1 files changed, 97 insertions, 0 deletions
diff --git a/internal/programs/programs.go b/internal/programs/programs.go
new file mode 100644
index 0000000..1dbfff7
--- /dev/null
+++ b/internal/programs/programs.go
@@ -0,0 +1,97 @@
+package programs
+
+import (
+	"context"
+	"database/sql"
+	"fmt"
+	"os/exec"
+	"strings"
+
+	"github.com/pkg/errors"
+	"go.alanpearce.eu/searchix/internal/config"
+	"go.alanpearce.eu/x/log"
+	_ "modernc.org/sqlite" //nolint:blank-imports // sqlite driver needed for database/sql
+)
+
+type DB struct {
+	Path   string
+	Source *config.Source
+
+	logger *log.Logger
+	db     *sql.DB
+}
+
+func Instantiate(ctx context.Context, source *config.Source, logger *log.Logger) (*DB, error) {
+	// nix-instantiate --eval --json -I nixpkgs=channel:nixos-unstable --expr 'toString <nixpkgs/programs.sqlite>'
+	args := []string{
+		"--eval",
+		"--json",
+		"-I", fmt.Sprintf("%s=channel:%s", source.Key, source.Channel),
+		"--expr", fmt.Sprintf("toString <%s/%s>", source.Key, source.Programs.Attribute),
+	}
+
+	logger.Debug("nix-instantiate command", "args", args)
+	cmd := exec.CommandContext(ctx, "nix-instantiate", args...)
+	out, err := cmd.Output()
+	if err != nil {
+		return nil, errors.WithMessage(err, "failed to run nix-instantiate")
+	}
+
+	outPath := strings.Trim(strings.TrimSpace(string(out)), "\"")
+	logger.Debug("got output path", "outputPath", outPath)
+
+	return &DB{
+		Source: source,
+		Path:   outPath,
+
+		logger: logger,
+	}, nil
+}
+
+func (p *DB) Open() error {
+	db, err := sql.Open("sqlite", p.Path)
+	if err != nil {
+		return errors.WithMessage(err, "failed to open sqlite database")
+	}
+	p.db = db
+
+	return nil
+}
+
+func (p *DB) Close() error {
+	if err := p.db.Close(); err != nil {
+		return errors.WithMessage(err, "failed to close sqlite database")
+	}
+
+	return nil
+}
+
+func (p *DB) GetPackagePrograms(ctx context.Context, pkg string) (programs []string, err error) {
+	if p.db == nil {
+		return nil, errors.New("database not open")
+	}
+	rows, err := p.db.QueryContext(ctx, `
+SELECT name
+FROM Programs
+WHERE package = ?
+GROUP BY name, package`, pkg)
+	if err != nil {
+		return nil, errors.WithMessage(err, "failed to execute query")
+	}
+	defer rows.Close()
+
+	for rows.Next() {
+		var name string
+		if err := rows.Scan(&name); err != nil {
+			return nil, errors.WithMessage(err, "failed to scan row")
+		}
+
+		programs = append(programs, name)
+	}
+	rerr := rows.Close()
+	if rerr != nil {
+		return nil, errors.WithMessage(rerr, "sql error")
+	}
+
+	return
+}