From acd70ea821352efd3ed7253f2415edd7f75da765 Mon Sep 17 00:00:00 2001
From: wanyaoqi <wanyaoqi@yunion.cn>
Date: Wed, 2 Jan 2019 16:29:22 +0800
Subject: [PATCH] 20190102 tempcommit

---
 pkg/util/fileutils2/fileutils.go | 90 ++++++++++++++++++++++++++++++++
 pkg/util/fuseutils/fuseutils.go  | 64 +++++++++++++++++++++++
 2 files changed, 154 insertions(+)
 create mode 100644 pkg/util/fuseutils/fuseutils.go

diff --git a/pkg/util/fileutils2/fileutils.go b/pkg/util/fileutils2/fileutils.go
index c1db9e8d16..373f064c5b 100644
--- a/pkg/util/fileutils2/fileutils.go
+++ b/pkg/util/fileutils2/fileutils.go
@@ -232,3 +232,93 @@ func (hf HostsFile) String() string {
 func Writable(path string) bool {
 	return unix.Access(path, unix.W_OK) == nil
 }
+
+func FsFormatToDiskType(fsFormat string) string {
+	switch {
+	case fsFormat == "swap":
+		return "linux-swap"
+	case strings.HasPrefix(fsFormat, "ext") || fsFormat == "xfs":
+		return "ext2"
+	case strings.HasPrefix(fsFormat, "fat"):
+		return "fat32"
+	case fsFormat == "ntfs":
+		return fsFormat
+	default:
+		return ""
+	}
+}
+
+func Mkpartition(imagePath, fsFormat string) error {
+	var (
+		parted    = "/sbin/parted"
+		labelType = "gpt"
+		diskType  = FsFormatToDiskType(fsFormat)
+	)
+
+	if len(diskType) == 0 {
+		return fmt.Errorf("Unknown fsFormat %s", fsFormat)
+	}
+
+	// 创建一个新磁盘分区表类型, ex: mbr gpt msdos ...
+	err := exec.Command(parted, "-s", imagePath, "mklabel", labelType).Run()
+	if err != nil {
+		log.Errorf("mklabel %s %s error %s", imagePath, fsFormat, err)
+		return err
+	}
+
+	// 创建一个part-type类型的分区, part-type可以是："primary", "logical", "extended"
+	// 如果指定fs-type(即diskType)则在创建分区的同时进行格式化
+	err = exec.Command(parted, "-s", "-a", "cylinder",
+		imagePath, "mkpart", "primary", diskType, "0", "100%").Run()
+	if err != nil {
+		log.Errorf("mkpart %s %s error %s", imagePath, fsFormat, err)
+		return err
+	}
+	return nil
+}
+
+func FormatPartition(path, fs, uuid string) error {
+	var cmd, cmdUuid []string
+	switch {
+	case fs == "swap":
+		cmd = []string{"mkswap", "-U", uuid}
+	case fs == "ext2":
+		cmd = []string{"mkfs.ext2"}
+		cmdUuid = []string{"tune2fs", "-U", uuid}
+	case fs == "ext3":
+		cmd = []string{"mkfs.ext3"}
+		cmdUuid = []string{"tune2fs", "-U", uuid}
+	case fs == "ext4":
+		cmd = []string{"mkfs.ext4", "-E", "lazy_itable_init=1"}
+		cmdUuid = []string{"tune2fs", "-U", uuid}
+	case fs == "ext4dev":
+		cmd = []string{"mkfs.ext4dev", "-E", "lazy_itable_init=1"}
+		cmdUuid = []string{"tune2fs", "-U", uuid}
+	case strings.HasPrefix(fs, "fat"):
+		cmd = []string{"mkfs.msdos"}
+	// #case fs == "ntfs":
+	// #    cmd = []string{"/sbin/mkfs.ntfs"}
+	case fs == "xfs":
+		cmd = []string{"mkfs.xfs", "-f"}
+		cmdUuid = []string{"xfs_admin", "-U", uuid}
+	}
+
+	if len(cmd) > 0 {
+		var cmds = cmd
+		cmds = append(cmds, path)
+		if err := exec.Command(cmds[0], cmds[1:]...).Run(); err != nil {
+			log.Errorln(err)
+			return err
+		}
+		if len(cmdUuid) > 0 {
+			cmds = cmdUuid
+			cmds = append(cmds, path)
+			if err := exec.Command(cmds[0], cmds[1:]...).Run(); err != nil {
+				log.Errorln(err)
+				return err
+			}
+		}
+		return nil
+	}
+	return fmt.Errorf("Unknown fs %s", fs)
+}
diff --git a/pkg/util/fuseutils/fuseutils.go b/pkg/util/fuseutils/fuseutils.go
new file mode 100644
index 0000000000..ef33ab3dad
--- /dev/null
+++ b/pkg/util/fuseutils/fuseutils.go
@@ -0,0 +1,64 @@
+package fuseutils
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+	"path"
+	"strings"
+	"time"
+
+	"yunion.io/x/log"
+	"yunion.io/x/onecloud/pkg/util/fileutils2"
+)
+
+const DEFAULT_BLOCKSIZE = 8
+
+func MountFusefs(fetcherfsPath, url, tmpdir, token, mntpath string, blocksize int) error {
+	var metaPath = path.Join(mntpath, "meta")
+	if f, err := os.OpenFile(metaPath, os.O_RDONLY, 0644); err == nil {
+		f.Close()
+		log.Infof("Already mounted: %s, skip...", mntpath)
+		return nil
+	}
+
+	// is mounted
+	if err := exec.Command("mountpoint", mntpath).Run(); err == nil {
+		exec.Command("umount", mntpath).Run()
+	}
+
+	if !fileutils2.Exists(tmpdir) {
+		if err := exec.Command("mkdir", "-p", tmpdir).Run(); err != nil {
+			return err
+		}
+	}
+
+	if !fileutils2.Exists(mntpath) {
+		if err := exec.Command("mkdir", "-p", mntpath).Run(); err != nil {
+			return err
+		}
+	}
+
+	var opts = fmt.Sprintf("url=%s", url)
+	opts += fmt.Sprintf(",tmpdir=%s", tmpdir)
+	opts += fmt.Sprintf(",token=%s", token)
+	opts += fmt.Sprintf(",blocksize=%d", blocksize)
+
+	var cmd = []string{fetcherfsPath, "-s", "-p", opts, mntpath}
+	log.Infof("%s", strings.Join(cmd, " "))
+	err := exec.Command(cmd[0], cmd[1:]...).Run()
+	if err != nil {
+		log.Errorf("Mount fetcherfs filed: %s", err)
+		exec.Command("umount", mntpath).Run()
+		return err
+	}
+
+	time.Sleep(200 * time.Millisecond)
+	if f, err := os.OpenFile(metaPath, os.O_RDONLY, 0644); err == nil {
+		f.Close()
+		return nil
+	} else {
+		log.Errorln(err)
+		return err
+	}
+}