// Copyright 2019 Yunion
//
// 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 drivers

import (
	"fmt"

	"yunion.io/x/jsonutils"
	"yunion.io/x/log"
	"yunion.io/x/pkg/errors"

	api "yunion.io/x/onecloud/pkg/apis/compute"
	"yunion.io/x/onecloud/pkg/baremetal/utils/raid"
	_ "yunion.io/x/onecloud/pkg/baremetal/utils/raid/adaptec"
	_ "yunion.io/x/onecloud/pkg/baremetal/utils/raid/hpssactl"
	_ "yunion.io/x/onecloud/pkg/baremetal/utils/raid/mdadm"
	_ "yunion.io/x/onecloud/pkg/baremetal/utils/raid/megactl"
	_ "yunion.io/x/onecloud/pkg/baremetal/utils/raid/mvcli"
	_ "yunion.io/x/onecloud/pkg/baremetal/utils/raid/sas2iru"
	"yunion.io/x/onecloud/pkg/compute/baremetal"
)

func GetDriver(name string, term raid.IExecTerm) raid.IRaidDriver {
	factory := raid.RaidDrivers[name]
	if factory == nil {
		return nil
	}
	return factory(term)
}

func GetLocalDriver(name string) raid.IRaidDriver {
	factory := raid.RaidDrivers[name]
	if factory == nil {
		return nil
	}
	return factory(NewExecutor())
}

func GetDriverWithInit(name string, term raid.IExecTerm) (raid.IRaidDriver, error) {
	drv := GetDriver(name, term)
	if drv == nil {
		return nil, errors.Errorf("Not found raid driver %q", name)
	}
	return drv, drv.ParsePhyDevs()
}

func GetDriverByKernelModule(module string, term raid.IExecTerm) (raid.IRaidDriver, error) {
	name := ""
	switch module {
	case raid.MODULE_MEGARAID:
		name = baremetal.DISK_DRIVER_MEGARAID
	case raid.MODULE_HPSA:
		name = baremetal.DISK_DRIVER_HPSARAID
	case raid.MODULE_MPT2SAS, raid.MODULE_MPT3SAS:
		name = baremetal.DISK_DRIVER_MPT2SAS
	case raid.MODULE_AACRAID, raid.MODULE_SMARTPQI:
		name = baremetal.DISK_DRIVER_ADAPTECRAID
	}
	if name == "" {
		return nil, errors.Errorf("Not support module %q", module)
	}
	return GetDriverWithInit(name, term)
}

func GetDrivers(term raid.IExecTerm) []raid.IRaidDriver {
	ret := []raid.IRaidDriver{}
	for _, factory := range raid.RaidDrivers {
		ret = append(ret, factory(term))
	}
	return ret
}

func BuildRaid(driver raid.IRaidDriver, confs []*api.BaremetalDiskConfig, adapterIdx int) error {
	if err := driver.PreBuildRaid(confs, adapterIdx); err != nil {
		return fmt.Errorf("PreBuildRaid: %v", err)
	}
	var adapter raid.IRaidAdapter
	for _, tmp := range driver.GetAdapters() {
		if tmp.GetIndex() == adapterIdx {
			adapter = tmp
			break
		}
	}
	if adapter == nil {
		return fmt.Errorf("Not found adapter by index %d", adapterIdx)
	}
	if err := buildRaid(driver, adapter, confs); err != nil {
		return fmt.Errorf("Driver %s, adapter %d build raid: %v", driver.GetName(), adapterIdx, err)
	}
	return nil
}

func buildRaid(driver raid.IRaidDriver, adapter raid.IRaidAdapter, confs []*api.BaremetalDiskConfig) error {
	if err := adapter.PreBuildRaid(confs); err != nil {
		return fmt.Errorf("PreBuildRaid: %v", err)
	}
	if err := adapter.RemoveLogicVolumes(); err != nil {
		return fmt.Errorf("RemoveLogicVolumes: %v", err)
	}
	devs := adapter.GetDevices()
	if len(devs) == 0 {
		// no disk to build
		return nil
	}

	var selected []*baremetal.BaremetalStorage
	var nonDisks []*baremetal.BaremetalStorage
	var err error
	left := devs

	for _, conf := range confs {
		selected, left, err = baremetal.RetrieveStorages(conf, left)
		if len(selected) == 0 {
			return errors.Wrapf(err, "no enough disks for config %#v", conf)
		}
		var err error
		switch conf.Conf {
		case baremetal.DISK_CONF_RAID5:
			err = adapter.BuildRaid5(selected, conf)
		case baremetal.DISK_CONF_RAID10:
			err = adapter.BuildRaid10(selected, conf)
		case baremetal.DISK_CONF_NONE:
			nonDisks = append(nonDisks, selected...)
		case baremetal.DISK_CONF_RAID0:
			err = adapter.BuildRaid0(selected, conf)
		case baremetal.DISK_CONF_RAID1:
			err = adapter.BuildRaid1(selected, conf)
		default:
			return fmt.Errorf("Unknown raid config %s", conf.Conf)
		}
		if err != nil {
			return fmt.Errorf("Build raid %s: %v", conf.Conf, err)
		}
		log.Infof("Build %s:%d raid %s", driver.GetName(), adapter.GetIndex(), conf.Conf)
	}
	if len(nonDisks) > 0 {
		if err := adapter.BuildNoneRaid(nonDisks); err != nil {
			return fmt.Errorf("Build raw disks: %v", err)
		}
	}
	if err := adapter.PostBuildRaid(); err != nil {
		return errors.Wrap(err, "Post build raid")
	}
	return nil
}

func PostBuildRaid(driver raid.IRaidDriver, adapterIdx int) error {
	for _, a := range driver.GetAdapters() {
		if a.GetIndex() == adapterIdx {
			if err := a.PostBuildRaid(); err != nil {
				return errors.Wrap(err, "PostBuildRaid")
			}
		}
	}
	return nil
}

func GetFirstLogicalVolume(drv raid.IRaidDriver, adapterIdx int) (*raid.RaidLogicalVolume, error) {
	var adapter raid.IRaidAdapter = nil
	for _, ada := range drv.GetAdapters() {
		if ada.GetIndex() == adapterIdx {
			adapter = ada
		}
	}
	if adapter == nil {
		return nil, errors.Errorf("Not found raid %s adapter %d", drv.GetName(), adapterIdx)
	}
	lvs, err := adapter.GetLogicVolumes()
	if err != nil {
		return nil, errors.Wrapf(err, "Get adapter logical volume")
	}
	if len(lvs) == 0 {
		return nil, errors.Errorf("Adapter %d empty logical volume", adapterIdx)
	}
	return lvs[0], nil
}

func GetBlockDeviceLogicalVolume(drv raid.IRaidDriver, blockDev string) (*raid.RaidLogicalVolume, error) {
	lvs := make([]*raid.RaidLogicalVolume, 0)
	for _, ada := range drv.GetAdapters() {
		aLvs, err := ada.GetLogicVolumes()
		if err != nil {
			return nil, errors.Errorf("Get adapter %d logical volumes: %v", ada.GetIndex(), err)
		}
		lvs = append(lvs, aLvs...)
	}
	log.Debugf("Get logic volumes: %s", jsonutils.Marshal(lvs).String())
	for _, lv := range lvs {
		if lv.BlockDev == blockDev {
			return lv, nil
		}
	}
	return nil, errors.Errorf("Not found logical volume by block device %s", blockDev)
}