// Copyright 2022 Linka Cloud  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 d2vm

type ConvertOption func(o *convertOptions)

type convertOptions struct {
	size           uint64
	password       string
	output         string
	cmdLineExtra   string
	networkManager NetworkManager
	bootLoader     string
	raw            bool

	splitBoot bool
	bootSize  uint64
	bootFS    BootFS

	luksPassword string

	keepCache bool
	platform  string
	pull      bool

	hostname  string
	dns       []string
	dnsSearch []string
}

func (o *convertOptions) hasGrubBIOS() bool {
	return o.bootLoader == "grub" || o.bootLoader == "grub-bios"
}

func (o *convertOptions) hasGrubEFI() bool {
	return o.bootLoader == "grub" || o.bootLoader == "grub-efi"
}

func WithSize(size uint64) ConvertOption {
	return func(o *convertOptions) {
		o.size = size
	}
}

func WithPassword(password string) ConvertOption {
	return func(o *convertOptions) {
		o.password = password
	}
}

func WithOutput(output string) ConvertOption {
	return func(o *convertOptions) {
		o.output = output
	}
}

func WithCmdLineExtra(cmdLineExtra string) ConvertOption {
	return func(o *convertOptions) {
		o.cmdLineExtra = cmdLineExtra
	}
}

func WithNetworkManager(networkManager NetworkManager) ConvertOption {
	return func(o *convertOptions) {
		o.networkManager = networkManager
	}
}

func WithBootLoader(bootLoader string) ConvertOption {
	return func(o *convertOptions) {
		o.bootLoader = bootLoader
	}
}

func WithRaw(raw bool) ConvertOption {
	return func(o *convertOptions) {
		o.raw = raw
	}
}

func WithSplitBoot(b bool) ConvertOption {
	return func(o *convertOptions) {
		o.splitBoot = b
	}
}

func WithBootSize(bootSize uint64) ConvertOption {
	return func(o *convertOptions) {
		o.bootSize = bootSize
	}
}

func WithBootFS(bootFS BootFS) ConvertOption {
	return func(o *convertOptions) {
		o.bootFS = bootFS
	}
}

func WithLuksPassword(password string) ConvertOption {
	return func(o *convertOptions) {
		o.luksPassword = password
	}
}

func WithKeepCache(b bool) ConvertOption {
	return func(o *convertOptions) {
		o.keepCache = b
	}
}

func WithPlatform(platform string) ConvertOption {
	return func(o *convertOptions) {
		o.platform = platform
	}
}

func WithPull(b bool) ConvertOption {
	return func(o *convertOptions) {
		o.pull = b
	}
}

func WithHostname(hostname string) ConvertOption {
	return func(o *convertOptions) {
		o.hostname = hostname
	}
}

func WithDNS(dns []string) ConvertOption {
	return func(o *convertOptions) {
		o.dns = dns
	}
}

func WithDNSSearch(dnsSearch []string) ConvertOption {
	return func(o *convertOptions) {
		o.dnsSearch = dnsSearch
	}
}