Exemplo n.º 1
0
func main() {
	// check input parameter,-p package.rar
	if len(os.Args) < 2 {
		panic("Usage: distribute.exe -p xxxxx.rar[zip]")
	}
	dstPkg := flag.String("p", "C:/distribute004.zip", "for distribute file package")
	//dstPkg := os.Args[0]
	flag.Parse()

	dstPkgU := util.ReplaceWindowsPathSeparator(*dstPkg)
	//check file if not exist
	err := util.CheckFile(dstPkgU)

	if err != nil {
		log.Fatal(err)
	}

	// get config filename from package name
	confName := path.Base(dstPkgU)
	t1 := strings.Split(confName, ".")
	headerName := strings.Join(t1[:len(t1)-1], "")
	cfgFile := headerName + ".json"
	fullCfgFile := path.Join(path.Dir(dstPkgU), cfgFile)

	// Change path to os temporary path
	os.Chdir(os.TempDir())

	// Create sub temporary directory
	tmpName, err := ioutil.TempDir(os.TempDir(), "Dst")
	if err != nil {
		panic(err)
	}
	//fmt.Println(tmpName)
	//if crush clear temporary dir
	//defer func() {
	//	os.RemoveAll(tmpName)
	//}()

	fmt.Println("Begin uncompress file")

	// Uncompress pkg file to sub temporary path
	if path.Ext(dstPkgU) == ".rar" {
		if err := util.CheckFile(cmd); err != nil {
			panic(err)
		}
		if err := tar.UnRar(cmd, dstPkgU, tmpName); err != nil {
			panic(err)
		}
	} else {
		if err := tar.UnZip(dstPkgU, tmpName); err != nil {
			panic(err)
		}
	}

	// Change dir to sub temporary dir
	os.Chdir(tmpName)

	// Check Json
	// Read json config file
	body, _ := util.ReadTxt(fullCfgFile)
	var r apiR

	if err := json.Unmarshal([]byte(body), &r); err != nil {
		panic(err)
	}

	// Global parameter
	ownerVar := r.Global.USER
	groupVar := r.Global.GROUP
	modeVar := r.Global.ALLAUTH

	//check user and group in window os
	fmt.Println("Check owner")
	if err := util.CheckUG(ownerVar); err != nil {
		panic(err)
	}

	//check mode,only contain fwr-
	fmt.Println("Check mode")
	if !util.CheckM(modeVar) {
		panic("AllAUTH ERROR")
	}

	// dependence config file
	// Chmod all file;
	fmt.Println("Begin Change Globle Mode")
	for src := range r.Detail {
		//fmt.Println(key)
		mode.Chown(src, ownerVar, groupVar, modeVar)
	}

	fmt.Println("Begin Change Special Mode")
	for key, value := range r.Special {
		mode.Chown(key, ownerVar, groupVar, value)
	}

	fmt.Println("Begin copy files")
	// manager file
	//fmt.Println(tmpName)
	for key, value := range r.Detail {
		fi, _ := os.Stat(key)
		if fi.IsDir() {
			//fmt.Printf("Src:%v,Dst:%v\n", key, value)
			err := copy.Copy(key, path.Join(value, key))
			if err != nil {
				fmt.Printf("Copy Dir Error ---- %s\n", err)
			}
		} else {
			//fmt.Printf("Src:%v,Dst:%v\n", key, value)
			err := copy.Copy(key, path.Join(value, key))
			if err != nil {
				fmt.Printf("Copy File Error ---- %s\n", err)
			}
		}

	}

}
Exemplo n.º 2
0
func main() {
	// check input parameter,-p package.rar
	if len(os.Args) < 2 {
		panic("Usage: distribute.exe -p xxxxx.rar[zip]")
	}
	dstPkg := flag.String("p", " Need the fullpath pkg name", "for distribute file package")
	//dstPkg := os.Args[0]
	flag.Parse()

	dstPkgU := util.ReplaceWindowsPathSeparator(*dstPkg)
	//check file if not exist
	err := util.CheckFile(dstPkgU)

	if err != nil {
		log.Fatal(err)
	}

	// get config filename from package name
	confName := path.Base(dstPkgU)
	t1 := strings.Split(confName, ".")
	headerName := strings.Join(t1[:len(t1)-1], ".")
	cfgFile := headerName + ".json"
	fullCfgFile := path.Join(path.Dir(dstPkgU), cfgFile)

	// Change path to os temporary path
	os.Chdir(os.TempDir())

	// Create sub temporary directory
	tmpName, err := ioutil.TempDir(os.TempDir(), "Dst")
	if err != nil {
		log.Fatal(err)
	}
	//fmt.Println(tmpName)
	//if crush clear temporary dir
	defer func() {
		os.RemoveAll(tmpName)
	}()

	fmt.Println("Begin uncompress file")

	// Uncompress pkg file to sub temporary path
	if path.Ext(dstPkgU) == ".rar" {
		if err := util.CheckFile(cmd); err != nil {
			fmt.Printf("Uncompress file error: %v", err)
			os.Exit(1)
		}
		if err := tar.UnRar(cmd, dstPkgU, tmpName); err != nil {
			log.Fatal(err)
		}
	} else {
		if err := tar.UnZip(dstPkgU, tmpName); err != nil {
			log.Fatal(err)
		}
	}

	// Change dir to sub temporary dir
	os.Chdir(tmpName)

	// Check Json
	// Read json config file
	body, _ := util.ReadTxt(fullCfgFile)
	var r apiR

	if err := json.Unmarshal([]byte(body), &r); err != nil {
		log.Fatal(err)
	}

	// Global parameter
	ownerVar := r.Global.USER
	groupVar := r.Global.GROUP
	modeVar := r.Global.ALLAUTH

	//check user and group in window os
	fmt.Println("Check owner")
	if err := util.CheckUG(ownerVar); err != nil {
		log.Fatalf("The user [%v] is not found\n", ownerVar)
	}

	//check mode,only contain fwr-
	fmt.Println("Check mode")
	if !util.CheckM(modeVar) {
		log.Fatalf("AllAUTH %v FORMAT ERROR,ONLY f,w,r", modeVar)
	}

	//check source file(s)
	fmt.Println("Check src")
	var errPath []string
	for key, _ := range r.Special {
		if util.CheckFile(key) != nil {
			errPath = append(errPath, key)
		}
	}

	for key, _ := range r.Detail {
		if util.CheckFile(key) != nil {
			errPath = append(errPath, key)
		}
	}

	util.RemoveDuplicate(&errPath)

	if len(errPath) > 0 {
		log.Fatal("Src File(s) :" + strings.Join(errPath, ",") + " is not exist!!!")
	}

	// dependence config file
	// Chmod all file;
	fmt.Println("Begin Change Globle Mode")
	for src := range r.Detail {
		//fmt.Println(key)
		mode.ResetMode(src)
		mode.Chown(src, ownerVar, groupVar, modeVar)
	}

	fmt.Println("Begin Change Special Mode")
	for key, value := range r.Special {
		mode.Chown(key, ownerVar, groupVar, value)
	}

	fmt.Println("Begin copy files")
	// manager file
	//fmt.Println(tmpName)
	for key, value := range r.Detail {

		switch vv := value.(type) {
		case string:
			err := copy.Copy(key, path.Join(vv, key))
			if err != nil {
				fmt.Printf("Copy File Error ---- %s\n", err)
			}
		case []interface{}:
			for _, v := range vv {
				err := copy.Copy(key, path.Join(v.(string), key))
				if err != nil {
					fmt.Printf("Copy File Error ---- %s\n", err)
				}
			}
		default:
			fmt.Println("Detail format is error:", vv)
		}

		//fi, _ := os.Stat(key)
		//if fi.IsDir() {
		//fmt.Printf("Src:%v,Dst:%v\n", key, value)
		//	err := copy.Copy(key, path.Join(value, key))
		//	if err != nil {
		//		fmt.Printf("Copy Dir Error ---- %s\n", err)
		//	}
		//} else {
		//fmt.Printf("Src:%v,Dst:%v\n", key, value)
		//	err := copy.Copy(key, path.Join(value, key))
		//	if err != nil {
		//		fmt.Printf("Copy File Error ---- %s\n", err)
		//	}

	}

}