cmd/vpkflags: Implement vpkflags command

cmd/cmd.go

@@ -7,6 +7,7 @@ import (
 	_ "github.com/pg9182/tf2vpk/cmd/lzham"
 	_ "github.com/pg9182/tf2vpk/cmd/verify"
 	_ "github.com/pg9182/tf2vpk/cmd/version"
+	_ "github.com/pg9182/tf2vpk/cmd/vpkflags"
 )
 
 func Execute() {

cmd/vpkflags/main.go

@@ -0,0 +1,54 @@
+package vpkflags
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/pg9182/tf2vpk"
+	"github.com/pg9182/tf2vpk/cmd/root"
+	"github.com/pg9182/tf2vpk/vpkutil"
+	"github.com/spf13/cobra"
+)
+
+var Flags struct {
+	VPK      tf2vpk.ValvePak
+	Explicit bool
+}
+
+var Command = &cobra.Command{
+	Use:   "vpkflags vpk_path",
+	Short: "Generates a vpkflags file for the specified VPK.",
+	Args:  cobra.ExactArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		main()
+	},
+	GroupID: root.GroupVPKRepacking.ID,
+}
+
+func init() {
+	root.ArgVPK(&Flags.VPK, Command, -1, false, false, false)
+	Command.Flags().BoolVarP(&Flags.Explicit, "explicit", "x", false, "do not compute inherited vpkflags; generate one line for each file")
+	root.Command.AddCommand(Command)
+}
+
+func main() {
+	r, err := tf2vpk.NewReader(Flags.VPK)
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "error: open vpk: %v\n", err)
+		os.Exit(1)
+	}
+
+	var vpkflags vpkutil.VPKFlags
+	if Flags.Explicit {
+		err = vpkflags.GenerateExplicit(r.Root)
+	} else {
+		err = vpkflags.Generate(r.Root)
+	}
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "error: generate vpkflags: %v\n", err)
+		os.Exit(1)
+	}
+	if _, err := os.Stdout.WriteString(vpkflags.String()); err != nil {
+		os.Exit(1)
+	}
+}