package registry

import (
	"net/url"
	"os"
	"path/filepath"

	"forge.cadoles.com/arcad/edge/pkg/storage"
	"forge.cadoles.com/arcad/edge/pkg/storage/registry"
	"forge.cadoles.com/arcad/edge/pkg/storage/sqlite"
	"github.com/pkg/errors"
)

func init() {
	registry.AddDocumentStoreFactory("sqlite", documentStoreFactory)
	registry.AddBlobStoreFactory("sqlite", blobStoreFactory)
}

func documentStoreFactory(url *url.URL) (storage.DocumentStore, error) {
	dir := filepath.Dir(url.Host + url.Path)

	if err := os.MkdirAll(dir, os.FileMode(0750)); err != nil {
		return nil, errors.WithStack(err)
	}

	path := url.Host + url.Path + "?" + url.RawQuery

	db, err := sqlite.Open(path)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	return sqlite.NewDocumentStoreWithDB(db), nil
}

func blobStoreFactory(url *url.URL) (storage.BlobStore, error) {
	dir := filepath.Dir(url.Host + url.Path)

	if err := os.MkdirAll(dir, os.FileMode(0750)); err != nil {
		return nil, errors.WithStack(err)
	}

	path := url.Host + url.Path + "?" + url.RawQuery

	db, err := sqlite.Open(path)
	if err != nil {
		return nil, errors.WithStack(err)
	}

	return sqlite.NewBlobStoreWithDB(db), nil
}