Merge pull request #110 from x-motemen/without-draft

add "--no-drafts" and "--only-drafts" flags to pull subcommand for drafts management
x-motemen · Oct 22, 2023 · 3174648 · 3174648
2 parents 5fe10ba + 9b95420
commit 3174648
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/broker.go b/broker.go
@@ -33,7 +33,7 @@ func newBroker(bc *blogConfig) *broker {
 	}
 }
 
-func (b *broker) FetchRemoteEntries() ([]*entry, error) {
+func (b *broker) FetchRemoteEntries(published, drafts bool) ([]*entry, error) {
 	entries := []*entry{}
 	fixedPageURL := fixedPageEndpointURL(b.blogConfig)
 	urls := []string{
@@ -63,7 +63,9 @@ func (b *broker) FetchRemoteEntries() ([]*entry, error) {
 			if err != nil {
 				return nil, err
 			}
-			entries = append(entries, e)
+			if (e.IsDraft && drafts) || (!e.IsDraft && published) {
+				entries = append(entries, e)
+			}
 		}
 
 		nextLink := feed.Links.Find("next")

diff --git a/main.go b/main.go
@@ -95,6 +95,10 @@ func loadConfigFiles(pwd string) (*config, error) {
 var commandPull = &cli.Command{
 	Name:  "pull",
 	Usage: "Pull entries from remote",
+	Flags: []cli.Flag{
+		&cli.BoolFlag{Name: "no-drafts"},
+		&cli.BoolFlag{Name: "only-drafts"},
+	},
 	Action: func(c *cli.Context) error {
 		blog := c.Args().First()
 		if blog == "" {
@@ -112,7 +116,8 @@ var commandPull = &cli.Command{
 		}
 
 		b := newBroker(blogConfig)
-		remoteEntries, err := b.FetchRemoteEntries()
+		remoteEntries, err := b.FetchRemoteEntries(
+			!c.Bool("only-drafts"), !c.Bool("no-drafts"))
 		if err != nil {
 			return err
 		}