// Copyright (c) 2022 Yaohui Wang (yaohuiwang@outlook.com)
// FlexLB is licensed under Mulan PubL v2.
// You can use this software according to the terms and conditions of the Mulan PubL v2.
// You may obtain a copy of Mulan PubL v2 at:
//         http://license.coscl.org.cn/MulanPubL-2.0
// THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
// EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
// MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
// See the Mulan PubL v2 for more details.

package client

import (
	"encoding/json"
	"io/ioutil"

	"gitee.com/flexlb/flexlb-client-go/client/instance"
	"gitee.com/flexlb/flexlb-client-go/client/service"
	"gitee.com/flexlb/flexlb-client-go/models"

	httptransport "github.com/go-openapi/runtime/client"
	"github.com/go-openapi/strfmt"
)

func NewTLSClient(host string, ca string, cert string, key string, insecure bool, formats strfmt.Registry) (*Flexlb, error) {
	tlsOpts := httptransport.TLSClientOptions{
		CA:                 ca,
		Key:                key,
		Certificate:        cert,
		InsecureSkipVerify: insecure,
	}
	client, err := httptransport.TLSClient(tlsOpts)
	if err != nil {
		return nil, err
	}
	transport := httptransport.NewWithClient(host, DefaultBasePath, DefaultSchemes, client)
	return New(transport, formats), nil
}

func (lb *Flexlb) GetReadyStatus() (models.ReadyStatus, error) {
	params := service.NewReadyzParams()
	if resp, err := lb.Service.Readyz(params); err != nil {
		return nil, err
	} else {
		return resp.Payload, nil
	}
}

func (lb *Flexlb) ListInstances(name *string) ([]*models.Instance, error) {
	params := instance.NewListParams()
	params.Name = name
	if resp, err := lb.Instance.List(params); err != nil {
		return nil, err
	} else {
		return resp.Payload, nil
	}
}

func (lb *Flexlb) GetInstance(name string) (*models.Instance, error) {
	params := instance.NewGetParams()
	params.Name = name
	if resp, err := lb.Instance.Get(params); err != nil {
		return nil, err
	} else {
		return resp.Payload, nil
	}
}

func (lb *Flexlb) StartInstance(name string) (*models.Instance, error) {
	params := instance.NewStartParams()
	params.Name = name
	if resp, err := lb.Instance.Start(params); err != nil {
		return nil, err
	} else {
		return resp.Payload, nil
	}
}

func (lb *Flexlb) StopInstance(name string) (*models.Instance, error) {
	params := instance.NewStopParams()
	params.Name = name
	if resp, err := lb.Instance.Stop(params); err != nil {
		return nil, err
	} else {
		return resp.Payload, nil
	}
}

func (lb *Flexlb) DeleteInstance(name string) error {
	params := instance.NewDeleteParams()
	params.Name = name
	if _, err := lb.Instance.Delete(params); err != nil {
		return err
	} else {
		return nil
	}
}

func (lb *Flexlb) CreateInstance(cfg *models.InstanceConfig) (*models.Instance, error) {
	params := instance.NewCreateParams()
	params.Config = cfg
	if resp, err := lb.Instance.Create(params); err != nil {
		return nil, err
	} else {
		return resp.Payload, nil
	}
}

func (lb *Flexlb) ModifyInstance(cfg *models.InstanceConfig) (*models.Instance, error) {
	params := instance.NewModifyParams()
	params.Config = cfg
	if resp, err := lb.Instance.Modify(params); err != nil {
		return nil, err
	} else {
		return resp.Payload, nil
	}
}

func LoadConfig(file string) (*models.InstanceConfig, error) {
	if raw, err := ioutil.ReadFile(file); err != nil {
		return nil, err
	} else {
		var cfg models.InstanceConfig
		if err := json.Unmarshal(raw, &cfg); err != nil {
			return nil, err
		} else {
			return &cfg, nil
		}
	}
}