package rpc

import (
	"net/url"

	"forge.cadoles.com/arcad/edge/pkg/storage"
	"forge.cadoles.com/arcad/edge/pkg/storage/driver"
	"forge.cadoles.com/arcad/edge/pkg/storage/driver/rpc/client"
	"forge.cadoles.com/arcad/edge/pkg/storage/share"
)

func init() {
	driver.RegisterDocumentStoreFactory("rpc", documentStoreFactory)
	driver.RegisterBlobStoreFactory("rpc", blobStoreFactory)
	driver.RegisterShareStoreFactory("rpc", shareStoreFactory)
}

func documentStoreFactory(url *url.URL) (storage.DocumentStore, error) {
	return client.NewDocumentStore(url), nil
}

func blobStoreFactory(url *url.URL) (storage.BlobStore, error) {
	return client.NewBlobStore(url), nil
}

func shareStoreFactory(url *url.URL) (share.Store, error) {
	return client.NewShareStore(url), nil
}