waveterm/pkg/wstore/wstore_dbsetup.go

// Copyright 2024, Command Line Inc.
// SPDX-License-Identifier: Apache-2.0

package wstore

import (
	"context"
	"encoding/json"
	"fmt"
	"log"
	"path"
	"time"

	"github.com/jmoiron/sqlx"
	"github.com/sawka/txwrap"
	"github.com/wavetermdev/thenextwave/pkg/util/migrateutil"
	"github.com/wavetermdev/thenextwave/pkg/wavebase"

	dbfs "github.com/wavetermdev/thenextwave/db"
)

const WStoreDBName = "waveterm.db"

type TxWrap = txwrap.TxWrap

var globalDB *sqlx.DB

func InitWStore() error {
	ctx, cancelFn := context.WithTimeout(context.Background(), 2*time.Second)
	defer cancelFn()
	var err error
	globalDB, err = MakeDB(ctx)
	if err != nil {
		return err
	}
	err = migrateutil.Migrate("wstore", globalDB.DB, dbfs.WStoreMigrationFS, "migrations-wstore")
	if err != nil {
		return err
	}
	log.Printf("wstore initialized\n")
	return nil
}

func GetDBName() string {
	waveHome := wavebase.GetWaveHomeDir()
	return path.Join(waveHome, WStoreDBName)
}

func MakeDB(ctx context.Context) (*sqlx.DB, error) {
	dbName := GetDBName()
	rtn, err := sqlx.Open("sqlite3", fmt.Sprintf("file:%s?mode=rwc&_journal_mode=WAL&_busy_timeout=5000", dbName))
	if err != nil {
		return nil, err
	}
	rtn.DB.SetMaxOpenConns(1)
	return rtn, nil
}

func WithTx(ctx context.Context, fn func(tx *TxWrap) error) error {
	return txwrap.WithTx(ctx, globalDB, fn)
}

func WithTxRtn[RT any](ctx context.Context, fn func(tx *TxWrap) (RT, error)) (RT, error) {
	return txwrap.WithTxRtn(ctx, globalDB, fn)
}

func TxJson(tx *TxWrap, v any) string {
	barr, err := json.Marshal(v)
	if err != nil {
		tx.SetErr(fmt.Errorf("json marshal (%T): %w", v, err))
		return ""
	}
	return string(barr)
}

func TxReadJson[T any](tx *TxWrap, jsonData string) *T {
	if jsonData == "" {
		return nil
	}
	var v T
	err := json.Unmarshal([]byte(jsonData), &v)
	if err != nil {
		tx.SetErr(fmt.Errorf("json unmarshal (%T): %w", v, err))
	}
	return &v
}