wip: reorganized to go pkg structure

pkg/storage/interfaces.go

@@ -0,0 +1 @@
+package storage

pkg/storage/storage.go

@@ -0,0 +1,83 @@
+package storage
+
+import (
+	"errors"
+	"fmt"
+	"io/fs"
+	"log/slog"
+	"os"
+	"path"
+	"path/filepath"
+
+	co "github.com/hohn/ghes-mirva-server/common"
+)
+
+type StorageSingle struct {
+	CurrentID int
+}
+
+func (s *StorageSingle) NextID() int {
+	s.CurrentID += 1
+	return s.CurrentID
+}
+
+func (s *StorageSingle) SaveQueryPack(tgz []byte, sessionId int) (string, error) {
+	// Save the tar.gz body
+	cwd, err := os.Getwd()
+	if err != nil {
+		slog.Error("No working directory")
+		panic(err)
+	}
+
+	dirpath := path.Join(cwd, "var", "codeql", "querypacks")
+	if err := os.MkdirAll(dirpath, 0755); err != nil {
+		slog.Error("Unable to create query pack output directory",
+			"dir", dirpath)
+		return "", err
+	}
+
+	fpath := path.Join(dirpath, fmt.Sprintf("qp-%d.tgz", sessionId))
+	err = os.WriteFile(fpath, tgz, 0644)
+	if err != nil {
+		slog.Error("unable to save querypack body decoding error", "path", fpath)
+		return "", err
+	} else {
+		slog.Info("Query pack saved to ", "path", fpath)
+	}
+
+	return fpath, nil
+}
+
+//		Determine for which repositories codeql databases are available.
+//
+//	 Those will be the analysis_repos.  The rest will be skipped.
+func (s *StorageSingle) FindAvailableDBs(analysisReposRequested []co.OwnerRepo) (not_found_repos []co.OwnerRepo,
+	analysisRepos *map[co.OwnerRepo]DBLocation) {
+	slog.Debug("Looking for available CodeQL databases")
+
+	cwd, err := os.Getwd()
+	if err != nil {
+		slog.Error("No working directory")
+		return
+	}
+
+	analysisRepos = &map[co.OwnerRepo]DBLocation{}
+
+	not_found_repos = []co.OwnerRepo{}
+
+	for _, rep := range analysisReposRequested {
+		dbPrefix := filepath.Join(cwd, "codeql", "dbs", rep.Owner, rep.Repo)
+		dbName := fmt.Sprintf("%s_%s_db.zip", rep.Owner, rep.Repo)
+		dbPath := filepath.Join(dbPrefix, dbName)
+
+		if _, err := os.Stat(dbPath); errors.Is(err, fs.ErrNotExist) {
+			slog.Info("Database does not exist for repository ", "owner/repo", rep,
+				"path", dbPath)
+			not_found_repos = append(not_found_repos, rep)
+		} else {
+			slog.Info("Found database for ", "owner/repo", rep, "path", dbPath)
+			(*analysisRepos)[rep] = DBLocation{Prefix: dbPrefix, File: dbName}
+		}
+	}
+	return not_found_repos, analysisRepos
+}

pkg/storage/types.go

@@ -0,0 +1,17 @@
+package storage
+
+import (
+	co "github.com/hohn/ghes-mirva-server/common"
+)
+
+type Storage interface {
+	NextID() int
+	SaveQueryPack(tgz []byte, sessionID int) (storagePath string, error error)
+	FindAvailableDBs(analysisReposRequested []co.OwnerRepo) (not_found_repos []co.OwnerRepo,
+		analysisRepos *map[co.OwnerRepo]DBLocation)
+}
+
+type DBLocation struct {
+	Prefix string
+	File   string
+}