package server

import (
	"context"
	"os"
	"path/filepath"

	"github.com/VictorVargas/rony-chat-bot/internal/portfolio"
)

type ragStore = *portfolio.Store

func osWriteFile(path string, data []byte, perm os.FileMode) error {
	if err := os.MkdirAll(filepath.Dir(path), 0o755); err != nil {
		return err
	}
	return os.WriteFile(path, data, perm)
}

func ragOpenStore(dbPath, srcDir string) (ragStore, error) {
	store, err := portfolio.OpenStore(dbPath)
	if err != nil {
		return nil, err
	}
	if _, _, err := store.Reindex(context.Background(), portfolio.SourcesFor(srcDir, ""), portfolio.DefaultChunkerConfig()); err != nil {
		_ = store.Close()
		return nil, err
	}
	return store, nil
}