Albirew/nyaa-pantsu
Archivé
1
0
Bifurcation 0
Ce dépôt a été archivé le 2022-05-07. Vous pouvez voir ses fichiers ou le cloner, mais pas ouvrir de ticket ou de demandes d'ajout, ni soumettre de changements.
nyaa-pantsu/config/config.go

150 lignes
4 Kio
Go
Brut Vue normale Historique

package config
import (
"bufio"
"encoding/json"
"flag"
"fmt"
"io"
"os"
)
const (
// LastOldTorrentID is the highest torrent ID
// that was copied from the original Nyaa
LastOldTorrentID = 923000
TableName = "torrents"
CommentsTableName = "comments"
// for sukebei
//TableName = "sukebei_torrents"
//CommentsTableName = "sukebei_comments"
)
type Config struct {
Host string `json:"host"`
Port int `json:"port"`
DBType string `json:"db_type"`
// DBParams will be directly passed to Gorm, and its internal
// structure depends on the dialect for each db type
DBParams string `json:"db_params"`
DBLogMode string `json:"db_logmode"`
2017-05-10 19:29:35 +02:00
// tracker scraper config (required)
Scrape ScraperConfig `json:"scraper"`
2017-05-11 15:01:53 +02:00
// cache config
Cache CacheConfig `json:"cache"`
// search config
Search SearchConfig `json:"search"`
2017-05-07 21:51:37 +02:00
// optional i2p configuration
I2P *I2PConfig `json:"i2p"`
2017-05-13 19:58:48 +02:00
// filesize fetcher config
2017-05-14 13:20:34 +02:00
MetainfoFetcher MetainfoFetcherConfig `json:"metainfo_fetcher"`
// internationalization config
2017-05-14 00:55:17 +02:00
I18n I18nConfig `json:"i18n"`
}
2017-05-14 13:20:34 +02:00
var Defaults = Config{"localhost", 9999, "sqlite3", "./nyaa.db?cache_size=50", "default", DefaultScraperConfig, DefaultCacheConfig, DefaultSearchConfig, nil, DefaultMetainfoFetcherConfig, DefaultI18nConfig}
2017-05-14 00:55:17 +02:00
var allowedDatabaseTypes = map[string]bool{
"sqlite3": true,
"postgres": true,
"mysql": true,
"mssql": true,
}
var allowedDBLogModes = map[string]bool{
"default": true, // errors only
"detailed": true,
"silent": true,
}
func New() *Config {
var config Config
config.Host = Defaults.Host
config.Port = Defaults.Port
config.DBType = Defaults.DBType
config.DBParams = Defaults.DBParams
config.DBLogMode = Defaults.DBLogMode
2017-05-10 19:29:35 +02:00
config.Scrape = Defaults.Scrape
2017-05-11 15:01:53 +02:00
config.Cache = Defaults.Cache
2017-05-14 13:20:34 +02:00
config.MetainfoFetcher = Defaults.MetainfoFetcher
config.I18n = Defaults.I18n
return &config
}
// BindFlags returns a function which is to be used after
// flag.Parse to check and copy the flags' values to the Config instance.
func (config *Config) BindFlags() func() error {
confFile := flag.String("conf", "", "path to the configuration file")
dbType := flag.String("dbtype", Defaults.DBType, "database backend")
host := flag.String("host", Defaults.Host, "binding address of the server")
port := flag.Int("port", Defaults.Port, "port of the server")
dbParams := flag.String("dbparams", Defaults.DBParams, "parameters to open the database (see Gorm's doc)")
dbLogMode := flag.String("dblogmode", Defaults.DBLogMode, "database log verbosity (errors only by default)")
return func() error {
// You can override fields in the config file with flags.
config.Host = *host
config.Port = *port
config.DBParams = *dbParams
err := config.SetDBType(*dbType)
2017-05-06 20:43:15 +02:00
if err != nil {
return err
}
err = config.SetDBLogMode(*dbLogMode)
if err != nil {
return err
}
err = config.HandleConfFileFlag(*confFile)
return err
}
}
func (config *Config) HandleConfFileFlag(path string) error {
if path != "" {
file, err := os.Open(path)
if err != nil {
return fmt.Errorf("can't read file '%s'", path)
}
err = config.Read(bufio.NewReader(file))
if err != nil {
return fmt.Errorf("failed to parse file '%s' (%s)", path, err)
}
}
return nil
}
func (config *Config) SetDBType(db_type string) error {
if !allowedDatabaseTypes[db_type] {
return fmt.Errorf("unknown database backend '%s'", db_type)
}
config.DBType = db_type
return nil
}
func (config *Config) SetDBLogMode(db_logmode string) error {
if !allowedDBLogModes[db_logmode] {
return fmt.Errorf("unknown database log mode '%s'", db_logmode)
}
config.DBLogMode = db_logmode
return nil
}
func (config *Config) Read(input io.Reader) error {
return json.NewDecoder(input).Decode(config)
}
func (config *Config) Write(output io.Writer) error {
return json.NewEncoder(output).Encode(config)
}
func (config *Config) Pretty(output io.Writer) error {
data, err := json.MarshalIndent(config, "", "\t")
if err != nil {
return err
}
data = append(data, []byte("\n")...)
_, err = output.Write(data)
return err
2017-05-07 21:51:37 +02:00
}