Add bpm-repo functionality

src/bpm-utils-shared/database.go

@@ -0,0 +1,127 @@
+package bpm_utils_shared
+
+import (
+	"fmt"
+	"io/fs"
+	"log"
+	"os"
+	"os/exec"
+	"path"
+	"path/filepath"
+	"strconv"
+	"strings"
+
+	"gopkg.in/yaml.v3"
+)
+
+type BPMDatabase struct {
+	DatabaseVersion int                         `yaml:"database_version"`
+	Entries         map[string]BPMDatabaseEntry `yaml:"entries"`
+}
+
+type BPMDatabaseEntry struct {
+	PackageInfo   *PackageInfo `yaml:"info"`
+	Filepath      string       `yaml:"filepath"`
+	DownloadSize  int64        `yaml:"download_size"`
+	InstalledSize int64        `yaml:"installed_size"`
+}
+
+func ReadDatabase(path string) (*BPMDatabase, error) {
+	// Read database file
+	output, err := os.ReadFile(path)
+	if err != nil {
+		return nil, err
+	}
+
+	database := &BPMDatabase{}
+
+	// Unmarshal yaml
+	err = yaml.Unmarshal(output, &database)
+	if err != nil {
+		return nil, err
+	}
+
+	return database, nil
+}
+
+func GenerateDatabase(path string) error {
+	database := BPMDatabase{
+		DatabaseVersion: 2,
+		Entries:         make(map[string]BPMDatabaseEntry),
+	}
+
+	err := filepath.Walk(path, func(packagePath string, info fs.FileInfo, err error) error {
+		if !strings.HasSuffix(packagePath, ".bpm") {
+			return nil
+		}
+
+		// Initialize database entry
+		entry := BPMDatabaseEntry{}
+		entry.DownloadSize = info.Size()
+		entry.Filepath, err = filepath.Rel(path, packagePath)
+		if err != nil {
+			return err
+		}
+
+		// Get package installed size
+		cmd := exec.Command("tar", "-t", "-v", "-f", packagePath, "files.tar.gz")
+		output, err := cmd.Output()
+		if err == nil {
+			entry.InstalledSize, err = strconv.ParseInt(strings.Fields(string(output))[2], 10, 64)
+			if err != nil {
+				return err
+			}
+		} else if err.(*exec.ExitError).ExitCode() == 2 {
+			entry.InstalledSize = 0
+		} else {
+			return err
+		}
+
+		// Get package info
+		entry.PackageInfo, err = ReadPacakgeInfo(packagePath)
+		if err != nil {
+			return err
+		}
+
+		// Add entry to database
+		if _, ok := database.Entries[entry.PackageInfo.Name]; ok {
+			return fmt.Errorf("package (%s) has already been added to the database", entry.PackageInfo.Name)
+		}
+		database.Entries[entry.PackageInfo.Name] = entry
+
+		return nil
+	})
+	if err != nil {
+		return err
+	}
+
+	data, err := yaml.Marshal(&database)
+	if err != nil {
+		return err
+	}
+
+	err = os.WriteFile(filepath.Join(path, "database.bpmdb"), data, 0644)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
+func UpdateDatabases(repo string) {
+	if _, err := os.Stat(path.Join(repo, "source")); err == nil {
+		err = GenerateDatabase(path.Join(repo, "source"))
+		if err != nil {
+			log.Fatalf("Error: could not generate source directory database: %s", err)
+		}
+		fmt.Println("Source directory database was generated successfully!")
+	}
+
+	if _, err := os.Stat(path.Join(repo, "binary")); err == nil {
+		err = GenerateDatabase(path.Join(repo, "binary"))
+		if err != nil {
+			log.Fatalf("Error: could not generate binary directory database: %s", err)
+		}
+		fmt.Println("Binary directory database was generated successfully!")
+	}
+}

src/bpm-utils-shared/package.go

@@ -0,0 +1,62 @@
+package bpm_utils_shared
+
+import (
+	"os/exec"
+
+	"gopkg.in/yaml.v3"
+)
+
+type PackageInfo struct {
+	Name            string         `yaml:"name,omitempty"`
+	Description     string         `yaml:"description,omitempty"`
+	Version         string         `yaml:"version,omitempty"`
+	Revision        int            `yaml:"revision,omitempty"`
+	Url             string         `yaml:"url,omitempty"`
+	License         string         `yaml:"license,omitempty"`
+	Arch            string         `yaml:"architecture,omitempty"`
+	Type            string         `yaml:"type,omitempty"`
+	Keep            []string       `yaml:"keep,omitempty"`
+	Depends         []string       `yaml:"depends,omitempty"`
+	MakeDepends     []string       `yaml:"make_depends,omitempty"`
+	OptionalDepends []string       `yaml:"optional_depends,omitempty"`
+	Conflicts       []string       `yaml:"conflicts,omitempty"`
+	Replaces        []string       `yaml:"replaces,omitempty"`
+	Provides        []string       `yaml:"provides,omitempty"`
+	SplitPackages   []*PackageInfo `yaml:"split_packages,omitempty"`
+}
+
+func ReadPacakgeInfo(path string) (*PackageInfo, error) {
+	// Extract package info using tar
+	cmd := exec.Command("tar", "-x", "-f", path, "pkg.info", "-O")
+	output, err := cmd.Output()
+	if err != nil {
+		return nil, err
+	}
+
+	pkgInfo := &PackageInfo{
+		Name:            "",
+		Description:     "",
+		Version:         "",
+		Revision:        1,
+		Url:             "",
+		License:         "",
+		Arch:            "",
+		Type:            "",
+		Keep:            make([]string, 0),
+		Depends:         make([]string, 0),
+		MakeDepends:     make([]string, 0),
+		OptionalDepends: make([]string, 0),
+		Conflicts:       make([]string, 0),
+		Replaces:        make([]string, 0),
+		Provides:        make([]string, 0),
+		SplitPackages:   make([]*PackageInfo, 0),
+	}
+
+	// Unmarshal yaml
+	err = yaml.Unmarshal(output, pkgInfo)
+	if err != nil {
+		return nil, err
+	}
+
+	return pkgInfo, nil
+}

src/bpm-utils-shared/repo.go

@@ -0,0 +1,25 @@
+package bpm_utils_shared
+
+import (
+	"os"
+	"path"
+)
+
+func GetRepository() string {
+	dir, err := os.Getwd()
+	if err != nil {
+		return ""
+	}
+
+	for dir != "/" {
+		if _, err := os.Stat(path.Join(dir, "bpm-repo.conf")); err == nil {
+			return dir
+		}
+		dir = path.Dir(dir)
+	}
+
+	if dir == "/" {
+		return ""
+	}
+	return dir
+}