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

import (
	"context"
	"fmt"
	"net/http"
	"strings"

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

	"yunion.io/x/onecloud/pkg/apis"
	computeapi "yunion.io/x/onecloud/pkg/apis/compute"
	hostapi "yunion.io/x/onecloud/pkg/apis/host"
	schedapi "yunion.io/x/onecloud/pkg/apis/scheduler"
	"yunion.io/x/onecloud/pkg/appsrv"
	"yunion.io/x/onecloud/pkg/hostman/guestman"
	"yunion.io/x/onecloud/pkg/hostman/guestman/desc"
	"yunion.io/x/onecloud/pkg/hostman/hostutils"
	"yunion.io/x/onecloud/pkg/hostman/monitor"
	"yunion.io/x/onecloud/pkg/hostman/storageman"
	"yunion.io/x/onecloud/pkg/httperrors"
	"yunion.io/x/onecloud/pkg/mcclient"
	"yunion.io/x/onecloud/pkg/mcclient/auth"
)

type strDict map[string]string
type actionFunc func(context.Context, mcclient.TokenCredential, string, jsonutils.JSONObject) (interface{}, error)

var (
	keyWords = []string{"servers"}
)

func AddGuestTaskHandler(prefix string, app *appsrv.Application) {
	for _, keyWord := range keyWords {
		app.AddHandler("GET",
			fmt.Sprintf("%s/%s/<sid>/status", prefix, keyWord),
			auth.Authenticate(getStatus))

		app.AddHandler("POST",
			fmt.Sprintf("%s/%s/cpu-node-balance", prefix, keyWord),
			auth.Authenticate(cpusetBalance))

		app.AddHandler("POST",
			fmt.Sprintf("%s/%s/prepare-import-from-libvirt", prefix, keyWord),
			auth.Authenticate(guestPrepareImportFormLibvirt))

		app.AddHandler("POST",
			fmt.Sprintf("%s/%s/upload-status", prefix, keyWord),
			auth.Authenticate(uploadStatus))

		app.AddHandler("DELETE",
			fmt.Sprintf("%s/%s/<sid>", prefix, keyWord),
			auth.Authenticate(deleteGuest))

		for action, f := range map[string]actionFunc{
			"create":                   guestCreate,
			"deploy":                   guestDeploy,
			"rebuild":                  guestRebuild,
			"start":                    guestStart,
			"stop":                     guestStop,
			"monitor":                  guestMonitor,
			"sync":                     guestSync,
			"suspend":                  guestSuspend,
			"io-throttle":              guestIoThrottle,
			"snapshot":                 guestSnapshot,
			"delete-snapshot":          guestDeleteSnapshot,
			"reload-disk-snapshot":     guestReloadDiskSnapshot,
			"src-prepare-migrate":      guestSrcPrepareMigrate,
			"dest-prepare-migrate":     guestDestPrepareMigrate,
			"live-migrate":             guestLiveMigrate,
			"cancel-live-migrate":      guestCancelLiveMigrate,
			"resume":                   guestResume,
			"block-replication":        guestBlockReplication,
			"slave-block-stream-disks": slaveGuestBlockStreamDisks,
			"hotplug-cpu-mem":          guestHotplugCpuMem,
			"cancel-block-jobs":        guestCancelBlockJobs,
			"cancel-block-replication": guestCancelBlockReplication,
			"create-from-libvirt":      guestCreateFromLibvirt,
			"create-form-esxi":         guestCreateFromEsxi,
			"create-from-cloudpods":    guestCreateFromCloudpods,
			"open-forward":             guestOpenForward,
			"list-forward":             guestListForward,
			"close-forward":            guestCloseForward,
			"storage-clone-disk":       guestStorageCloneDisk,
			"live-change-disk":         guestLiveChangeDisk,
			"cpuset":                   guestCPUSet,
			"cpuset-remove":            guestCPUSetRemove,
			"memory-snapshot":          guestMemorySnapshot,
			"memory-snapshot-reset":    guestMemorySnapshotReset,
			"qga-set-password":         qgaGuestSetPassword,
			"qga-guest-ping":           qgaGuestPing,
			"qga-command":              qgaCommand,
			"reset-nic-traffic-limit":  guestResetNicTrafficLimit,
			"set-nic-traffic-limit":    guestSetNicTrafficLimit,
			"qga-guest-info-task":      qgaGuestInfoTask,
			"qga-get-network":          qgaGetNetwork,
			"qga-set-network":          qgaSetNetwork,
			"qga-get-os-info":          qgaGetOsInfo,
			"start-rescue":             guestStartRescue,
			"guest-screen-dump":        guestScreenDump,
			"upload-status":            guestUploadStatus,
		} {
			app.AddHandler("POST",
				fmt.Sprintf("%s/%s/<sid>/%s", prefix, keyWord, action),
				auth.Authenticate(guestActions(f)),
			)
		}

		app.AddHandler("DELETE",
			fmt.Sprintf("%s/%s/memory-snapshot", prefix, keyWord),
			auth.Authenticate(guestMemorySnapshotDelete))
	}
}

func guestActions(f actionFunc) appsrv.FilterHandler {
	return func(ctx context.Context, w http.ResponseWriter, r *http.Request) {
		params, _, body := appsrv.FetchEnv(ctx, w, r)
		userCred := auth.FetchUserCredential(ctx, nil)
		if body == nil {
			body = jsonutils.NewDict()
		}
		var sid = params["<sid>"]
		res, err := f(ctx, userCred, sid, body)
		if err != nil {
			hostutils.Response(ctx, w, err)
		} else if !gotypes.IsNil(res) {
			hostutils.Response(ctx, w, res)
		} else {
			hostutils.ResponseOk(ctx, w)
		}
	}
}

func getStatus(ctx context.Context, w http.ResponseWriter, r *http.Request) {
	params, _, _ := appsrv.FetchEnv(ctx, w, r)
	sid := params["<sid>"]
	hostutils.DelayTaskWithoutReqctx(ctx, guestman.GetGuestManager().GetGuestStatus, sid)
	hostutils.ResponseOk(ctx, w)
}

func cpusetBalance(ctx context.Context, w http.ResponseWriter, r *http.Request) {
	hostutils.DelayTask(ctx, guestman.GetGuestManager().CpusetBalance, nil)
	hostutils.ResponseOk(ctx, w)
}

func deleteGuest(ctx context.Context, w http.ResponseWriter, r *http.Request) {
	params, _, body := appsrv.FetchEnv(ctx, w, r)
	var sid = params["<sid>"]
	var migrated bool
	if body != nil {
		migrated = jsonutils.QueryBoolean(body, "migrated", false)
	}
	guest, err := guestman.GetGuestManager().Delete(sid)
	if err != nil {
		hostutils.Response(ctx, w, err)
	} else {
		hostutils.DelayTask(ctx, guest.CleanGuest, migrated)
		hostutils.Response(ctx, w, map[string]bool{"delay_clean": true})
	}
}

func guestCreate(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	if guestman.GetGuestManager().IsGuestExist(sid) {
		return nil, httperrors.NewBadRequestError("Guest %s is exist", sid)
	}
	hostutils.DelayTaskWithWorker(ctx,
		guestman.GetGuestManager().GuestCreate,
		&guestman.SGuestDeploy{
			UserCred: userCred,

			Sid:    sid,
			Body:   body,
			IsInit: true,
		},
		guestman.NbdWorker,
	)
	return nil, nil
}

func guestDeploy(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	err := guestman.GetGuestManager().PrepareDeploy(sid)
	if err != nil {
		return nil, err
	}
	hostutils.DelayTaskWithWorker(ctx,
		guestman.GetGuestManager().GuestDeploy,
		&guestman.SGuestDeploy{
			UserCred: userCred,

			Sid:    sid,
			Body:   body,
			IsInit: false,
		},
		guestman.NbdWorker,
	)
	return nil, nil
}

func guestRebuild(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	err := guestman.GetGuestManager().PrepareDeploy(sid)
	if err != nil {
		return nil, err
	}
	hostutils.DelayTaskWithWorker(ctx,
		guestman.GetGuestManager().GuestDeploy,
		&guestman.SGuestDeploy{
			UserCred: userCred,

			Sid:    sid,
			Body:   body,
			IsInit: true,
		},
		guestman.NbdWorker,
	)
	return nil, nil
}

func guestStart(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	return guestman.GetGuestManager().GuestStart(ctx, userCred, sid, body)
}

func guestStop(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	timeout, err := body.Int("timeout")
	if err != nil {
		timeout = 30
	}
	return nil, guestman.GetGuestManager().GuestStop(ctx, sid, timeout)
}

func guestMonitor(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	if !guestman.GetGuestManager().IsGuestExist(sid) {
		return nil, httperrors.NewNotFoundError("Guest %s not found", sid)
	}

	if body.Contains("cmd") {
		var c = make(chan string)
		cb := func(res string) {
			c <- res
		}
		cmd, _ := body.GetString("cmd")
		qmp := jsonutils.QueryBoolean(body, "qmp", false)
		err := guestman.GetGuestManager().Monitor(sid, cmd, qmp, cb)
		if err != nil {
			return nil, err
		} else {
			var res = <-c
			lines := strings.Split(res, "\\r\\n")

			return strDict{"results": strings.Join(lines, "\n")}, nil
		}
	} else {
		return nil, httperrors.NewMissingParameterError("cmd")
	}
}

func guestSync(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	if !guestman.GetGuestManager().IsGuestExist(sid) {
		return nil, httperrors.NewNotFoundError("Guest %s not found", sid)
	}
	hostutils.DelayTask(ctx, guestman.GetGuestManager().GuestSync, &guestman.SBaseParams{
		Sid:  sid,
		Body: body,
	})
	return nil, nil
}

func guestSuspend(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	if !guestman.GetGuestManager().IsGuestExist(sid) {
		return nil, httperrors.NewNotFoundError("Guest %s not found", sid)
	}
	hostutils.DelayTaskWithoutReqctx(ctx, guestman.GetGuestManager().GuestSuspend, sid)
	return nil, nil
}

func guestIoThrottle(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	guest, ok := guestman.GetGuestManager().GetServer(sid)
	if !ok {
		return nil, httperrors.NewNotFoundError("Guest %s not found", sid)
	}
	if !guest.IsRunning() {
		return nil, httperrors.NewInvalidStatusError("Not running")
	}

	input := new(computeapi.ServerSetDiskIoThrottleInput)
	if err := body.Unmarshal(input); err != nil {
		return nil, httperrors.NewInputParameterError("unmarshal params failed %s", err)
	}

	hostutils.DelayTaskWithoutReqctx(ctx, guestman.GetGuestManager().GuestIoThrottle, &guestman.SGuestIoThrottle{
		Sid:   sid,
		Input: input,
	})
	return nil, nil
}

func guestSrcPrepareMigrate(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	if !guestman.GetGuestManager().IsGuestExist(sid) {
		return nil, httperrors.NewNotFoundError("Guest %s not found", sid)
	}
	liveMigrate := jsonutils.QueryBoolean(body, "live_migrate", false)
	liveMigrateEnableTls := jsonutils.QueryBoolean(body, "enable_tls", false)
	hostutils.DelayTask(ctx, guestman.GetGuestManager().SrcPrepareMigrate,
		&guestman.SSrcPrepareMigrate{
			Sid:               sid,
			LiveMigrate:       liveMigrate,
			LiveMigrateUseTLS: liveMigrateEnableTls,
		})
	return nil, nil
}

func guestDestPrepareMigrate(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	err := guestDestPrepareMigrateInternal(ctx, userCred, sid, body)
	if err != nil {
		guestman.GetGuestManager().CleanServer(sid)
		return nil, errors.Wrapf(err, "guestDestPrepareMigrateInternal %s", sid)
	}
	return nil, nil
}

func guestDestPrepareMigrateInternal(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) error {
	if !guestman.GetGuestManager().CanMigrate(sid) {
		return httperrors.NewBadRequestError("Guest exist")
	}
	var guestDesc = new(desc.SGuestDesc)
	err := body.Unmarshal(guestDesc, "desc")
	if err != nil {
		return httperrors.NewBadRequestError("Failed unmarshal guest desc %s", err)
	}

	var params = &guestman.SDestPrepareMigrate{}
	qemuVersion, err := body.GetString("qemu_version")
	if err != nil {
		return httperrors.NewMissingParameterError("qemu_version")
	}
	liveMigrate := jsonutils.QueryBoolean(body, "live_migrate", false)
	if liveMigrate {
		var sourceDesc = new(desc.SGuestDesc)
		err := body.Unmarshal(sourceDesc, "src_desc")
		if err != nil {
			return httperrors.NewBadRequestError("Failed unmarshal guest source desc %s", err)
		}
		params.SrcDesc = sourceDesc
	}

	isLocal, err := body.Bool("is_local_storage")
	if err != nil {
		return httperrors.NewMissingParameterError("is_local_storage")
	}

	params.Sid = sid
	params.Desc = guestDesc
	params.QemuVersion = qemuVersion
	params.LiveMigrate = liveMigrate
	params.EnableTLS = jsonutils.QueryBoolean(body, "enable_tls", false)
	if params.EnableTLS {
		certsObj, err := body.Get("migrate_certs")
		if err != nil {
			return httperrors.NewMissingParameterError("migrate_certs")
		}
		certs := map[string]string{}
		if err := certsObj.Unmarshal(&certs); err != nil {
			return httperrors.NewInputParameterError("unmarshal migrate_certs to map: %s", err)
		}
		params.MigrateCerts = certs
	}
	if isLocal {
		serverUrl, err := body.GetString("server_url")
		if err != nil {
			return httperrors.NewMissingParameterError("server_url")
		} else {
			params.ServerUrl = serverUrl
		}
		snapshotsUri, err := body.GetString("snapshots_uri")
		if err != nil {
			return httperrors.NewMissingParameterError("snapshots_uri")
		} else {
			params.SnapshotsUri = snapshotsUri
		}
		disksUri, err := body.GetString("disks_uri")
		if err != nil {
			return httperrors.NewMissingParameterError("disks_uri")
		} else {
			params.DisksUri = disksUri
		}
		diskSnapsChain, err := body.Get("disk_snaps_chain")
		if err != nil {
			params.DiskSnapsChain = jsonutils.NewDict()
		} else {
			params.DiskSnapsChain = diskSnapsChain
		}
		outChainSnaps, err := body.Get("out_chain_snapshots")
		if err != nil {
			params.OutChainSnaps = jsonutils.NewDict()
		} else {
			params.OutChainSnaps = outChainSnaps
		}
		params.SysDiskHasTemplate = jsonutils.QueryBoolean(body, "sys_disk_has_template", false)

		disksBack, err := body.Get("disks_back")
		if err != nil {
			params.DisksBackingFile = jsonutils.NewDict()
		} else {
			params.DisksBackingFile = disksBack
		}
		disks := guestDesc.Disks
		if disks == nil {
			return httperrors.NewInputParameterError("Get desc disks error")
		} else {
			targetStorageIds := []string{}
			for i := 0; i < len(disks); i++ {
				targetStorageId := disks[i].TargetStorageId
				if len(targetStorageId) == 0 {
					return httperrors.NewMissingParameterError("target_storage_id")
				}
				targetStorageIds = append(targetStorageIds, targetStorageId)
				params.TargetStorageIds = targetStorageIds
			}

		}
		params.RebaseDisks = jsonutils.QueryBoolean(body, "rebase_disks", false)
	}

	msUri, err := body.GetString("memory_snapshots_uri")
	if err != nil {
		return httperrors.NewMissingParameterError("memory_snapshots_uri")
	}
	params.MemorySnapshotsUri = msUri
	msIds, _ := jsonutils.GetStringArray(body, "src_memory_snapshots")
	params.SrcMemorySnapshots = msIds

	params.UserCred = userCred

	hostutils.DelayTask(ctx, guestman.GetGuestManager().DestPrepareMigrate, params)
	return nil
}

func guestLiveMigrate(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	if !guestman.GetGuestManager().IsGuestExist(sid) {
		return nil, httperrors.NewNotFoundError("Guest %s not found", sid)
	}
	destPort, err := body.Int("live_migrate_dest_port")
	if err != nil {
		return nil, httperrors.NewMissingParameterError("live_migrate_dest_port")
	}
	var nbdServerPort int64 = -1
	if body.Contains("nbd_server_port") {
		nbdServerPort, err = body.Int("nbd_server_port")
		if err != nil {
			return nil, httperrors.NewMissingParameterError("live_migrate_dest_port")
		}
	}

	destIp, err := body.GetString("dest_ip")
	if err != nil {
		return nil, httperrors.NewMissingParameterError("dest_ip")
	}
	isLocal, err := body.Bool("is_local_storage")
	if err != nil {
		return nil, httperrors.NewMissingParameterError("is_local_storage")
	}
	enableTLS := jsonutils.QueryBoolean(body, "enable_tls", false)
	quicklyFinish := jsonutils.QueryBoolean(body, "quickly_finish", false)
	params := &guestman.SLiveMigrate{
		Sid:           sid,
		DestPort:      int(destPort),
		NbdServerPort: int(nbdServerPort),
		DestIp:        destIp,
		IsLocal:       isLocal,
		EnableTLS:     enableTLS,
		QuicklyFinish: quicklyFinish,
	}
	if body.Contains("max_bandwidth_mb") {
		maxBandwidthMb, _ := body.Int("max_bandwidth_mb")
		params.MaxBandwidthMB = &maxBandwidthMb
	}

	hostutils.DelayTaskWithoutReqctx(ctx, guestman.GetGuestManager().LiveMigrate, params)
	return nil, nil
}

func guestCancelLiveMigrate(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	guest, ok := guestman.GetGuestManager().GetKVMServer(sid)
	if !ok {
		return nil, httperrors.NewNotFoundError("Guest %s not found", sid)
	}
	if guest.MigrateTask == nil {
		return nil, httperrors.NewBadRequestError("Guest %s not in migrating", sid)
	}
	guest.MigrateTask.SetLiveMigrateCancelled()
	var c = make(chan string)
	cb := func(res string) {
		c <- res
	}
	guest.Monitor.MigrateCancel(cb)
	var res = <-c
	lines := strings.Split(res, "\\r\\n")
	return strDict{"results": strings.Join(lines, "\n")}, nil
}

func guestResume(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	if !guestman.GetGuestManager().IsGuestExist(sid) {
		return nil, httperrors.NewNotFoundError("Guest %s not found", sid)
	}
	isLiveMigrate := jsonutils.QueryBoolean(body, "live_migrate", false)
	cleanTLS := jsonutils.QueryBoolean(body, "clean_tls", false)
	guestman.GetGuestManager().Resume(ctx, sid, isLiveMigrate, cleanTLS)
	return nil, nil
}

// func guestStartNbdServer(ctx context.Context, sid string, body jsonutils.JSONObject) (interface{}, error) {
// 	if !guestManger.IsGuestExist(sid) {
// 		return nil, httperrors.NewNotFoundError("Guest %s not found", sid)
// 	}
// 	hostutils.DelayTask(ctx, guestManger.StartNbdServer, sid)
// 	return nil, nil
// }

func guestBlockReplication(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	if !guestman.GetGuestManager().IsGuestExist(sid) {
		return nil, httperrors.NewNotFoundError("Guest %s not found", sid)
	}
	backupNbdServerUri, err := body.GetString("backup_nbd_server_uri")
	if err != nil {
		return nil, httperrors.NewMissingParameterError("backup_nbd_server_uri")
	}
	var guestDesc = new(desc.SGuestDesc)
	err = body.Unmarshal(guestDesc, "desc")
	if err != nil {
		return nil, httperrors.NewInputParameterError("failed unmarshal desc %s", err)
	}
	hostutils.DelayTaskWithoutReqctx(ctx, guestman.GetGuestManager().StartBlockReplication,
		&guestman.SDriverMirror{
			Sid:          sid,
			NbdServerUri: backupNbdServerUri,
			Desc:         guestDesc,
		})
	return nil, nil
}

func slaveGuestBlockStreamDisks(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	guest, ok := guestman.GetGuestManager().GetKVMServer(sid)
	if !ok {
		return nil, httperrors.NewNotFoundError("Guest %s not found", sid)
	}
	return nil, guest.SlaveDisksBlockStream()
}

func guestCancelBlockJobs(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	if !guestman.GetGuestManager().IsGuestExist(sid) {
		return nil, httperrors.NewNotFoundError("Guest %s not found", sid)
	}
	hostutils.DelayTaskWithoutReqctx(ctx, guestman.GetGuestManager().CancelBlockJobs, sid)
	return nil, nil
}

func guestCancelBlockReplication(
	ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject,
) (interface{}, error) {
	if !guestman.GetGuestManager().IsGuestExist(sid) {
		return nil, httperrors.NewNotFoundError("Guest %s not found", sid)
	}
	hostutils.DelayTaskWithoutReqctx(ctx, guestman.GetGuestManager().CancelBlockReplication, sid)
	return nil, nil
}

func guestHotplugCpuMem(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	if !guestman.GetGuestManager().IsGuestExist(sid) {
		return nil, httperrors.NewNotFoundError("Guest %s not found", sid)
	}

	if guestman.GetGuestManager().Status(sid) != "running" {
		return nil, httperrors.NewBadRequestError("Guest %s not running", sid)
	}

	addCpuCount, _ := body.Int("add_cpu")
	addMemSize, _ := body.Int("add_mem")

	input := &guestman.SGuestHotplugCpuMem{
		Sid:         sid,
		AddCpuCount: addCpuCount,
		AddMemSize:  addMemSize,
	}
	totalMemSize, err := body.Int("total_mem")
	if err == nil {
		input.TotalMemSize = &totalMemSize
	}
	totalCpuCount, err := body.Int("total_cpu")
	if err == nil {
		input.TotalCpuCount = &totalCpuCount
	}

	if body.Contains("cpu_numa_pin") {
		cpuNumaPin := make([]schedapi.SCpuNumaPin, 0)
		if err := body.Unmarshal(&cpuNumaPin, "cpu_numa_pin"); err != nil {
			return nil, httperrors.NewInputParameterError("failed parse cpu_numa_pin %s", err)
		}

		if len(cpuNumaPin) > 0 {
			descCpuNumaPin := make([]*desc.SCpuNumaPin, len(cpuNumaPin))
			for i := range cpuNumaPin {
				if cpuNumaPin[i].MemSizeMB != nil {
					descCpuNumaPin[i].SizeMB = int64(*cpuNumaPin[i].MemSizeMB)
					nodeId := uint16(cpuNumaPin[i].NodeId)
					descCpuNumaPin[i].NodeId = &nodeId
				}
				if len(cpuNumaPin[i].CpuPin) > 0 {
					vcpuPin := make([]desc.SVCpuPin, len(cpuNumaPin[i].CpuPin))
					for j := range vcpuPin {
						vcpuPin[j].Pcpu = cpuNumaPin[i].CpuPin[j]
					}
					descCpuNumaPin[i].VcpuPin = vcpuPin
				}
			}
			input.CpuNumaPin = descCpuNumaPin
		}
	}
	hostutils.DelayTaskWithoutReqctx(ctx, guestman.GetGuestManager().HotplugCpuMem, input)
	return nil, nil
}

func guestReloadDiskSnapshot(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	diskId, err := body.GetString("disk_id")
	if err != nil {
		return nil, httperrors.NewMissingParameterError("disk_id")
	}
	guest, ok := guestman.GetGuestManager().GetKVMServer(sid)
	if !ok {
		return nil, httperrors.NewNotFoundError("guest %s not found", sid)
	}

	var disk storageman.IDisk
	disks := guest.Desc.Disks
	for _, d := range disks {
		if diskId == d.DiskId {
			disk, _ = storageman.GetManager().GetDiskByPath(d.Path)
			break
		}
	}
	if disk == nil {
		return nil, httperrors.NewNotFoundError("Disk not found")
	}

	hostutils.DelayTask(ctx, guestman.GetGuestManager().ReloadDiskSnapshot, &guestman.SReloadDisk{
		Sid:  sid,
		Disk: disk,
	})
	return nil, nil
}

func guestSnapshot(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	snapshotId, err := body.GetString("snapshot_id")
	if err != nil {
		return nil, httperrors.NewMissingParameterError("snapshot_id")
	}
	diskId, err := body.GetString("disk_id")
	if err != nil {
		return nil, httperrors.NewMissingParameterError("disk_id")
	}
	guest, ok := guestman.GetGuestManager().GetServer(sid)
	if !ok {
		return nil, httperrors.NewNotFoundError("guest %s not found", sid)
	}

	var disk storageman.IDisk
	disks := guest.GetDesc().Disks
	for _, d := range disks {
		if diskId == d.DiskId {
			disk, err = storageman.GetManager().GetDiskById(diskId)
			if err != nil {
				return nil, errors.Wrapf(err, "GetDiskById(%s)", diskId)
			}
			break
		}
	}
	if disk == nil {
		return nil, httperrors.NewNotFoundError("Disk not found")
	}
	input := &guestman.SDiskSnapshot{
		UserCred:   userCred,
		Sid:        sid,
		SnapshotId: snapshotId,
		Disk:       disk,
	}

	if body.Contains("backup_disk_config") {
		backupDiskConfig := new(guestman.SBackupDiskConfig)
		if err := body.Unmarshal(backupDiskConfig, "backup_disk_config"); err != nil {
			return nil, httperrors.NewInputParameterError("unmarshal backup_disk_config")
		}
		input.BackupDiskConfig = backupDiskConfig
	}

	hostutils.DelayBackupTask(ctx, guestman.GetGuestManager().DoSnapshot, input)
	return nil, nil
}

func guestDeleteSnapshot(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	deleteSnapshot, err := body.GetString("delete_snapshot")
	if err != nil {
		return nil, httperrors.NewMissingParameterError("delete_snapshot")
	}
	diskId, err := body.GetString("disk_id")
	if err != nil {
		return nil, httperrors.NewMissingParameterError("disk_id")
	}
	var totalCnt, deletedCnt int64 = 1, 0
	if body.Contains("snapshot_total_count") {
		totalCnt, _ = body.Int("snapshot_total_count")
		deletedCnt, _ = body.Int("deleted_snapshot_count")
	}

	guest, ok := guestman.GetGuestManager().GetServer(sid)
	if !ok {
		return nil, httperrors.NewNotFoundError("guest %s not found", sid)
	}

	var disk storageman.IDisk
	disks := guest.GetDesc().Disks
	for _, d := range disks {
		if diskId == d.DiskId {
			disk, err = storageman.GetManager().GetDiskById(diskId)
			if err != nil {
				return nil, errors.Wrapf(err, "GetDiskById(%s)", diskId)
			}
			break
		}
	}
	if disk == nil {
		return nil, httperrors.NewNotFoundError("Disk not found")
	}

	params := &guestman.SDeleteDiskSnapshot{
		Sid:                      sid,
		DeleteSnapshot:           deleteSnapshot,
		Disk:                     disk,
		TotalDeleteSnapshotCount: int(totalCnt),
		DeletedSnapshotCount:     int(deletedCnt),
	}

	if body.Contains("encrypt_info") {
		encryptInfo := apis.SEncryptInfo{}
		if err = body.Unmarshal(&encryptInfo, "encrypt_info"); err != nil {
			return nil, httperrors.NewInputParameterError("unmarshal encrypt_info failed %s", err)
		}
		params.EncryptInfo = encryptInfo
	}

	// blockStream indicate snapshot<-disk
	blockStream := jsonutils.QueryBoolean(body, "block_stream", false)
	autoDeleted := jsonutils.QueryBoolean(body, "auto_deleted", false)

	if !blockStream && !autoDeleted {
		convertSnapshot, err := body.GetString("convert_snapshot")
		if err != nil {
			return nil, httperrors.NewMissingParameterError("convert_snapshot")
		}
		params.ConvertSnapshot = convertSnapshot
	}
	params.BlockStream = blockStream
	hostutils.DelayTask(ctx, guestman.GetGuestManager().DeleteSnapshot, params)
	return nil, nil
}

func formatCloneDiskParams(sid string, body jsonutils.JSONObject) (*guestman.SStorageCloneDisk, error) {
	input := new(computeapi.ServerChangeDiskStorageInternalInput)
	if err := body.Unmarshal(input); err != nil {
		return nil, err
	}
	srcStorage := storageman.GetManager().GetStorage(input.StorageId)
	if srcStorage == nil {
		return nil, httperrors.NewNotFoundError("Source storage %q not found", input.StorageId)
	}
	srcDisk, err := srcStorage.GetDiskById(input.DiskId)
	if err != nil {
		return nil, errors.Wrapf(err, "Get source disk %q on storage %q", input.DiskId, srcStorage.GetId())
	}
	targetStorage := storageman.GetManager().GetStorage(input.TargetStorageId)
	if targetStorage == nil {
		return nil, httperrors.NewNotFoundError("Target storage %s not found", input.TargetStorageId)
	}
	if input.TargetDiskId == "" {
		return nil, httperrors.NewMissingParameterError("Target disk id is empty")
	}

	params := &guestman.SStorageCloneDisk{
		ServerId:           sid,
		SourceStorage:      srcStorage,
		SourceDisk:         srcDisk,
		TargetStorage:      targetStorage,
		TargetDiskId:       input.TargetDiskId,
		DiskFormat:         input.DiskFormat,
		TargetDiskDesc:     input.TargetDiskDesc,
		CompletedDiskCount: input.CompletedDiskCount,
		CloneDiskCount:     input.CloneDiskCount,
	}
	return params, nil
}

func guestStorageCloneDisk(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	params, err := formatCloneDiskParams(sid, body)
	if err != nil {
		return nil, err
	}

	hostutils.DelayTaskWithoutReqctx(ctx, guestman.GetGuestManager().StorageCloneDisk, params)
	return nil, nil
}

func guestLiveChangeDisk(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	params, err := formatCloneDiskParams(sid, body)
	if err != nil {
		return nil, err
	}

	hostutils.DelayTaskWithoutReqctx(ctx, guestman.GetGuestManager().LiveChangeDisk, params)
	return nil, nil
}

func guestCPUSet(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	input := new(computeapi.ServerCPUSetInput)
	if err := body.Unmarshal(input); err != nil {
		return nil, err
	}
	gm := guestman.GetGuestManager()
	return gm.CPUSet(ctx, sid, input)
}

func guestCPUSetRemove(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	gm := guestman.GetGuestManager()
	if err := gm.CPUSetRemove(ctx, sid); err != nil {
		return nil, err
	}
	return nil, nil
}

func guestMemorySnapshot(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	input := new(hostapi.GuestMemorySnapshotRequest)
	if err := body.Unmarshal(input); err != nil {
		return nil, err
	}
	gm := guestman.GetGuestManager()
	hostutils.DelayTaskWithoutReqctx(ctx, gm.DoMemorySnapshot, &guestman.SMemorySnapshot{
		GuestMemorySnapshotRequest: input,
		Sid:                        sid,
	})
	return nil, nil
}

func guestMemorySnapshotReset(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	input := new(hostapi.GuestMemorySnapshotResetRequest)
	if err := body.Unmarshal(input); err != nil {
		return nil, err
	}
	if input.InstanceSnapshotId == "" {
		return nil, httperrors.NewMissingParameterError("instance_snapshot_id")
	}
	if input.Path == "" {
		return nil, httperrors.NewMissingParameterError("path")
	}
	gm := guestman.GetGuestManager()
	hostutils.DelayTaskWithoutReqctx(ctx, gm.DoResetMemorySnapshot, &guestman.SMemorySnapshotReset{
		GuestMemorySnapshotResetRequest: input,
		Sid:                             sid,
	})
	return nil, nil
}

func guestMemorySnapshotDelete(ctx context.Context, w http.ResponseWriter, r *http.Request) {
	_, _, body := appsrv.FetchEnv(ctx, w, r)
	input := new(hostapi.GuestMemorySnapshotDeleteRequest)
	if err := body.Unmarshal(input); err != nil {
		hostutils.Response(ctx, w, err)
		return
	}
	if input.InstanceSnapshotId == "" {
		hostutils.Response(ctx, w, httperrors.NewMissingParameterError("instance_snapshot_id"))
		return
	}
	if input.Path == "" {
		hostutils.Response(ctx, w, httperrors.NewMissingParameterError("path"))
		return
	}
	gm := guestman.GetGuestManager()
	hostutils.DelayTask(ctx, gm.DoDeleteMemorySnapshot, &guestman.SMemorySnapshotDelete{
		GuestMemorySnapshotDeleteRequest: input,
	})
}

func guestResetNicTrafficLimit(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	input := []computeapi.ServerNicTrafficLimit{}
	if err := body.Unmarshal(&input); err != nil {
		return nil, httperrors.NewInputParameterError("failed unmarshal input %s", err)
	}

	hostutils.DelayTask(ctx, func(ctx context.Context, params interface{}) (jsonutils.JSONObject, error) {
		return nil, guestman.GetGuestManager().ResetGuestNicTrafficLimit(sid, input)
	}, nil)

	return nil, nil
}

func guestSetNicTrafficLimit(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	input := []computeapi.ServerNicTrafficLimit{}
	if err := body.Unmarshal(&input); err != nil {
		return nil, httperrors.NewInputParameterError("failed unmarshal input %s", err)
	}

	hostutils.DelayTask(ctx, func(ctx context.Context, params interface{}) (jsonutils.JSONObject, error) {
		return nil, guestman.GetGuestManager().SetGuestNicTrafficLimit(sid, input)
	}, nil)
	return nil, nil
}

func qgaGuestSetPassword(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	input := new(hostapi.GuestSetPasswordRequest)
	if err := body.Unmarshal(input); err != nil {
		return nil, err
	}
	if input.Username == "" {
		return nil, httperrors.NewMissingParameterError("username")
	}
	if input.Password == "" {
		return nil, httperrors.NewMissingParameterError("password")
	}
	gm := guestman.GetGuestManager()
	hostutils.DelayTask(ctx, gm.QgaGuestSetPassword, &guestman.SQgaGuestSetPassword{
		GuestSetPasswordRequest: input,
		Sid:                     sid,
	})
	return nil, nil
}

func qgaGuestPing(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	gm := guestman.GetGuestManager()
	params := &guestman.SBaseParams{Sid: sid, Body: body}
	if jsonutils.QueryBoolean(body, "async", true) {
		hostutils.DelayTask(ctx, gm.QgaGuestPing, params)
		return nil, nil
	} else {
		return gm.QgaGuestPing(ctx, params)
	}
}

func qgaCommand(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	gm := guestman.GetGuestManager()
	input := computeapi.ServerQgaCommandInput{}
	err := body.Unmarshal(&input)
	if err != nil {
		return nil, httperrors.NewInputParameterError("unmarshal input to ServerQgaCommandInput: %s", err.Error())
	}
	cmdJson, err := jsonutils.ParseString(input.Command)
	if err != nil {
		return nil, httperrors.NewInputParameterError("failed parse qga command")
	}
	qgaCmd := &monitor.Command{}
	qgaCmd.Execute, err = cmdJson.GetString("execute")
	if err != nil {
		return nil, httperrors.NewInputParameterError("failed get qga command")
	}
	if cmdJson.Contains("arguments") {
		qgaCmd.Args, _ = cmdJson.Get("arguments")
	}
	return gm.QgaCommand(qgaCmd, sid, input.Timeout)
}

func qgaGuestInfoTask(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	gm := guestman.GetGuestManager()
	return gm.QgaGuestInfoTask(sid)
}

func qgaGetNetwork(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	gm := guestman.GetGuestManager()
	return gm.QgaGetNetwork(sid)
}

func qgaSetNetwork(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	input := computeapi.ServerQgaSetNetworkInput{}
	err := body.Unmarshal(&input)
	if err != nil {
		return nil, httperrors.NewInputParameterError("unmarshal input to ServerQgaSetNetworkInput: %s", err.Error())
	}
	if input.Device == "" {
		return nil, httperrors.NewMissingParameterError("device")
	}
	if input.Ipmask == "" && input.Ip6mask == "" {
		return nil, httperrors.NewMissingParameterError("ipmask")
	}
	if input.Gateway == "" && input.Gateway6 == "" {
		return nil, httperrors.NewMissingParameterError("gateway")
	}

	hostutils.DelayTask(ctx, guestman.GetGuestManager().QgaSetNetwork, &guestman.SQgaGuestSetNetwork{
		Sid:      sid,
		Device:   input.Device,
		Ipmask:   input.Ipmask,
		Gateway:  input.Gateway,
		Ip6mask:  input.Ip6mask,
		Gateway6: input.Gateway6,
	})

	return nil, nil
}

func qgaGetOsInfo(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	gm := guestman.GetGuestManager()
	return gm.QgaGetOsInfo(sid)
}

func guestScreenDump(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	gm := guestman.GetGuestManager()
	return gm.RequestGuestScreenDump(sid)
}

// prepare rescue files
func guestStartRescue(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	return guestman.GetGuestManager().GuestStartRescue(ctx, userCred, sid, body)
}

func uploadStatus(ctx context.Context, w http.ResponseWriter, r *http.Request) {
	_, _, body := appsrv.FetchEnv(ctx, w, r)
	input := new(computeapi.HostUploadGuestsStatusRequest)
	if err := body.Unmarshal(input); err != nil {
		hostutils.Response(ctx, w, httperrors.NewInputParameterError("Unmarshal body to input: %v", err))
		return
	}
	hostutils.DelayTask(ctx, guestman.GetGuestManager().UploadGuestsStatus, input)
	hostutils.ResponseOk(ctx, w)
}

func guestUploadStatus(ctx context.Context, userCred mcclient.TokenCredential, sid string, body jsonutils.JSONObject) (interface{}, error) {
	hostutils.DelayTask(ctx, func(ctx context.Context, params interface{}) (jsonutils.JSONObject, error) {
		return guestman.GetGuestManager().UploadGuestStatus(ctx, sid)
	}, nil)

	return nil, nil
}