146 lines
4.1 KiB
Go
146 lines
4.1 KiB
Go
package engine
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"golang.org/x/time/rate"
|
|
|
|
"github.com/anacrolix/dht"
|
|
|
|
"github.com/anacrolix/torrent"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
//FullClientSettings contains all of the settings for our entire application
|
|
type FullClientSettings struct {
|
|
LoggingLevel logrus.Level
|
|
HTTPAddr string
|
|
Version int
|
|
TorrentConfig torrent.Config
|
|
TFileUploadFolder string
|
|
}
|
|
|
|
//default is called if there is a parsing error
|
|
func defaultConfig() FullClientSettings {
|
|
var Config FullClientSettings
|
|
Config.Version = 1.0
|
|
Config.LoggingLevel = 3 //Warn level
|
|
Config.TorrentConfig.DataDir = "downloads" //the full OR relative path of the default download directory for torrents
|
|
Config.TFileUploadFolder = "uploadedTorrents"
|
|
Config.TorrentConfig.Seed = true
|
|
Config.HTTPAddr = ":8000"
|
|
|
|
Config.TorrentConfig.DHTConfig = dht.ServerConfig{
|
|
StartingNodes: dht.GlobalBootstrapAddrs,
|
|
}
|
|
|
|
return Config
|
|
}
|
|
|
|
func dhtServerSettings(dhtConfig dht.ServerConfig) dht.ServerConfig {
|
|
|
|
viper.UnmarshalKey("DHTConfig", &dhtConfig)
|
|
fmt.Println("dhtconfig", dhtConfig)
|
|
|
|
return dhtConfig
|
|
}
|
|
|
|
//FullClientSettingsNew creates a new set of setting from config.toml
|
|
func FullClientSettingsNew() FullClientSettings {
|
|
viper.SetConfigName("config")
|
|
viper.AddConfigPath("./")
|
|
err := viper.ReadInConfig()
|
|
if err != nil {
|
|
fmt.Println("Error reading in config, using defaults", err)
|
|
FullClientSettings := defaultConfig()
|
|
return FullClientSettings
|
|
}
|
|
|
|
var httpAddr string
|
|
|
|
httpAddrIP := viper.GetString("serverConfig.ServerAddr")
|
|
httpAddrPort := viper.GetString("serverConfig.ServerPort")
|
|
httpAddr = httpAddrIP + httpAddrPort
|
|
|
|
fmt.Println("HttpAddr", httpAddr)
|
|
|
|
dataDir := viper.GetString("torrentClientConfig.DownloadDir")
|
|
listenAddr := viper.GetString("torrentClientConfig.ListenAddr")
|
|
disablePex := viper.GetBool("torrentClientConfig.DisablePEX")
|
|
noDHT := viper.GetBool("torrentClientConfig.NoDHT")
|
|
noUpload := viper.GetBool("torrentClientConfig.NoUpload")
|
|
seed := viper.GetBool("torrentClientConfig.Seed")
|
|
peerID := viper.GetString("torrentClientConfig.PeerID")
|
|
disableUTP := viper.GetBool("torrentClientConfig.DisableUTP")
|
|
disableTCP := viper.GetBool("torrentClientConfig.DisableTCP")
|
|
disableIPv6 := viper.GetBool("torrentClientConfig.DisableIPv6")
|
|
debug := viper.GetBool("torrentClientConfig.Debug")
|
|
logLevelString := viper.GetString("serverConfig.LogLevel")
|
|
var logLevel logrus.Level
|
|
switch logLevelString { //Options = Debug 5, Info 4, Warn 3, Error 2, Fatal 1, Panic 0
|
|
case "Panic":
|
|
logLevel = 0
|
|
case "Fatal":
|
|
logLevel = 1
|
|
case "Error":
|
|
logLevel = 2
|
|
case "Warn":
|
|
logLevel = 3
|
|
case "Info":
|
|
logLevel = 4
|
|
case "Debug":
|
|
logLevel = 5
|
|
default:
|
|
logLevel = 3
|
|
|
|
}
|
|
|
|
dhtServerConfig := dht.ServerConfig{
|
|
StartingNodes: dht.GlobalBootstrapAddrs,
|
|
}
|
|
if viper.IsSet("DHTConfig") {
|
|
fmt.Println("Reading in custom DHT config")
|
|
dhtServerConfig = dhtServerSettings(dhtServerConfig)
|
|
}
|
|
|
|
uploadRateLimiter := new(rate.Limiter)
|
|
viper.UnmarshalKey("UploadRateLimiter", &uploadRateLimiter)
|
|
|
|
downloadRateLimiter := new(rate.Limiter)
|
|
viper.UnmarshalKey("DownloadRateLimiter", &downloadRateLimiter)
|
|
|
|
rreferNoEncryption := viper.GetBool("EncryptionPolicy.PreferNoEncryption")
|
|
fmt.Println("Encryption", rreferNoEncryption)
|
|
|
|
encryptionPolicy := torrent.EncryptionPolicy{
|
|
DisableEncryption: viper.GetBool("EncryptionPolicy.DisableEncryption"),
|
|
ForceEncryption: viper.GetBool("EncryptionPolicy.ForceEncryption"),
|
|
PreferNoEncryption: viper.GetBool("EncryptionPolicy.PreferNoEncryption"),
|
|
}
|
|
|
|
tConfig := torrent.Config{
|
|
DataDir: dataDir,
|
|
ListenAddr: listenAddr,
|
|
DisablePEX: disablePex,
|
|
NoDHT: noDHT,
|
|
DHTConfig: dhtServerConfig,
|
|
NoUpload: noUpload,
|
|
Seed: seed,
|
|
//UploadRateLimiter: uploadRateLimiter,
|
|
//DownloadRateLimiter: downloadRateLimiter,
|
|
PeerID: peerID,
|
|
DisableUTP: disableUTP,
|
|
DisableTCP: disableTCP,
|
|
DisableIPv6: disableIPv6,
|
|
Debug: debug,
|
|
EncryptionPolicy: encryptionPolicy,
|
|
}
|
|
|
|
Config := FullClientSettings{LoggingLevel: logLevel, HTTPAddr: httpAddr, TorrentConfig: tConfig, TFileUploadFolder: "uploadedTorrents"}
|
|
|
|
return Config
|
|
|
|
}
|