feat(binana): add tokens database

go/app/cmd/query/query.go

@@ -0,0 +1,72 @@
+package query
+
+import (
+	"math"
+
+	"github.com/spf13/cobra"
+	"github.com/thunderbrewhq/binana/go/app"
+	"github.com/thunderbrewhq/binana/go/app/cmd/root"
+	"github.com/thunderbrewhq/binana/go/app/util"
+)
+
+var query_cmd = cobra.Command{
+	Use:   "q regexp",
+	Args:  cobra.MinimumNArgs(1),
+	Short: "query the token database for information",
+	Run:   run_query_cmd,
+}
+
+func init() {
+	f := query_cmd.Flags()
+	f.Uint32("min-build", 0, "the minimum build to return tokens for")
+	f.Uint32("max-build", math.MaxUint32, "the maximum build to return tokens for")
+	f.StringSlice("program", nil, "a list of programs to return tokens for")
+	f.StringSlice("os", nil, "a list of kernel names to return tokens for (windows, darwin, linux)")
+	f.StringSlice("arch", nil, "a list of CPU architectures to return tokens for (ppc, 386, amd64)")
+	f.String("present", "normal", "control the way tokens are presented to console (normal, name-only)")
+	root.RootCmd.AddCommand(&query_cmd)
+}
+
+func run_query_cmd(cmd *cobra.Command, args []string) {
+	f := cmd.Flags()
+	var (
+		params            util.QueryParams
+		err               error
+		presentation_mode string
+	)
+	params.MinBuild, err = f.GetUint32("min-build")
+	if err != nil {
+		app.Fatal(err)
+	}
+	params.MaxBuild, err = f.GetUint32("max-build")
+	if err != nil {
+		app.Fatal(err)
+	}
+	params.Program, err = f.GetStringSlice("program")
+	if err != nil {
+		app.Fatal(err)
+	}
+	params.OS, err = f.GetStringSlice("os")
+	if err != nil {
+		return
+	}
+	params.Arch, err = f.GetStringSlice("arch")
+	if err != nil {
+		return
+	}
+	presentation_mode, err = f.GetString("present")
+	if err != nil {
+		return
+	}
+	switch presentation_mode {
+	case "normal":
+		params.Present = util.PresentQueryNormal
+	case "name-only":
+		params.Present = util.PresentQueryNameOnly
+	default:
+		cmd.Help()
+		return
+	}
+	params.Token = args[0]
+	util.Query(&params)
+}