Implements Wake On Lan and management features (#164)

store/jsondb/jsondb.go

@@ -38,6 +38,7 @@ func New(dbPath string) (*JsonDB, error) {
 func (o *JsonDB) Init() error {
 	var clientPath string = path.Join(o.dbPath, "clients")
 	var serverPath string = path.Join(o.dbPath, "server")
+	var wakeOnLanHostsPath string = path.Join(o.dbPath, "wake_on_lan_hosts")
 	var serverInterfacePath string = path.Join(serverPath, "interfaces.json")
 	var serverKeyPairPath string = path.Join(serverPath, "keypair.json")
 	var globalSettingPath string = path.Join(serverPath, "global_settings.json")
@@ -49,6 +50,9 @@ func (o *JsonDB) Init() error {
 	if _, err := os.Stat(serverPath); os.IsNotExist(err) {
 		os.MkdirAll(serverPath, os.ModePerm)
 	}
+	if _, err := os.Stat(wakeOnLanHostsPath); os.IsNotExist(err) {
+		os.MkdirAll(wakeOnLanHostsPath, os.ModePerm)
+	}
 
 	// server's interface
 	if _, err := os.Stat(serverInterfacePath); os.IsNotExist(err) {

store/jsondb/jsondb_wake_on_lan.go

@@ -0,0 +1,78 @@
+package jsondb
+
+import (
+	"encoding/json"
+	"fmt"
+	"github.com/ngoduykhanh/wireguard-ui/model"
+)
+
+func (o *JsonDB) GetWakeOnLanHosts() ([]model.WakeOnLanHost, error) {
+	var hosts []model.WakeOnLanHost
+
+	// read all client json file in "hosts" directory
+	records, err := o.conn.ReadAll(model.WakeOnLanHostCollectionName)
+	if err != nil {
+		return hosts, err
+	}
+
+	// build the ClientData list
+	for _, f := range records {
+		host := model.WakeOnLanHost{}
+
+		// get client info
+		if err := json.Unmarshal([]byte(f), &host); err != nil {
+			return hosts, fmt.Errorf("cannot decode client json structure: %v", err)
+		}
+
+		// create the list of hosts and their qrcode data
+		hosts = append(hosts, host)
+	}
+
+	return hosts, nil
+}
+
+func (o *JsonDB) GetWakeOnLanHost(macAddress string) (*model.WakeOnLanHost, error) {
+	host := &model.WakeOnLanHost{
+		MacAddress: macAddress,
+	}
+	resourceName, err := host.ResolveResourceName()
+	if err != nil {
+		return nil, err
+	}
+
+	err = o.conn.Read(model.WakeOnLanHostCollectionName, resourceName, host)
+	if err != nil {
+		host = nil
+	}
+	return host, err
+}
+
+func (o *JsonDB) DeleteWakeOnHostLanHost(macAddress string) error {
+	host := &model.WakeOnLanHost{
+		MacAddress: macAddress,
+	}
+	resourceName, err := host.ResolveResourceName()
+	if err != nil {
+		return err
+	}
+
+	return o.conn.Delete(model.WakeOnLanHostCollectionName, resourceName)
+}
+
+func (o *JsonDB) SaveWakeOnLanHost(host model.WakeOnLanHost) error {
+	resourceName, err := host.ResolveResourceName()
+	if err != nil {
+		return err
+	}
+
+	return o.conn.Write(model.WakeOnLanHostCollectionName, resourceName, host)
+}
+
+func (o *JsonDB) DeleteWakeOnHost(host model.WakeOnLanHost) error {
+	resourceName, err := host.ResolveResourceName()
+	if err != nil {
+		return err
+	}
+
+	return o.conn.Delete(model.WakeOnLanHostCollectionName, resourceName)
+}