Fully implement local and container MRVA
This commit is contained in:
66
pkg/qldbstore/qldbstore_local.go
Normal file
66
pkg/qldbstore/qldbstore_local.go
Normal file
@@ -0,0 +1,66 @@
|
||||
package qldbstore
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"mrvacommander/pkg/common"
|
||||
"os"
|
||||
"path/filepath"
|
||||
)
|
||||
|
||||
type FilesystemCodeQLDatabaseStore struct {
|
||||
basePath string
|
||||
}
|
||||
|
||||
func NewLocalFilesystemCodeQLDatabaseStore(basePath string) *FilesystemCodeQLDatabaseStore {
|
||||
return &FilesystemCodeQLDatabaseStore{
|
||||
basePath: basePath,
|
||||
}
|
||||
}
|
||||
|
||||
func (store *FilesystemCodeQLDatabaseStore) FindAvailableDBs(analysisReposRequested []common.NameWithOwner) (
|
||||
notFoundRepos []common.NameWithOwner,
|
||||
foundRepos *map[common.NameWithOwner]CodeQLDatabaseLocation) {
|
||||
|
||||
foundReposMap := make(map[common.NameWithOwner]CodeQLDatabaseLocation)
|
||||
for _, repo := range analysisReposRequested {
|
||||
location, err := store.GetDatabaseLocationByNWO(repo)
|
||||
if err != nil {
|
||||
notFoundRepos = append(notFoundRepos, repo)
|
||||
} else {
|
||||
foundReposMap[repo] = location
|
||||
}
|
||||
}
|
||||
|
||||
return notFoundRepos, &foundReposMap
|
||||
}
|
||||
|
||||
func (store *FilesystemCodeQLDatabaseStore) GetDatabase(location CodeQLDatabaseLocation) ([]byte, error) {
|
||||
path, exists := location.data["path"]
|
||||
if !exists {
|
||||
return nil, fmt.Errorf("path not specified in location")
|
||||
}
|
||||
|
||||
data, err := os.ReadFile(path)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return data, nil
|
||||
}
|
||||
|
||||
func (store *FilesystemCodeQLDatabaseStore) GetDatabaseLocationByNWO(nwo common.NameWithOwner) (CodeQLDatabaseLocation, error) {
|
||||
filePath := filepath.Join(store.basePath, fmt.Sprintf("%s/%s/%s_%s_db.zip", nwo.Owner, nwo.Repo, nwo.Owner, nwo.Repo))
|
||||
|
||||
// Check if the file exists
|
||||
if _, err := os.Stat(filePath); os.IsNotExist(err) {
|
||||
return CodeQLDatabaseLocation{}, fmt.Errorf("database not found for %s", nwo)
|
||||
}
|
||||
|
||||
location := CodeQLDatabaseLocation{
|
||||
data: map[string]string{
|
||||
"path": filePath,
|
||||
},
|
||||
}
|
||||
|
||||
return location, nil
|
||||
}
|
||||
Reference in New Issue
Block a user