add datadir and cmd/search.go

1 files changed, 76 insertions, 0 deletions

diff --git a/cmd/search.go b/cmd/search.go
new file mode 100644
index 0000000..ba831df
--- /dev/null
+++ b/cmd/search.go
@@ -0,0 +1,76 @@
+package cmd
+
+import (
+  "context"
+  "encoding/json"
+  "github.com/pablotron/cvez/datadir"
+  "github.com/pablotron/cvez/dbstore"
+  "github.com/spf13/cobra"
+  "github.com/rs/zerolog"
+  "github.com/rs/zerolog/log"
+  "os"
+  "strings"
+  "time"
+)
+
+// Build query string.
+func getSearchQuery(args []string) string {
+  r := make([]string, len(args))
+
+  for i, v := range(args) {
+    r[i] = "\"" + strings.ReplaceAll(strings.ToLower(v), "\"", "\"\"") + "\""
+  }
+
+  return strings.Join(r, " AND ")
+}
+
+// Get database store.
+func getDb() dbstore.DbStore {
+  // build database path
+  dbPath, err := datadir.Join("cvez.db")
+  if err != nil {
+    log.Error().Err(err).Msg("Join")
+    os.Exit(-1)
+  }
+
+  // open database
+  db, err := dbstore.Open(dbPath)
+  if err != nil {
+    log.Error().Err(err).Msg("Open")
+    os.Exit(-1)
+  }
+
+  return db
+}
+
+var searchCmd = &cobra.Command{
+  Use:   "search",
+  Args:  cobra.MinimumNArgs(1),
+  Short: "Search CVEs.",
+  Long: `Common Vulnerability Enumeration (CVE) search.`,
+
+  Run: func(cmd *cobra.Command, args []string) {
+    ctx := context.Background()
+
+    // set global logging options
+    zerolog.TimeFieldFormat = time.RFC3339
+    zerolog.SetGlobalLevel(zerolog.InfoLevel)
+
+    // search for CVEs
+    rows, err := getDb().CveSearch(ctx, getSearchQuery(args))
+    if err != nil {
+      log.Error().Err(err).Msg("CveSearch")
+      os.Exit(-1)
+    }
+
+    // search for CVEs, write result
+    e := json.NewEncoder(os.Stdout)
+    if err := e.Encode(rows); err != nil {
+      log.Error().Err(err).Msg("Encode")
+    }
+  },
+}
+
+func init() {
+  rootCmd.AddCommand(searchCmd)
+}