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.
go-fly/config/config.go

146 lines
2.8 KiB

package config
import (
"encoding/json"
"fmt"
"github.com/taoshihan1991/imaptool/tools"
"io/ioutil"
"os"
)
var (
PageSize uint = 10
VisitorPageSize uint = 8
Version = "0.1.2"
GoflyConfig *Config
)
const Dir = "config/"
4 years ago
const AccountConf = Dir + "account.json"
const MysqlConf = Dir + "mysql.json"
4 years ago
const MailConf = Dir + "mail.json"
const MainConf = Dir + "config.json"
func init() {
//配置文件
GoflyConfig = CreateConfig()
}
type Mysql struct {
Server string
Port string
4 years ago
Database string
Username string
Password string
}
4 years ago
type MailServer struct {
Server, Email, Password string
}
4 years ago
type Config struct {
Upload string
NoticeServerJiang bool
4 years ago
}
func CreateConfig() *Config {
var configObj Config
c := &Config{
Upload: "static/upload/",
NoticeServerJiang: false,
4 years ago
}
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
4 years ago
}
4 years ago
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
}
4 years ago
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
}
4 years ago
func GetAccount() map[string]string {
var account map[string]string
4 years ago
isExist, _ := tools.IsFileExist(AccountConf)
if !isExist {
return account
}
4 years ago
info, err := ioutil.ReadFile(AccountConf)
if err != nil {
return account
}
4 years ago
err = json.Unmarshal(info, &account)
return account
}
4 years ago
func GetUserInfo(uid string) map[string]string {
var userInfo map[string]string
4 years ago
userFile := Dir + "sess_" + uid + ".json"
isExist, _ := tools.IsFileExist(userFile)
if !isExist {
return userInfo
}
4 years ago
info, err := ioutil.ReadFile(userFile)
if err != nil {
return userInfo
}
4 years ago
err = json.Unmarshal(info, &userInfo)
return userInfo
}
4 years ago
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)
4 years ago
str := "{\r\n"
for k, v := range info {
str += fmt.Sprintf(`"%s":"%s",`, k, v)
}
4 years ago
str += fmt.Sprintf(`"session_id":"%s"%s}`, uid, "\r\n")
file.WriteString(str)
4 years ago
}