// Copyright 2025 The Casibase Authors. All Rights Reserved.
//
// 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 pkgmachine

import (
	"fmt"
	"net/url"

	"github.com/casibase/casibase/i18n"
	"github.com/digitalocean/go-libvirt"
)

type MachineKvmClient struct {
	L *libvirt.Libvirt
}

// The URI format is "driver[+transport]://[username @][hostname][:post]/[path][?Extraparameters]", for example "kvm+ssh://root @192.168.1.100/system ".
// Use TCP connection method here: “kvm+tcp://192.168.1.100/system”.
func newMachineKvmClient(username string, hostname string) (MachineKvmClient, error) {
	uri, err := url.Parse("kvm+tcp://" + hostname + "/system")
	if err != nil {
		return MachineKvmClient{}, err
	}
	l, err := libvirt.ConnectToURI(uri)
	if err != nil {
		return MachineKvmClient{}, err
	}

	return MachineKvmClient{L: l}, nil
}

func getMachineFromDom(l *libvirt.Libvirt, dom libvirt.Domain) (*Machine, error) {
	machine := &Machine{}

	name := dom.Name
	ID := dom.ID
	os, err := l.DomainGetOsType(dom)
	if err != nil {
		return nil, err
	}
	cpuSize, err := l.DomainGetMaxVcpus(dom)
	if err != nil {
		return nil, err
	}
	memSize, err := l.DomainGetMaxMemory(dom)
	if err != nil {
		return nil, err
	}

	machine.Name = name
	machine.Id = fmt.Sprintf("%d", ID)
	machine.Os = os
	machine.CpuSize = fmt.Sprintf("%d", cpuSize)
	machine.MemSize = fmt.Sprintf("%d", memSize)

	return machine, nil
}

func (client MachineKvmClient) GetMachines(lang string) ([]*Machine, error) {
	flags := libvirt.ConnectListDomainsActive | libvirt.ConnectListDomainsInactive
	doms, _, err := client.L.ConnectListAllDomains(1, flags)
	if err != nil {
		return nil, err
	}

	machines := []*Machine{}
	for _, dom := range doms {
		machine, err := getMachineFromDom(client.L, dom)
		if err != nil {
			return nil, err
		}
		machines = append(machines, machine)
	}

	return machines, nil
}

func (client MachineKvmClient) GetMachine(name string, lang string) (*Machine, error) {
	dom, err := client.L.DomainLookupByName(name)
	if err != nil {
		return nil, err
	}

	machine, err := getMachineFromDom(client.L, dom)
	if err != nil {
		return nil, err
	}

	return machine, nil
}

func (client MachineKvmClient) UpdateMachineState(name string, state string, lang string) (bool, string, error) {
	dom, err := client.L.DomainLookupByName(name)
	if err != nil {
		return false, "", err
	}

	switch state {
	case "Running":
		err = client.L.DomainCreate(dom)
	case "Stopped":
		err = client.L.DomainShutdown(dom)
	default:
		return false, fmt.Sprintf(i18n.Translate(lang, "pkgmachine:Unsupported state: %s"), state), nil
	}

	if err != nil {
		return false, "", err
	}

	return true, fmt.Sprintf(i18n.Translate(lang, "pkgmachine:Domain: [%s]'s state has been successfully updated to: [%s]"), name, state), nil
}