mirror of
https://github.com/FloatTech/ZeroBot-Plugin.git
synced 2026-02-12 18:20:27 +00:00
Feature aipaint (#450)
This commit is contained in:
54
plugin/aipaint/config.go
Normal file
54
plugin/aipaint/config.go
Normal file
@@ -0,0 +1,54 @@
|
||||
package aipaint
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"os"
|
||||
|
||||
"github.com/FloatTech/floatbox/file"
|
||||
)
|
||||
|
||||
// 配置结构体
|
||||
type serverConfig struct {
|
||||
BaseURL string `json:"base_url"`
|
||||
Token string `json:"token"`
|
||||
file string
|
||||
}
|
||||
|
||||
func newServerConfig(file string) *serverConfig {
|
||||
return &serverConfig{
|
||||
file: file,
|
||||
}
|
||||
}
|
||||
|
||||
func (cfg *serverConfig) save(baseURL, token string) (err error) {
|
||||
cfg.BaseURL = baseURL
|
||||
cfg.Token = token
|
||||
reader, err := os.Create(cfg.file)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer reader.Close()
|
||||
return json.NewEncoder(reader).Encode(cfg)
|
||||
}
|
||||
|
||||
func (cfg *serverConfig) load() (aipaintServer, token string, err error) {
|
||||
if cfg.BaseURL != "" && cfg.Token != "" {
|
||||
aipaintServer = cfg.BaseURL
|
||||
token = cfg.Token
|
||||
return
|
||||
}
|
||||
if file.IsNotExist(cfg.file) {
|
||||
err = errors.New("no server config")
|
||||
return
|
||||
}
|
||||
reader, err := os.Open(cfg.file)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
defer reader.Close()
|
||||
err = json.NewDecoder(reader).Decode(cfg)
|
||||
aipaintServer = cfg.BaseURL
|
||||
token = cfg.Token
|
||||
return
|
||||
}
|
||||
Reference in New Issue
Block a user