/*
 *     Copyright 2020 The Dragonfly Authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package storedriver

import (
	"fmt"
	"path/filepath"
	"strings"

	"github.com/mitchellh/mapstructure"

	"d7y.io/dragonfly/v2/cdn/plugins"
	"d7y.io/dragonfly/v2/pkg/util/fileutils"
)

// DriverBuilder is a function that creates a new storage driver plugin instant with the giving Config.
type DriverBuilder func(cfg *Config) (Driver, error)

// Register defines an interface to register a driver with specified name.
// All drivers should call this function to register itself to the driverFactory.
func Register(name string, builder DriverBuilder) error {
	name = strings.ToLower(name)
	// plugin builder
	var f = func(conf interface{}) (plugins.Plugin, error) {
		cfg := &Config{}
		if err := mapstructure.Decode(conf, cfg); err != nil {
			return nil, fmt.Errorf("parse config: %v", err)
		}
		// prepare the base dir
		if !filepath.IsAbs(cfg.BaseDir) {
			return nil, fmt.Errorf("not absolute path: %s", cfg.BaseDir)
		}
		if err := fileutils.MkdirAll(cfg.BaseDir); err != nil {
			return nil, fmt.Errorf("create baseDir %s: %v", cfg.BaseDir, err)
		}

		return newDriverPlugin(name, builder, cfg)
	}
	return plugins.RegisterPluginBuilder(plugins.StorageDriverPlugin, name, f)
}

// Get a store from manager with specified name.
func Get(name string) (Driver, bool) {
	v, ok := plugins.GetPlugin(plugins.StorageDriverPlugin, strings.ToLower(name))
	if !ok {
		return nil, false
	}
	return v.(*driverPlugin).instance, true
}