130 lines
4.3 KiB
Go
130 lines
4.3 KiB
Go
package generate
|
|
|
|
import (
|
|
"github.com/Akilan1999/p2p-rendering-computation/config"
|
|
"os"
|
|
)
|
|
|
|
var (
|
|
defaults = map[string]interface{}{}
|
|
configPaths []string
|
|
defaultEnvName = "P2PRC"
|
|
)
|
|
|
|
// GetPathP2PRC Getting P2PRC Directory from environment variable
|
|
func GetPathP2PRC() (string, error) {
|
|
curDir := os.Getenv(defaultEnvName)
|
|
return curDir + "/", nil
|
|
}
|
|
|
|
// SetEnvName Sets the environment name
|
|
// This is to ensure that the Path of your project is detected from
|
|
// your environment variable
|
|
// This is useful when extending the use case of P2PRC
|
|
func SetEnvName(EnvName string) error {
|
|
if EnvName != "" {
|
|
defaultEnvName = EnvName
|
|
}
|
|
// Handling error to be implemented only if needed
|
|
return nil
|
|
}
|
|
|
|
// GetCurrentPath Getting P2PRC Directory from environment variable
|
|
func GetCurrentPath() (string, error) {
|
|
curDir := os.Getenv("PWD")
|
|
return curDir + "/", nil
|
|
}
|
|
|
|
// SetDefaults This function to be called only during a
|
|
// make install
|
|
func SetDefaults(envName string, forceDefault bool, ConfigUpdate ...*config.Config) (*config.Config, error) {
|
|
//Setting current directory to default path
|
|
defaultPath, err := GetCurrentPath()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// Set Env name
|
|
err = config.SetEnvName(envName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
////Creates ip_table.json in the json directory
|
|
//err = Copy("p2p/ip_table.json", "p2p/iptable/ip_table.json")
|
|
//if err != nil {
|
|
// return err
|
|
//}
|
|
//
|
|
////Creates a copy of trackcontainers.json in the appropriate directory
|
|
//err = Copy("client/trackcontainers.json", "client/trackcontainers/trackcontainers.json")
|
|
//if err != nil {
|
|
// return err
|
|
//}
|
|
//
|
|
////Creates a copy of trackcontainers.json in the appropriate directory
|
|
//err = Copy("client/grouptrackcontainers.json", "client/trackcontainers/grouptrackcontainers.json")
|
|
//if err != nil {
|
|
// return err
|
|
//}
|
|
|
|
if len(ConfigUpdate) == 0 {
|
|
//Setting default paths for the config file
|
|
defaults["IPTable"] = defaultPath + "p2p/iptable/ip_table.json"
|
|
defaults["DefaultDockerFile"] = defaultPath + "server/docker/containers/docker-ubuntu-sshd/"
|
|
defaults["DockerContainers"] = defaultPath + "server/docker/containers/"
|
|
defaults["SpeedTestFile"] = defaultPath + "p2p/50.bin"
|
|
defaults["IPV6Address"] = ""
|
|
defaults["PluginPath"] = defaultPath + "plugin/deploy"
|
|
defaults["TrackContainersPath"] = defaultPath + "client/trackcontainers/trackcontainers.json"
|
|
defaults["GroupTrackContainersPath"] = defaultPath + "client/trackcontainers/grouptrackcontainers.json"
|
|
defaults["ServerPort"] = "8088"
|
|
defaults["FRPServerPort"] = "0"
|
|
defaults["BehindNAT"] = "True"
|
|
// Random name generator
|
|
hostname, err := os.Hostname()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
defaults["MachineName"] = hostname
|
|
} else {
|
|
defaults["IPTable"] = ConfigUpdate[0].IPTable
|
|
defaults["DefaultDockerFile"] = ConfigUpdate[0].DefaultDockerFile
|
|
defaults["DockerContainers"] = ConfigUpdate[0].DockerContainers
|
|
defaults["SpeedTestFile"] = ConfigUpdate[0].SpeedTestFile
|
|
defaults["IPV6Address"] = ConfigUpdate[0].IPV6Address
|
|
defaults["PluginPath"] = ConfigUpdate[0].PluginPath
|
|
defaults["TrackContainersPath"] = ConfigUpdate[0].TrackContainersPath
|
|
defaults["GroupTrackContainersPath"] = ConfigUpdate[0].GroupTrackContainersPath
|
|
defaults["ServerPort"] = ConfigUpdate[0].ServerPort
|
|
defaults["FRPServerPort"] = ConfigUpdate[0].FRPServerPort
|
|
defaults["BehindNAT"] = ConfigUpdate[0].BehindNAT
|
|
}
|
|
|
|
//defaults["NetworkInterface"] = "wlp0s20f3"
|
|
//defaults["NetworkInterfaceIPV6Index"] = "2"
|
|
|
|
//Paths to search for config file
|
|
configPaths = append(configPaths, defaultPath)
|
|
|
|
if fileExists(defaultPath + "config.json") && forceDefault {
|
|
err := os.Remove(defaultPath + "config.json")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
//Calling configuration file
|
|
Config, err := config.ConfigInit(defaults, envName)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = GenerateFiles()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return Config, nil
|
|
}
|