flytalk/config/config.go

package config

import (
	"encoding/json"
	"fmt"
	"git.wanbits.cc/sin/flytalk/tools"
	"io/ioutil"
	"os"
)

var (
	PageSize uint = 10

	GoflyConfig *Config
	QiniuConfig *Qiniu
)

const (
	VERSION           = "0.1.2"
	PAGE_SIZE_VISITOR = 8
	DIR               = "config"
)

const Dir = "config/"
const MysqlConf = Dir + "mysql.json"
const QiniuFile = Dir + "qiniu.json"
const MailConf = Dir + "mail.json"
const MainConf = Dir + "config.json"

func init() {
	//配置文件
	GoflyConfig = CreateConfig()
	QiniuConfig = CreateQiniu(QiniuFile)
}

type Database struct {
	Driver   string
	Server   string
	Port     string
	Db       string
	Username string
	Password string
}

type Qiniu struct {
	Access string
	Secret string
	Bucket string
	Zone   string
	Domain string
}

type MailServer struct {
	Server, Email, Password string
}

type Config struct {
	Upload       string
	NoticeServer bool
}

func CreateConfig() *Config {
	var configObj Config
	c := &Config{
		Upload:       "static/upload/",
		NoticeServer: false,
	}
	isExist, _ := tools.IsFileExist(MainConf)
	if !isExist {
		return c
	}
	info, err := ioutil.ReadFile(MainConf)
	if err != nil {
		return c
	}
	err = json.Unmarshal(info, &configObj)
	return &configObj
}

func CreateMailServer() *MailServer {
	var imap MailServer
	isExist, _ := tools.IsFileExist(MailConf)
	if !isExist {
		return &imap
	}
	info, err := ioutil.ReadFile(MailConf)
	if err != nil {
		return &imap
	}

	err = json.Unmarshal(info, &imap)
	return &imap
}

func CreateMysql(f string) *Database {
	var mysql Database
	isExist, _ := tools.IsFileExist(f)
	if !isExist {
		return &mysql
	}
	info, err := ioutil.ReadFile(f)
	if err != nil {
		return &mysql
	}

	err = json.Unmarshal(info, &mysql)
	return &mysql
}

func CreateQiniu(f string) *Qiniu {
	var q Qiniu
	isExist, _ := tools.IsFileExist(f)
	if !isExist {
		fmt.Println(f)
		return &q
	}
	info, err := ioutil.ReadFile(f)
	if err != nil {
		fmt.Println("bb")
		return &q
	}

	_ = json.Unmarshal(info, &q)
	return &q
}

func GetMysql() map[string]string {
	var mysql map[string]string
	isExist, _ := tools.IsFileExist(MysqlConf)
	if !isExist {
		return mysql
	}
	info, err := ioutil.ReadFile(MysqlConf)
	if err != nil {
		return mysql
	}

	err = json.Unmarshal(info, &mysql)
	return mysql
}

func GetUserInfo(uid string) map[string]string {
	var userInfo map[string]string
	userFile := Dir + "sess_" + uid + ".json"
	isExist, _ := tools.IsFileExist(userFile)
	if !isExist {
		return userInfo
	}
	info, err := ioutil.ReadFile(userFile)
	if err != nil {
		return userInfo
	}

	err = json.Unmarshal(info, &userInfo)
	return userInfo
}

func SetUserInfo(uid string, info map[string]string) {
	userFile := Dir + "sess_" + uid + ".json"
	isExist, _ := tools.IsFileExist(Dir)
	if !isExist {
		os.Mkdir(Dir, os.ModePerm)
	}
	file, _ := os.OpenFile(userFile, os.O_RDWR|os.O_CREATE, os.ModePerm)
	str := "{\r\n"
	for k, v := range info {
		str += fmt.Sprintf(`"%s":"%s",`, k, v)
	}
	str += fmt.Sprintf(`"session_id":"%s"%s}`, uid, "\r\n")
	file.WriteString(str)
}