mirror of
https://github.com/gravitl/netmaker.git
synced 2025-10-23 17:03:17 +08:00
Began converting to rqlite
This commit is contained in:
afeiszli
@@ -1,137 +1,92 @@
|
||||
package serverctl
|
||||
|
||||
import (
|
||||
"log"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"io"
|
||||
"log"
|
||||
"os"
|
||||
"os/exec"
|
||||
"time"
|
||||
|
||||
"github.com/gravitl/netmaker/database"
|
||||
"github.com/gravitl/netmaker/functions"
|
||||
"github.com/gravitl/netmaker/models"
|
||||
"github.com/gravitl/netmaker/mongoconn"
|
||||
"github.com/gravitl/netmaker/servercfg"
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/mongo/options"
|
||||
"io"
|
||||
"time"
|
||||
"context"
|
||||
"errors"
|
||||
"os"
|
||||
"os/exec"
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/mongo/options"
|
||||
)
|
||||
|
||||
func CreateDefaultNetwork() (bool, error) {
|
||||
|
||||
log.Println("Creating default network...")
|
||||
|
||||
iscreated := false
|
||||
exists, err := functions.NetworkExists("default")
|
||||
|
||||
if exists || err != nil {
|
||||
log.Println("Default network already exists. Skipping...")
|
||||
return iscreated, err
|
||||
} else {
|
||||
|
||||
var network models.Network
|
||||
|
||||
network.NetID = "default"
|
||||
network.AddressRange = "10.10.10.0/24"
|
||||
network.DisplayName = "default"
|
||||
network.SetDefaults()
|
||||
network.SetNodesLastModified()
|
||||
network.SetNetworkLastModified()
|
||||
network.KeyUpdateTimeStamp = time.Now().Unix()
|
||||
priv := false
|
||||
network.IsLocal = &priv
|
||||
network.KeyUpdateTimeStamp = time.Now().Unix()
|
||||
allow := true
|
||||
network.AllowManualSignUp = &allow
|
||||
|
||||
collection := mongoconn.Client.Database("netmaker").Collection("networks")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||||
|
||||
// insert our network into the network table
|
||||
_, err = collection.InsertOne(ctx, network)
|
||||
defer cancel()
|
||||
|
||||
}
|
||||
if err == nil {
|
||||
iscreated = true
|
||||
}
|
||||
return iscreated, err
|
||||
|
||||
|
||||
}
|
||||
|
||||
func GetServerWGConf() (models.IntClient, error) {
|
||||
var server models.IntClient
|
||||
collection := mongoconn.Client.Database("netmaker").Collection("intclients")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||||
var server models.IntClient
|
||||
collection := mongoconn.Client.Database("netmaker").Collection("intclients")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||||
filter := bson.M{"network": "comms", "isserver": "yes"}
|
||||
err := collection.FindOne(ctx, filter, options.FindOne().SetProjection(bson.M{"_id": 0})).Decode(&server)
|
||||
err := collection.FindOne(ctx, filter, options.FindOne().SetProjection(bson.M{"_id": 0})).Decode(&server)
|
||||
defer cancel()
|
||||
|
||||
return server, err
|
||||
}
|
||||
|
||||
|
||||
func CreateCommsNetwork() (bool, error) {
|
||||
|
||||
iscreated := false
|
||||
exists, err := functions.NetworkExists("comms")
|
||||
iscreated := false
|
||||
exists, err := functions.NetworkExists("comms")
|
||||
|
||||
if exists || err != nil {
|
||||
log.Println("comms network already exists. Skipping...")
|
||||
return true, nil
|
||||
} else {
|
||||
if exists || err != nil {
|
||||
log.Println("comms network already exists. Skipping...")
|
||||
return true, err
|
||||
} else {
|
||||
var network models.Network
|
||||
|
||||
var network models.Network
|
||||
network.NetID = "comms"
|
||||
network.IsIPv6 = "no"
|
||||
network.IsIPv4 = "yes"
|
||||
network.IsGRPCHub = "yes"
|
||||
network.AddressRange = servercfg.GetGRPCWGAddressRange()
|
||||
network.DisplayName = "comms"
|
||||
network.SetDefaults()
|
||||
network.SetNodesLastModified()
|
||||
network.SetNetworkLastModified()
|
||||
network.KeyUpdateTimeStamp = time.Now().Unix()
|
||||
priv := false
|
||||
network.IsLocal = &priv
|
||||
network.KeyUpdateTimeStamp = time.Now().Unix()
|
||||
|
||||
network.NetID = "comms"
|
||||
network.IsIPv6 = "no"
|
||||
network.IsIPv4 = "yes"
|
||||
network.IsGRPCHub = "yes"
|
||||
network.AddressRange = servercfg.GetGRPCWGAddressRange()
|
||||
network.DisplayName = "comms"
|
||||
network.SetDefaults()
|
||||
network.SetNodesLastModified()
|
||||
network.SetNetworkLastModified()
|
||||
network.KeyUpdateTimeStamp = time.Now().Unix()
|
||||
priv := false
|
||||
network.IsLocal = &priv
|
||||
network.KeyUpdateTimeStamp = time.Now().Unix()
|
||||
|
||||
log.Println("Creating comms network...")
|
||||
|
||||
collection := mongoconn.Client.Database("netmaker").Collection("networks")
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||||
|
||||
// insert our network into the network table
|
||||
_, err = collection.InsertOne(ctx, network)
|
||||
defer cancel()
|
||||
|
||||
}
|
||||
if err == nil {
|
||||
iscreated = true
|
||||
}
|
||||
return iscreated, err
|
||||
log.Println("Creating comms network...")
|
||||
value, err := json.Marshal(network)
|
||||
if err != nil {
|
||||
return false, err
|
||||
}
|
||||
database.Insert(network.NetID, string(value), database.NETWORKS_TABLE_NAME)
|
||||
}
|
||||
if err == nil {
|
||||
iscreated = true
|
||||
}
|
||||
return iscreated, err
|
||||
}
|
||||
|
||||
func DownloadNetclient() error {
|
||||
/*
|
||||
// Get the data
|
||||
resp, err := http.Get("https://github.com/gravitl/netmaker/releases/download/latest/netclient")
|
||||
if err != nil {
|
||||
log.Println("could not download netclient")
|
||||
return err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
// Get the data
|
||||
resp, err := http.Get("https://github.com/gravitl/netmaker/releases/download/latest/netclient")
|
||||
if err != nil {
|
||||
log.Println("could not download netclient")
|
||||
return err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
// Create the file
|
||||
out, err := os.Create("/etc/netclient/netclient")
|
||||
*/
|
||||
if !FileExists("/etc/netclient/netclient") {
|
||||
// Create the file
|
||||
out, err := os.Create("/etc/netclient/netclient")
|
||||
*/
|
||||
if !FileExists("/etc/netclient/netclient") {
|
||||
_, err := copy("./netclient/netclient", "/etc/netclient/netclient")
|
||||
if err != nil {
|
||||
log.Println("could not create /etc/netclient")
|
||||
return err
|
||||
}
|
||||
if err != nil {
|
||||
log.Println("could not create /etc/netclient")
|
||||
return err
|
||||
}
|
||||
}
|
||||
//defer out.Close()
|
||||
|
||||
@@ -141,98 +96,97 @@ func DownloadNetclient() error {
|
||||
}
|
||||
|
||||
func FileExists(f string) bool {
|
||||
info, err := os.Stat(f)
|
||||
if os.IsNotExist(err) {
|
||||
return false
|
||||
}
|
||||
return !info.IsDir()
|
||||
info, err := os.Stat(f)
|
||||
if os.IsNotExist(err) {
|
||||
return false
|
||||
}
|
||||
return !info.IsDir()
|
||||
}
|
||||
|
||||
func copy(src, dst string) (int64, error) {
|
||||
sourceFileStat, err := os.Stat(src)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
sourceFileStat, err := os.Stat(src)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
if !sourceFileStat.Mode().IsRegular() {
|
||||
return 0, errors.New(src + " is not a regular file")
|
||||
}
|
||||
if !sourceFileStat.Mode().IsRegular() {
|
||||
return 0, errors.New(src + " is not a regular file")
|
||||
}
|
||||
|
||||
source, err := os.Open(src)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer source.Close()
|
||||
source, err := os.Open(src)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer source.Close()
|
||||
|
||||
destination, err := os.Create(dst)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer destination.Close()
|
||||
nBytes, err := io.Copy(destination, source)
|
||||
err = os.Chmod(dst, 0755)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
return nBytes, err
|
||||
destination, err := os.Create(dst)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
defer destination.Close()
|
||||
nBytes, err := io.Copy(destination, source)
|
||||
err = os.Chmod(dst, 0755)
|
||||
if err != nil {
|
||||
log.Println(err)
|
||||
}
|
||||
return nBytes, err
|
||||
}
|
||||
|
||||
func RemoveNetwork(network string) (bool, error) {
|
||||
_, err := os.Stat("/etc/netclient/netclient")
|
||||
if err != nil {
|
||||
log.Println("could not find /etc/netclient")
|
||||
if err != nil {
|
||||
log.Println("could not find /etc/netclient")
|
||||
return false, err
|
||||
}
|
||||
cmdoutput, err := exec.Command("/etc/netclient/netclient","leave","-n",network).Output()
|
||||
if err != nil {
|
||||
log.Println(string(cmdoutput))
|
||||
return false, err
|
||||
}
|
||||
log.Println("Server removed from network " + network)
|
||||
return true, err
|
||||
cmdoutput, err := exec.Command("/etc/netclient/netclient", "leave", "-n", network).Output()
|
||||
if err != nil {
|
||||
log.Println(string(cmdoutput))
|
||||
return false, err
|
||||
}
|
||||
log.Println("Server removed from network " + network)
|
||||
return true, err
|
||||
|
||||
}
|
||||
|
||||
func AddNetwork(network string) (bool, error) {
|
||||
pubip, err := servercfg.GetPublicIP()
|
||||
if err != nil {
|
||||
log.Println("could not get public IP.")
|
||||
return false, err
|
||||
}
|
||||
if err != nil {
|
||||
log.Println("could not get public IP.")
|
||||
return false, err
|
||||
}
|
||||
|
||||
_, err = os.Stat("/etc/netclient")
|
||||
if os.IsNotExist(err) {
|
||||
os.Mkdir("/etc/netclient", 744)
|
||||
} else if err != nil {
|
||||
log.Println("could not find or create /etc/netclient")
|
||||
return false, err
|
||||
if os.IsNotExist(err) {
|
||||
os.Mkdir("/etc/netclient", 744)
|
||||
} else if err != nil {
|
||||
log.Println("could not find or create /etc/netclient")
|
||||
return false, err
|
||||
}
|
||||
token, err := functions.CreateServerToken(network)
|
||||
if err != nil {
|
||||
log.Println("could not create server token for " + network)
|
||||
log.Println("could not create server token for " + network)
|
||||
return false, err
|
||||
}
|
||||
_, err = os.Stat("/etc/netclient/netclient")
|
||||
}
|
||||
_, err = os.Stat("/etc/netclient/netclient")
|
||||
if os.IsNotExist(err) {
|
||||
err = DownloadNetclient()
|
||||
if err != nil {
|
||||
return false, err
|
||||
}
|
||||
}
|
||||
err = os.Chmod("/etc/netclient/netclient", 0755)
|
||||
if err != nil {
|
||||
log.Println("could not change netclient directory permissions")
|
||||
return false, err
|
||||
}
|
||||
log.Println("executing network join: " + "/etc/netclient/netclient "+"join "+"-t "+token+" -name "+"netmaker"+" -endpoint "+pubip)
|
||||
out, err := exec.Command("/etc/netclient/netclient","join","-t",token,"-name","netmaker","-endpoint",pubip).Output()
|
||||
if string(out) != "" {
|
||||
log.Println(string(out))
|
||||
err = os.Chmod("/etc/netclient/netclient", 0755)
|
||||
if err != nil {
|
||||
log.Println("could not change netclient directory permissions")
|
||||
return false, err
|
||||
}
|
||||
log.Println("executing network join: " + "/etc/netclient/netclient " + "join " + "-t " + token + " -name " + "netmaker" + " -endpoint " + pubip)
|
||||
out, err := exec.Command("/etc/netclient/netclient", "join", "-t", token, "-name", "netmaker", "-endpoint", pubip).Output()
|
||||
if string(out) != "" {
|
||||
log.Println(string(out))
|
||||
}
|
||||
if err != nil {
|
||||
return false, errors.New(string(out) + err.Error())
|
||||
}
|
||||
return false, errors.New(string(out) + err.Error())
|
||||
}
|
||||
log.Println("Server added to network " + network)
|
||||
return true, err
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user