Golang goconfig.SaveConfigFile函數代碼示例

func SaveLocalNodes() {
	if err := goconfig.SaveConfigFile(LocalNodes,
		path.Join(HomeDir+LocalNodesFile)); err != nil {
		log.Error("", "Fail to save localnodes.list:")
		log.Error("", "\t"+err.Error())
	}
}

func InitModels() {
	if !com.IsFile(_CFG_PATH) {
		os.Create(_CFG_PATH)
	}

	var err error
	Cfg, err = goconfig.LoadConfigFile(_CFG_PATH)
	if err == nil {
		beego.Info("Initialize app.ini")
	}

	setGithubCredentials(Cfg.MustValue("github", "client_id"),
		Cfg.MustValue("github", "client_secret"))

	docLock = new(sync.RWMutex)
	blogLock = new(sync.RWMutex)

	initMaps()

	// Start check ticker.
	checkTicker = time.NewTicker(5 * time.Minute)
	go checkTickerTimer(checkTicker.C)

	// ATTENTION: you'd better comment following code when developing.
	if needCheckUpdate() {
		checkFileUpdates()

		Cfg.SetValue("app", "update_check_time", strconv.Itoa(int(time.Now().Unix())))
		goconfig.SaveConfigFile(Cfg, _CFG_PATH)
	}
}

func CreateIniFile() error {
	f, err := os.OpenFile("./server.ini", os.O_CREATE|os.O_RDWR, 0666)
	if err != nil {
		return err
	}

	defer f.Close()

	config, err = goconfig.LoadConfigFile("./server.ini")
	if err != nil {
		return err
	}

	config.SetSectionComments("App", "")
	config.SetSectionComments("User", "")

	config.SetValue("App", "IP", "127.0.0.1")
	config.SetValue("App", "Port", "8080")
	config.SetValue("User", "username", "admin")
	config.SetValue("User", "password", helper.MD5("admin"))

	err = goconfig.SaveConfigFile(config, "./server.ini")
	if err != nil {
		return err
	}

	return nil
}

// scan a directory and gen a gopm file
func runGen(ctx *cli.Context) {
	setup(ctx)

	if !com.IsExist(".gopmfile") {
		os.Create(".gopmfile")
	}

	gf, err := goconfig.LoadConfigFile(".gopmfile")
	if err != nil {
		log.Error("gen", "Cannot load gopmfile:")
		log.Fatal("", "\t"+err.Error())
	}

	// Get dependencies.
	imports := doc.GetAllImports([]string{workDir},
		parseTarget(gf.MustValue("target", "path")), ctx.Bool("example"))

	for _, p := range imports {
		p = doc.GetProjectPath(p)
		if strings.HasSuffix(workDir, p) {
			continue
		}
		if value := gf.MustValue("deps", p); len(value) == 0 {
			gf.SetValue("deps", p, "")
		}
	}

	err = goconfig.SaveConfigFile(gf, ".gopmfile")
	if err != nil {
		log.Error("gen", "Fail to save gopmfile:")
		log.Fatal("", "\t"+err.Error())
	}

	log.Success("SUCC", "gen", "Generate gopmfile successfully!")
}

func main() {
	read, err := goconfig.LoadConfigFile("read.ini")
	if err != nil {
		fmt.Println(err)
	}
	goconfig.PrettyFormat = false

	/*ini cfg配置文件讀取*/
	fmt.Println(read.GetSectionList())              // 獲取選項名稱
	fmt.Println(read.GetKeyList("Cody"))            // 獲取Cody選項下的key
	fmt.Println(read.GetSection("Cody"))            // 以map形式獲取Cody選項下的key和value的值
	fmt.Println(read.GetSectionComments("work"))    // 獲取選項work上的以“#”和“;”開頭的注解
	fmt.Println(read.GetKeyComments("Cody", "sex")) // 獲取選項Cody的sex鍵上的注解
	fmt.Println(read.GetValue("Cody", "name"))      // 獲取選項Cody的name鍵的值

	read.DeleteKey("Cody", "hell")
	// /**/
	// read.SetValue("Cody", "name", "cody.guo") // 設置選項Cody的name鍵的值為cody.guo
	goconfig.SaveConfigFile(read, "read.ini") // 保存

	// read.DeleteKey("work", "work")                 // 刪除選項work中的work鍵
	// goconfig.SaveConfigFile(read, "readWrite.ini") // 另存為readWrite.ini

	// read.DeleteSection("work")                // 刪除選項work
	// goconfig.SaveConfigFile(read, "read.ini") // 保存配置文件,會把 " : "的修改為" = "
}

// SaveConfig saves configuration file.
// if configKey has been set, the file will be encrypted.
func SaveConfig() {
	if len(configKey) == 0 {
		err := goconfig.SaveConfigFile(configData, ConfigPath)
		if err != nil {
			log.Fatalf("Failed to save config file: %v", err)
		}
		err = os.Chmod(ConfigPath, 0600)
		if err != nil {
			ErrorLog(nil, "Failed to set permissions on config file: %v", err)
		}
		return
	}
	var buf bytes.Buffer
	err := goconfig.SaveConfigData(configData, &buf)
	if err != nil {
		log.Fatalf("Failed to save config file: %v", err)
	}

	f, err := os.Create(ConfigPath)
	if err != nil {
		log.Fatalf("Failed to save config file: %v", err)
	}

	fmt.Fprintln(f, "# Encrypted rclone configuration File")
	fmt.Fprintln(f, "")
	fmt.Fprintln(f, "RCLONE_ENCRYPT_V0:")

	// Generate new nonce and write it to the start of the ciphertext
	var nonce [24]byte
	n, _ := rand.Read(nonce[:])
	if n != 24 {
		log.Fatalf("nonce short read: %d", n)
	}
	enc := base64.NewEncoder(base64.StdEncoding, f)
	_, err = enc.Write(nonce[:])
	if err != nil {
		log.Fatalf("Failed to write config file: %v", err)
	}

	var key [32]byte
	copy(key[:], configKey[:32])

	b := secretbox.Seal(nil, buf.Bytes(), &nonce, &key)
	_, err = enc.Write(b)
	if err != nil {
		log.Fatalf("Failed to write config file: %v", err)
	}
	_ = enc.Close()
	err = f.Close()
	if err != nil {
		log.Fatalf("Failed to close config file: %v", err)
	}

	err = os.Chmod(ConfigPath, 0600)
	if err != nil {
		ErrorLog(nil, "Failed to set permissions on config file: %v", err)
	}
}

// SaveConfig saves configuration file.
func SaveConfig() {
	err := goconfig.SaveConfigFile(ConfigFile, ConfigPath)
	if err != nil {
		log.Fatalf("Failed to save config file: %v", err)
	}
	err = os.Chmod(ConfigPath, 0600)
	if err != nil {
		log.Printf("Failed to set permissions on config file: %v", err)
	}
}

func (p *ProxyConf) Write(proxy []Proxy) {
	if len(proxy) == 0 {
		return
	}
	p.load()
	for _, pro := range proxy {
		switch pro.pTtype {
		case HTTP:
			p.SetValue(HTTP, pro.ip, pro.port)
		case HTTPS:
			p.SetValue(HTTPS, pro.ip, pro.port)
		}
	}
	goconfig.SaveConfigFile(p.ConfigFile, proxyFile)
}

func main() {
	cfg, err := goconfig.LoadConfigFile("conf.ini")
	if err != nil {
		log.Fatalf("無法加載配置文件: %s", err)
	}

	value, err := cfg.GetValue(goconfig.DEFAULT_SECTION, "key_default")
	if err != nil {
		log.Fatalf("無法獲取鍵值(%s): %s", "key_default", nil)
	}

	log.Printf("%s > %s: %s", goconfig.DEFAULT_SECTION, "key_default", value)

	isInsert := cfg.SetValue(goconfig.DEFAULT_SECTION, "key_default", "這是新的值")
	if err != nil {
		log.Fatalf("無法設置鍵值(%s): %s", "key_default", err)
	}
	log.Printf("設置鍵值 %s 為插入操作: %v", "key_default", isInsert)

	comment := cfg.GetSectionComments("super")
	log.Printf("分區 %s 的注釋: %s", "super", comment)

	v := cfg.SetSectionComments("super", "# 這是新的分區注釋")
	log.Printf("分區 %s 的注釋被插入或刪除: %v", "super", v)

	vInt, err := cfg.Int("must", "int")
	if err != nil {
		log.Fatalf("無法獲取鍵值(%s): %s)", "int", err)
	}

	log.Printf("%s > %s: %v", "must", "int", vInt)

	vBool := cfg.MustBool("must", "bool")
	log.Printf("%s > %s: %v", "must", "bool", vBool)

	vBool = cfg.MustBool("must", "bool404")
	log.Printf("%s > %s: %v", "must", "bool", vBool)

	ok := cfg.DeleteKey("must", "string")
	log.Printf("刪除鍵值 %s 是否成功: %v", "string", ok)

	err = goconfig.SaveConfigFile(cfg, "conf_save.ini")
	if err != nil {
		log.Fatalf("無法保存配置文件: %s", err)
	}
}

func runGen(ctx *cli.Context) {
	setup(ctx)

	if !com.IsExist(".gopmfile") {
		os.Create(".gopmfile")
	}

	gf, err := goconfig.LoadConfigFile(".gopmfile")
	if err != nil {
		log.Error("gen", "Cannot load gopmfile:")
		log.Fatal("", "\t"+err.Error())
	}

	targetPath := parseTarget(gf.MustValue("target", "path"))
	// Get and set dependencies.
	imports := doc.GetAllImports([]string{workDir}, targetPath, ctx.Bool("example"), false)
	for _, p := range imports {
		p = doc.GetProjectPath(p)
		// Skip subpackage(s) of current project.
		if isSubpackage(p, targetPath) {
			continue
		}

		// Check if user specified the version.
		if value := gf.MustValue("deps", p); len(value) == 0 {
			gf.SetValue("deps", p, "")
		}
	}

	// Get and set resources.
	res := make([]string, 0, len(commonRes))
	for _, cr := range commonRes {
		if com.IsExist(cr) {
			res = append(res, cr)
		}
	}
	gf.SetValue("res", "include", strings.Join(res, "|"))

	err = goconfig.SaveConfigFile(gf, ".gopmfile")
	if err != nil {
		log.Error("gen", "Fail to save gopmfile:")
		log.Fatal("", "\t"+err.Error())
	}

	log.Success("SUCC", "gen", "Generate gopmfile successfully!")
}

func runConfig(ctx *cli.Context) {
	setup(ctx)

	if len(ctx.Args()) == 0 {
		log.Error("config", "Cannot start command:")
		log.Fatal("", "\tNo section specified")
	}

	switch ctx.Args()[0] {
	case "github":
		if len(ctx.Args()) < 3 {
			log.Error("config", "Cannot config section 'github'")
			log.Fatal("", "\tNot enough arguments for client_id and client_secret")
		}
		doc.Cfg.SetValue("github", "client_id", ctx.Args()[1])
		doc.Cfg.SetValue("github", "client_secret", ctx.Args()[2])
		goconfig.SaveConfigFile(doc.Cfg, path.Join(doc.HomeDir, doc.GOPM_CONFIG_FILE))
	}

	log.Success("SUCC", "config", "Command executed successfully!")
}

func main() {
	cfg, err := goconfig.LoadConfigFile("conf.ini")
	if err != nil {
		log.Fatalf("無法加載配置文件:%s", err)
	}
	value, err := cfg.GetValue(goconfig.DEFAULT_SECTION, "key_default")
	if err != nil {
		log.Fatalf("無法獲取值:%s", err)
	}
	log.Printf("%s > %s:%s", goconfig.DEFAULT_SECTION, "key_default", value)
	isInsert := cfg.SetValue(goconfig.DEFAULT_SECTION, "key_default", "這是新的值")
	log.Printf("設在鍵值%s: %v", "key_default", isInsert)
	comment := cfg.GetSectionComments("super")
	log.Printf("分區%s的注釋:%s", "supper", comment)
	_ = cfg.SetSectionComments("super", "# 這是一個新的注釋")
	_, err = cfg.Int("must", "int")
	err = goconfig.SaveConfigFile(cfg, "conf_save.ini")
	if err != nil {
		log.Fatalf("無法獲取值:%s", err)
	}

}

func (p *ProxyConf) Update() {
	p.Read()
	ok := make(chan bool)
	for _, proxy := range p.proxy {
		ip := proxy.ip
		prot := proxy.port
		go func() {
			_, err := net.DialTimeout("tcp", fmt.Sprintf("%s:%s", ip, prot), 10*time.Second)
			if err != nil {
				fmt.Printf("代理服務器 [%s:%s] 不可用，已刪除.\n", ip, prot)
				p.DeleteKey(HTTP, ip)
				p.DeleteKey(HTTPS, ip)
			}
			ok <- true
		}()
	}
	for range p.proxy {
		<-ok
	}
	goconfig.SaveConfigFile(p.ConfigFile, proxyFile)
	p.Reload()
}

func main() {
	// 讀取證書文件
	rootPEM, err := ioutil.ReadFile("cert.pem")
	if err != nil {
		log.Println("讀取 cert.pem 出錯:", err, "請檢查文件是否存在")
		return
	}
	roots := x509.NewCertPool()
	ok := roots.AppendCertsFromPEM(rootPEM)
	if !ok {
		log.Println("證書分析失敗，請檢查證書文件是否正確")
		return
	}

	// 加載配置文件
	cfg, err := goconfig.LoadConfigFile("client.ini")
	if err != nil {
		log.Println("配置文件加載失敗，自動重置配置文件:", err)
		cfg, err = goconfig.LoadFromData([]byte{})
		if err != nil {
			log.Println(err)
			return
		}
	}

	var (
		port, ok1       = cfg.MustValueSet("client", "port", "7071")
		key, ok2        = cfg.MustValueSet("client", "key", "eGauUecvzS05U5DIsxAN4n2hadmRTZGBqNd2zsCkrvwEBbqoITj36mAMk4Unw6Pr")
		serverHost, ok3 = cfg.MustValueSet("server", "host", "localhost")
		serverPort, ok4 = cfg.MustValueSet("server", "port", "8081")
	)

	// 如果缺少配置則保存為默認配置
	if ok1 || ok2 || ok3 || ok4 {
		err = goconfig.SaveConfigFile(cfg, "client.ini")
		if err != nil {
			log.Println("配置文件保存失敗:", err)
		}
	}

	ln, err := net.Listen("tcp", ":"+port)
	if err != nil {
		log.Println(err)
		return
	}
	defer ln.Close()

	log.Println("|>>>>>>>>>>>>>>>|<<<<<<<<<<<<<<<|")
	log.Println("程序版本:" + VERSION)
	log.Println("代理端口:" + port)
	log.Println("Key:" + key)
	log.Println("服務器地址:" + serverHost + ":" + serverPort)
	log.Println("|>>>>>>>>>>>>>>>|<<<<<<<<<<<<<<<|")

	s := &serve{
		serverHost: serverHost,
		serverPort: serverPort,
		key:        key,
		conf: &tls.Config{
			RootCAs: roots,
		},
	}

	// 登錄
	if err = s.updateSession(); err != nil {
		log.Println("與服務器連接失敗:", err)
		return
	}
	log.Println("登錄成功,服務器連接完畢")
	go s.updateSessionLoop()

	for {
		conn, err := ln.Accept()
		if err != nil {
			log.Println(err)
			continue
		}
		go s.handleConnection(conn)
	}
}

// SaveConfig saves configuration file.
func SaveConfig() error {
	return goconfig.SaveConfigFile(Cfg, "conf/app.ini")
}