6543/wg-gen-web
0
0
Fork 0
mirror of https://github.com/vx3r/wg-gen-web.git synced 2025-09-11 12:24:27 +00:00
Code Issues Releases Wiki Activity

go fmt / badges

Browse Source
This commit is contained in:
vx3r
2020-01-30 16:12:26 +09:00
parent 024d2b4ebb
commit 9b6d610a9c
9 changed files with 85 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
util/tpl.go
View File

@ -40,6 +40,7 @@ AllowedIPs = {{.Address}}
{{end}}`
)
// dump client wg config with go template
func DumpClient(client *model.Client, server *model.Server) (bytes.Buffer, error) {
var tplBuff bytes.Buffer
@ -57,6 +58,7 @@ func DumpClient(client *model.Client, server *model.Server) (bytes.Buffer, error
})
}
// dump server wg config with go template
func DumpServerWg(clients []*model.Client, server *model.Server) (bytes.Buffer, error) {
var tplBuff bytes.Buffer
@ -66,17 +68,17 @@ func DumpServerWg(clients []*model.Client, server *model.Server) (bytes.Buffer,
}
return dump(t, struct {
Clients []*model.Client
Server *model.Server
Clients []*model.Client
Server *model.Server
ServerAdresses []string
}{
ServerAdresses: strings.Split(server.Address, ","),
Clients: clients,
Server: server,
Clients: clients,
Server: server,
})
}
func dump(tpl *template.Template , data interface{}) (bytes.Buffer, error) {
func dump(tpl *template.Template, data interface{}) (bytes.Buffer, error) {
var tplBuff bytes.Buffer
err := tpl.Execute(&tplBuff, data)

9
util/util.go
View File

@ -8,6 +8,7 @@ import (
"strings"
)
// read file content
func ReadFile(path string) (bytes []byte, err error) {
bytes, err = ioutil.ReadFile(path)
if err != nil {
@ -17,6 +18,7 @@ func ReadFile(path string) (bytes []byte, err error) {
return bytes, nil
}
// write content to file
func WriteFile(path string, bytes []byte) (err error) {
err = ioutil.WriteFile(path, bytes, 0644)
if err != nil {
@ -26,6 +28,7 @@ func WriteFile(path string, bytes []byte) (err error) {
return nil
}
// check if file exists
func FileExists(name string) bool {
info, err := os.Stat(name)
if os.IsNotExist(err) {
@ -34,6 +37,7 @@ func FileExists(name string) bool {
return !info.IsDir()
}
// check if directory exists
func DirectoryExists(name string) bool {
info, err := os.Stat(name)
if os.IsNotExist(err) {
@ -42,6 +46,7 @@ func DirectoryExists(name string) bool {
return info.IsDir()
}
// search for an available in cidr against a list of reserved ips
func GetAvailableIp(cidr string, reserved []string) (string, error) {
addresses, err := GetAllAddressesFromCidr(cidr)
if err != nil {
@ -64,6 +69,7 @@ func GetAvailableIp(cidr string, reserved []string) (string, error) {
return "", errors.New("no more available address from cidr")
}
// get all ip addresses from cidr
func GetAllAddressesFromCidr(cidr string) ([]string, error) {
ip, ipnet, err := net.ParseCIDR(cidr)
if err != nil {
@ -88,6 +94,7 @@ func inc(ip net.IP) {
}
}
// check if given ip is IPv6
func IsIPv6(address string) bool {
return strings.Count(address, ":") >= 2
}
}