mirror of https://github.com/vx3r/wg-gen-web.git
99 lines
2.2 KiB
Go
99 lines
2.2 KiB
Go
package core
|
|
|
|
import (
|
|
"gitlab.127-0-0-1.fr/vx3r/wg-gen-web/model"
|
|
"gitlab.127-0-0-1.fr/vx3r/wg-gen-web/storage"
|
|
"gitlab.127-0-0-1.fr/vx3r/wg-gen-web/template"
|
|
"gitlab.127-0-0-1.fr/vx3r/wg-gen-web/util"
|
|
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
|
"os"
|
|
"path/filepath"
|
|
"time"
|
|
)
|
|
|
|
// ReadServer object, create default one
|
|
func ReadServer() (*model.Server, error) {
|
|
if !util.FileExists(filepath.Join(os.Getenv("WG_CONF_DIR"), "server.json")) {
|
|
server := &model.Server{}
|
|
|
|
key, err := wgtypes.GeneratePrivateKey()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
server.PrivateKey = key.String()
|
|
server.PublicKey = key.PublicKey().String()
|
|
|
|
presharedKey, err := wgtypes.GenerateKey()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
server.PresharedKey = presharedKey.String()
|
|
|
|
server.Name = "Created with default values"
|
|
server.Endpoint = "wireguard.example.com:123"
|
|
server.ListenPort = 51820
|
|
server.Address = "fd9f:6666::10:6:6:1/112, 10.6.6.1/24"
|
|
server.Dns = "fd9f::10:0:0:2, 10.0.0.2"
|
|
server.PersistentKeepalive = 16
|
|
server.Created = time.Now().UTC()
|
|
server.Updated = server.Created
|
|
|
|
err = storage.Serialize("server.json", server)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
c, err := storage.Deserialize("server.json")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return c.(*model.Server), nil
|
|
}
|
|
|
|
// UpdateServer keep private values from existing one
|
|
func UpdateServer(server *model.Server) (*model.Server, error) {
|
|
current, err := storage.Deserialize("server.json")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
server.PrivateKey = current.(*model.Server).PrivateKey
|
|
server.PublicKey = current.(*model.Server).PublicKey
|
|
server.PresharedKey = current.(*model.Server).PresharedKey
|
|
server.Updated = time.Now().UTC()
|
|
|
|
err = storage.Serialize("server.json", server)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
v, err := storage.Deserialize("server.json")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
server = v.(*model.Server)
|
|
|
|
return server, UpdateServerConfigWg()
|
|
}
|
|
|
|
// UpdateServerConfigWg in wg format
|
|
func UpdateServerConfigWg() error {
|
|
clients, err := ReadClients()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
server, err := ReadServer()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = template.DumpServerWg(clients, server)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|