update mod

1 files changed, 220 insertions, 0 deletions

diff --git a/mkvlib/utils.go b/mkvlib/utils.go
new file mode 100644
index 0000000..a76a05d
--- /dev/null
+++ b/mkvlib/utils.go
@@ -0,0 +1,220 @@
+package mkvlib
+
+import (
+	"errors"
+	"fmt"
+	"io"
+	"math/rand"
+	"os"
+	"os/exec"
+	"path"
+	"path/filepath"
+	"regexp"
+	"strings"
+	"time"
+)
+
+func newProcess(stdin io.Reader, stdout, stderr io.Writer, dir, prog string, args ...string) (p *os.Process, err error) {
+	cmd := exec.Command(prog, args...)
+	if dir != "" {
+		cmd.Dir = dir
+	}
+	if stdin != nil {
+		cmd.Stdin = stdin
+	}
+	if stdout != nil {
+		cmd.Stdout = stdout
+	}
+	if stderr != nil {
+		cmd.Stderr = stderr
+	}
+	err = cmd.Start()
+	if err == nil {
+		p = cmd.Process
+	}
+	return
+}
+
+func newDir(path string) error {
+	return os.MkdirAll(path, os.ModePerm)
+}
+
+func queryPath(path string, cb func(string) bool) error {
+	return filepath.Walk(path, func(path string, f os.FileInfo, err error) error {
+		if f == nil {
+			return err
+		}
+		if f.IsDir() {
+			return nil
+		}
+		if cb(path) {
+			return nil
+		}
+		return errors.New("call cb return false")
+	})
+}
+
+func findPath(path, expr string) (list []string, err error) {
+	list = make([]string, 0)
+	reg, e := regexp.Compile(expr)
+	if e != nil {
+		err = e
+		return
+	}
+	err = queryPath(path, func(path string) bool {
+		if expr == "" || reg.MatchString(path) {
+			list = append(list, path)
+		}
+		return true
+	})
+	return
+}
+
+func copyFolder(src, dst string) error {
+	e, f := isExists(src)
+	if !e {
+		return errors.New("src is not exists")
+	}
+	if !f {
+		return errors.New("src is not folder")
+	}
+	if newDir(dst) != nil {
+		return errors.New("faild to create dst folder")
+	}
+	s := len(src)
+	if _, n, _, _ := splitPath(dst); n == "" {
+		_, n, _, _ = splitPath(src)
+		if n == "" {
+			_, n, _, _ = splitPath(src[:len(src)-1])
+		}
+		dst = fmt.Sprintf("%s/%s", dst, n)
+	}
+	return filepath.Walk(src, func(path string, f os.FileInfo, err error) error {
+		if f == nil {
+			return err
+		}
+		if f.IsDir() {
+			return nil
+		}
+		return copyFile(path, dst+"/"+path[s:])
+	})
+}
+
+func newFile(fp string) (file *os.File, err error) {
+	dir, _ := filepath.Split(fp)
+	if dir != "" {
+		err = newDir(dir)
+		if err != nil {
+			return
+		}
+	}
+	if err == nil {
+		file, err = os.Create(fp)
+	}
+	return
+}
+
+func openFile(filepath string, readOnly, create bool) (file *os.File, err error) {
+	f := os.O_RDWR | os.O_CREATE
+	if readOnly {
+		f = os.O_RDONLY
+	}
+	file, err = os.OpenFile(filepath, f, os.ModePerm)
+	if err != nil && create {
+		file, err = newFile(filepath)
+	}
+	return
+}
+
+func copyFile(src, dst string) error {
+	e, f := isExists(src)
+	if !e {
+		return errors.New("src is not exists")
+	}
+	if f {
+		return errors.New("src is not file")
+	}
+	if _, n, _, _ := splitPath(dst); n == "" {
+		_, n, _, _ = splitPath(src)
+		dst = path.Join(dst, n)
+	}
+	sf, err := openFile(src, true, false)
+	if err != nil {
+		return err
+	}
+	defer sf.Close()
+	df, err := openFile(dst, false, true)
+	if err != nil {
+		return err
+	}
+	defer df.Close()
+
+	_, err = io.Copy(df, sf)
+	return err
+}
+
+func splitPath(p string) (dir, name, ext, namewithoutext string) {
+	dir, name = filepath.Split(p)
+	ext = filepath.Ext(name)
+	n := strings.LastIndex(name, ".")
+	if n > 0 {
+		namewithoutext = name[:n]
+	}
+	return
+}
+
+func isExists(path string) (exists bool, isFolder bool) {
+	f, err := os.Stat(path)
+	exists = err == nil || os.IsExist(err)
+	if exists {
+		isFolder = f.IsDir()
+	}
+	return
+}
+
+func copyFileOrDir(src, dst string) error {
+	e, f := isExists(src)
+	if !e {
+		return errors.New("src is not exists")
+	}
+	if !f {
+		return copyFile(src, dst)
+	}
+	return copyFolder(src, dst)
+}
+
+func randomStr(l int) string {
+	str := "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ"
+	bytes := []byte(str)
+	var result []byte
+	lstr := len(str) - 1
+	for i := 0; i < l; i++ {
+		n := randomNumber(0, lstr)
+		result = append(result, bytes[n])
+	}
+	return string(result)
+}
+
+var r = rand.New(rand.NewSource(time.Now().UnixNano()))
+
+func randomN(n int) int {
+	return r.Intn(n)
+}
+
+func randomNumber(min, max int) int {
+	sub := max - min + 1
+	if sub <= 1 {
+		return min
+	}
+	return min + randomN(sub)
+}
+
+func findMKVs(dir string) []string {
+	list, _ := findPath(dir, `\.mkv$`)
+	return list
+}
+
+func findFonts(dir string) []string {
+	list, _ := findPath(dir, `\.((ttf)|(otf)|(ttc)|(fon))$`)
+	return list
+}