// 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 ksyun

import (
	"fmt"

	api "yunion.io/x/cloudmux/pkg/apis/compute"
	"yunion.io/x/cloudmux/pkg/cloudprovider"
	"yunion.io/x/cloudmux/pkg/multicloud"
	"yunion.io/x/jsonutils"
	"yunion.io/x/pkg/errors"
)

type SRegion struct {
	multicloud.SRegion
	multicloud.SNoObjectStorageRegion
	multicloud.SNoLbRegion
	client *SKsyunClient

	Region     string
	RegionName string
}

func (region *SRegion) GetId() string {
	return region.Region
}

func (region *SRegion) GetGlobalId() string {
	return fmt.Sprintf("%s/%s", api.CLOUD_PROVIDER_KSYUN, region.Region)
}

func (region *SRegion) GetProvider() string {
	return api.CLOUD_PROVIDER_KSYUN
}

func (region *SRegion) GetCloudEnv() string {
	return api.CLOUD_PROVIDER_KSYUN
}

func (region *SRegion) GetGeographicInfo() cloudprovider.SGeographicInfo {
	geo, ok := map[string]cloudprovider.SGeographicInfo{
		"cn-northwest-1": api.RegionQingYang,
		"ap-singapore-1": api.RegionSingapore,
		"cn-beijing-6":   api.RegionBeijing,
		"cn-guangzhou-1": api.RegionGuangzhou,
		"cn-shanghai-2":  api.RegionShanghai,
	}[region.Region]
	if ok {
		return geo
	}
	return cloudprovider.SGeographicInfo{}
}

func (region *SRegion) GetName() string {
	return region.RegionName
}

func (region *SRegion) GetI18n() cloudprovider.SModelI18nTable {
	table := cloudprovider.SModelI18nTable{}
	table["name"] = cloudprovider.NewSModelI18nEntry(region.GetName()).CN(region.GetName()).EN(region.Region)
	return table
}

func (region *SRegion) GetStatus() string {
	return api.CLOUD_REGION_STATUS_INSERVER
}

func (region *SRegion) GetClient() *SKsyunClient {
	return region.client
}

func (region *SRegion) CreateEIP(opts *cloudprovider.SEip) (cloudprovider.ICloudEIP, error) {
	return region.CreateEip(opts)
}

func (region *SRegion) CreateISecurityGroup(conf *cloudprovider.SecurityGroupCreateInput) (cloudprovider.ICloudSecurityGroup, error) {
	return region.CreateSecurityGroup(conf)
}

func (region *SRegion) CreateIVpc(opts *cloudprovider.VpcCreateOptions) (cloudprovider.ICloudVpc, error) {
	return region.CreateVpc(opts)
}

func (region *SRegion) GetIVpcs() ([]cloudprovider.ICloudVpc, error) {
	vpcs, err := region.GetVpcs([]string{})
	if err != nil {
		return nil, errors.Wrap(err, "GetVpcs")
	}
	res := []cloudprovider.ICloudVpc{}
	for i := 0; i < len(vpcs); i++ {
		vpcs[i].region = region
		res = append(res, &vpcs[i])
	}
	return res, nil
}

func (region *SRegion) GetIVpcById(id string) (cloudprovider.ICloudVpc, error) {
	vpc, err := region.GetVpc(id)
	if err != nil {
		return nil, errors.Wrap(err, "region.GetVpc")
	}
	vpc.region = region
	return vpc, nil
}

func (region *SRegion) GetCapabilities() []string {
	return region.client.GetCapabilities()
}

func (region *SRegion) GetISecurityGroupById(secgroupId string) (cloudprovider.ICloudSecurityGroup, error) {
	group, err := region.GetSecurityGroup(secgroupId)
	if err != nil {
		return nil, errors.Wrap(err, "region.GetSecurityGroup")
	}
	group.region = region
	return group, nil
}

func (region *SRegion) GetIEipById(eipId string) (cloudprovider.ICloudEIP, error) {
	eip, err := region.GetEip(eipId)
	if err != nil {
		return nil, errors.Wrap(err, "GetEipById")
	}
	eip.region = region
	return eip, nil
}

func (region *SRegion) GetIHosts() ([]cloudprovider.ICloudHost, error) {
	zones, err := region.GetIZones()
	if err != nil {
		return nil, errors.Wrap(err, "GetIZones")
	}
	hosts := []cloudprovider.ICloudHost{}
	for _, zone := range zones {
		zoneHosts, err := zone.GetIHosts()
		if err != nil {
			return nil, errors.Wrap(err, "zone.GetIHosts")
		}
		hosts = append(hosts, zoneHosts...)
	}
	return hosts, nil
}

func (region *SRegion) GetIHostById(id string) (cloudprovider.ICloudHost, error) {
	hosts, err := region.GetIHosts()
	if err != nil {
		return nil, err
	}
	for _, host := range hosts {
		if host.GetId() == id {
			return host, nil
		}
	}
	return nil, errors.ErrNotFound
}

func (region *SRegion) GetIStorageById(id string) (cloudprovider.ICloudStorage, error) {
	zones, err := region.GetIZones()
	if err != nil {
		return nil, errors.Wrap(err, "region.GetIZones")
	}
	for _, zone := range zones {
		storage, err := zone.GetIStorageById(id)
		if err == nil {
			return storage, nil
		} else if errors.Cause(err) != cloudprovider.ErrNotFound {
			return nil, errors.Wrap(err, "GetIStorageById")
		}
	}
	return nil, errors.ErrNotFound
}

func (region *SRegion) GetIEips() ([]cloudprovider.ICloudEIP, error) {
	eips, err := region.GetEips([]string{})
	if err != nil {
		return nil, errors.Wrap(err, "GetEips")
	}
	res := []cloudprovider.ICloudEIP{}
	for i := 0; i < len(eips); i++ {
		eips[i].region = region
		res = append(res, &eips[i])
	}
	return res, nil
}

func (region *SRegion) GetIZones() ([]cloudprovider.ICloudZone, error) {
	zones, err := region.GetZones()
	if err != nil {
		return nil, errors.Wrap(err, "GetZones")
	}
	res := []cloudprovider.ICloudZone{}
	for i := 0; i < len(zones); i++ {
		zones[i].region = region
		res = append(res, &zones[i])
	}
	return res, nil
}

func (region *SRegion) GetIZoneById(id string) (cloudprovider.ICloudZone, error) {
	izones, err := region.GetIZones()
	if err != nil {
		return nil, errors.Wrap(err, "GetIZones")
	}
	for _, izone := range izones {
		if izone.GetGlobalId() == id {
			return izone, nil
		}
	}
	return nil, errors.Wrapf(cloudprovider.ErrNotFound, "zone id:%s", id)
}

func (region *SRegion) GetIStorages() ([]cloudprovider.ICloudStorage, error) {
	iStores := make([]cloudprovider.ICloudStorage, 0)
	izones, err := region.GetIZones()
	if err != nil {
		return nil, err
	}
	for _, izone := range izones {
		iZoneStores, err := izone.GetIStorages()
		if err != nil {
			return nil, err
		}
		iStores = append(iStores, iZoneStores...)
	}
	return iStores, nil
}

func (r *SRegion) GetIVMs() ([]cloudprovider.ICloudVM, error) {
	vms, err := r.GetInstances("", nil)
	if err != nil {
		return nil, err
	}
	ret := []cloudprovider.ICloudVM{}
	for i := range vms {
		vms[i].region = r
		ret = append(ret, &vms[i])
	}
	return ret, nil
}

func (r *SRegion) GetIVMById(id string) (cloudprovider.ICloudVM, error) {
	vm, err := r.GetInstance(id)
	if err != nil {
		return nil, err
	}
	vm.region = r
	return vm, nil
}

func (region *SRegion) ecsRequest(action string, params map[string]interface{}) (jsonutils.JSONObject, error) {
	return region.client.ec2Request(region.Region, action, params)
}

func (region *SRegion) tagRequest(action string, params map[string]interface{}) (jsonutils.JSONObject, error) {
	return region.client.tagRequest(region.Region, action, params)
}

func (region *SRegion) eipRequest(action string, params map[string]interface{}) (jsonutils.JSONObject, error) {
	return region.client.eipRequest(region.Region, action, params)
}

func (region *SRegion) ebsRequest(action string, params map[string]interface{}) (jsonutils.JSONObject, error) {
	return region.client.ebsRequest(region.Region, action, params)
}

func (region *SRegion) vpcRequest(action string, params map[string]interface{}) (jsonutils.JSONObject, error) {
	return region.client.vpcRequest(region.Region, action, params)
}

func (region *SRegion) rdsRequest(action string, params map[string]interface{}) (jsonutils.JSONObject, error) {
	return region.client.rdsRequest(region.Region, action, params)
}