You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
104 lines
2.0 KiB
104 lines
2.0 KiB
package config
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"github.com/taoshihan1991/imaptool/tools"
|
|
"io/ioutil"
|
|
"os"
|
|
)
|
|
var(
|
|
PageSize uint=10
|
|
)
|
|
const Dir = "config/"
|
|
const AccountConf = Dir + "account.json"
|
|
const MysqlConf = Dir + "mysql.json"
|
|
type Mysql struct{
|
|
Server string
|
|
Port string
|
|
Database string
|
|
Username string
|
|
Password string
|
|
}
|
|
type Config struct {
|
|
Mysql *Mysql
|
|
}
|
|
func CreateConfig()*Config{
|
|
mysql :=CreateMysql()
|
|
c:=&Config{
|
|
Mysql: mysql,
|
|
}
|
|
return c
|
|
}
|
|
func CreateMysql() *Mysql {
|
|
var mysql Mysql
|
|
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 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 GetAccount() map[string]string {
|
|
var account map[string]string
|
|
isExist, _ := tools.IsFileExist(AccountConf)
|
|
if !isExist {
|
|
return account
|
|
}
|
|
info, err := ioutil.ReadFile(AccountConf)
|
|
if err != nil {
|
|
return account
|
|
}
|
|
|
|
err = json.Unmarshal(info, &account)
|
|
return account
|
|
}
|
|
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)
|
|
}
|