1 files changed, 51 insertions, 0 deletions
diff --git a/internal/commands/update.go b/internal/commands/update.go
new file mode 100644
index 0000000..66ccfb3
--- /dev/null
+++ b/internal/commands/update.go
@@ -0,0 +1,51 @@
+package commands
+
+import (
+	"fmt"
+
+	"github.com/Fuwn/yae/internal/yae"
+	"github.com/urfave/cli/v2"
+)
+
+func Update(sources *yae.Sources) func(c *cli.Context) error {
+	return func(c *cli.Context) error {
+		updates := []string{}
+		force := c.Bool("force-hashed")
+		forcePinned := c.Bool("force-pinned")
+
+		if c.Args().Len() == 0 {
+			for name, source := range *sources {
+				if updated, err := source.Update(sources, name, force, forcePinned); err != nil {
+					return err
+				} else if updated {
+					updates = append(updates, name)
+				}
+			}
+		} else {
+			name := c.Args().Get(0)
+			source := (*sources)[name]
+
+			if updated, err := source.Update(sources, name, force, forcePinned); err != nil {
+				return err
+			} else if updated {
+				updates = append(updates, name)
+			}
+		}
+
+		if len(updates) > 0 {
+			if err := sources.Save(c.String("sources")); err != nil {
+				return err
+			}
+		}
+
+		if c.Bool("output-updated-list") {
+			for _, update := range updates {
+				fmt.Println(update)
+			}
+		} else if c.Bool("output-formatted-updated-list") {
+			fmt.Println(yae.Lister(updates))
+		}
+
+		return nil
+	}
+}