107 lines
2.5 KiB
Go
107 lines
2.5 KiB
Go
package image
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"reflect"
|
|
"strings"
|
|
|
|
"github.com/abdfnx/gosh"
|
|
"github.com/rayaman/go-qemu/pkg/types"
|
|
)
|
|
|
|
func getOptions(m map[string]string) string {
|
|
str := []string{}
|
|
for k, v := range m {
|
|
if len(v) > 0 {
|
|
str = append(str, fmt.Sprintf("%v=%v", k, v))
|
|
}
|
|
}
|
|
n_str := strings.Join(str, ",")
|
|
if len(n_str) > 0 {
|
|
return "-o \"" + strings.Join(str, ",") + "\" "
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func getData(m map[string]string, key string) (string, bool) {
|
|
if d, ok := m[key]; ok {
|
|
delete(m, key)
|
|
if len(d) == 0 {
|
|
return "", false
|
|
}
|
|
return d, true
|
|
}
|
|
return "", false
|
|
}
|
|
|
|
func getMap(q any) map[string]string {
|
|
m := map[string]string{}
|
|
v := reflect.ValueOf(q).Elem()
|
|
for j := 0; j < v.NumField(); j++ { // Go through all fields of struct
|
|
if !v.Field(j).IsZero() {
|
|
index := strings.ReplaceAll(v.Type().Field(j).Tag.Get("json"), ",omitempty", "")
|
|
if v.Field(j).Type() == reflect.TypeOf(true) {
|
|
m[index] = types.SW[v.Field(j).Bool()]
|
|
} else {
|
|
m[index] = v.Field(j).String()
|
|
}
|
|
}
|
|
}
|
|
return m
|
|
}
|
|
|
|
type Options struct {
|
|
IsBaseImage bool `json:"is_base_image"`
|
|
}
|
|
|
|
// create [--object OBJECTDEF] [-q] [-f FMT] [-b BACKING_FILE [-F BACKING_FMT]] [-u] [-o OPTIONS] FILENAME [SIZE]
|
|
|
|
// Creates an image based of the supplied image
|
|
func Create(i Image, size types.Size, opts ...Options) error {
|
|
|
|
data := getMap(i)
|
|
|
|
var imagename string
|
|
var ok bool
|
|
|
|
if imagename, ok = getData(data, "image_name"); !(ok && imagename != "") {
|
|
return fmt.Errorf("invalid image data, ImageName is not optional")
|
|
}
|
|
|
|
os.MkdirAll(filepath.Join(filepath.Dir(imagename), "/base"), os.ModePerm)
|
|
basepath := filepath.Dir(imagename)
|
|
basename := filepath.Base(imagename)
|
|
|
|
if len(opts) > 0 && opts[0].IsBaseImage {
|
|
imagename = filepath.Join(basepath, "base", basename)
|
|
}
|
|
|
|
additional := []string{}
|
|
if backing_file, ok := getData(data, "backing_file"); ok {
|
|
additional = append(additional, "-b "+backing_file+" ")
|
|
}
|
|
|
|
if backing_fmt, ok := getData(data, "backing_fmt"); ok {
|
|
additional = append(additional, "-F "+backing_fmt+" ")
|
|
}
|
|
|
|
format := string(strings.ReplaceAll(strings.ToLower(reflect.TypeOf(i).String()), "*formats.", ""))
|
|
cmd := fmt.Sprintf("qemu-img create -f %s %v%v%v %v", format, strings.Join(additional, ""), getOptions(data), imagename, size)
|
|
|
|
// Todo remove this when done testing
|
|
fmt.Println(cmd)
|
|
err, _, errout := gosh.RunOutput(cmd)
|
|
|
|
if errout != "" {
|
|
return fmt.Errorf("%v", errout)
|
|
}
|
|
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
}
|
|
|
|
return nil
|
|
}
|