ooni-probe-cli/internal/database/database.go

package database

import (
	"path/filepath"

	"github.com/apex/log"
	"github.com/jmoiron/sqlx"
	_ "github.com/mattn/go-sqlite3" // this is needed to load the sqlite3 driver
	ooni "github.com/openobservatory/gooni"
	"github.com/openobservatory/gooni/internal/bindata"
	"github.com/pkg/errors"
	migrate "github.com/rubenv/sql-migrate"
)

// RunMigrations runs the database migrations
func RunMigrations(db *sqlx.DB) error {
	log.Debugf("running migrations")
	migrations := &migrate.AssetMigrationSource{
		Asset:    bindata.Asset,
		AssetDir: bindata.AssetDir,
		Dir:      "data/migrations",
	}
	n, err := migrate.Exec(db.DB, "sqlite3", migrations, migrate.Up)
	if err != nil {
		return err
	}
	log.Debugf("performed %d migrations", n)
	return nil
}

// Connect to the database
func Connect(path string) (*sqlx.DB, error) {
	db, err := sqlx.Connect("sqlite3", path)
	if err != nil {
		return nil, err
	}
	err = RunMigrations(db)
	if err != nil {
		return nil, err
	}
	return db, nil
}

// DefaultDatabasePath for the main database
func DefaultDatabasePath() (string, error) {
	home, err := ooni.GetOONIHome()
	if err != nil {
		return "", errors.Wrap(err, "default database path")
	}
	return filepath.Join(home, "db", "main.sqlite3"), nil
}