forked from IceWhaleTech/CasaOS
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
13 changed files
with
313 additions
and
3 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -39,4 +39,4 @@ dist | |
CasaOS | ||
|
||
# System Files | ||
.DS_Store | ||
.DS_Store |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,4 +1,14 @@ | ||
#!/bin/bash | ||
### | ||
# @Author: LinkLeong [email protected] | ||
# @Date: 2022-11-15 15:51:44 | ||
# @LastEditors: LinkLeong | ||
# @LastEditTime: 2022-11-15 15:53:37 | ||
# @FilePath: /CasaOS/build/sysroot/usr/share/casaos/cleanup/script.d/03-cleanup-casaos.sh | ||
# @Description: | ||
# @Website: https://www.casaos.io | ||
# Copyright (c) 2022 by icewhale, All Rights Reserved. | ||
### | ||
|
||
set -e | ||
|
||
|
@@ -44,4 +54,3 @@ readonly SETUP_SCRIPT_FILEPATH="${SETUP_SCRIPT_DIRECTORY}/${SETUP_SCRIPT_FILENAM | |
|
||
echo "🟩 Running ${SETUP_SCRIPT_FILENAME}..." | ||
$SHELL "${SETUP_SCRIPT_FILEPATH}" "${BUILD_PATH}" | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,89 @@ | ||
package common | ||
|
||
import ( | ||
"bytes" | ||
"encoding/json" | ||
"errors" | ||
"fmt" | ||
"net/http" | ||
"os" | ||
"path/filepath" | ||
"strings" | ||
) | ||
|
||
const ( | ||
CasaOSURLFilename = "casaos.url" | ||
APICasaOSNotify = "/v1/notify" | ||
) | ||
|
||
type NotifyService interface { | ||
SendNotify(path string, message map[string]interface{}) error | ||
SendSystemStatusNotify(message map[string]interface{}) error | ||
} | ||
type notifyService struct { | ||
address string | ||
} | ||
|
||
func (n *notifyService) SendNotify(path string, message map[string]interface{}) error { | ||
|
||
url := strings.TrimSuffix(n.address, "/") + APICasaOSNotify + "/" + path | ||
body, err := json.Marshal(message) | ||
if err != nil { | ||
return err | ||
} | ||
response, err := http.Post(url, "application/json", bytes.NewBuffer(body)) | ||
if err != nil { | ||
return err | ||
} | ||
|
||
if response.StatusCode != http.StatusOK { | ||
return errors.New("failed to send notify (status code: " + fmt.Sprint(response.StatusCode) + ")") | ||
} | ||
return nil | ||
|
||
} | ||
|
||
// disk: "sys_disk":{"size":56866869248,"avail":5855485952,"health":true,"used":48099700736} | ||
// usb: "sys_usb":[{"name": "sdc","size": 7747397632,"model": "DataTraveler_2.0","avail": 7714418688,"children": null}] | ||
func (n *notifyService) SendSystemStatusNotify(message map[string]interface{}) error { | ||
|
||
url := strings.TrimSuffix(n.address, "/") + APICasaOSNotify + "/system_status" | ||
fmt.Println(url) | ||
body, err := json.Marshal(message) | ||
if err != nil { | ||
return err | ||
} | ||
response, err := http.Post(url, "application/json", bytes.NewBuffer(body)) | ||
if err != nil { | ||
return err | ||
} | ||
|
||
if response.StatusCode != http.StatusOK { | ||
return errors.New("failed to send notify (status code: " + fmt.Sprint(response.StatusCode) + ")") | ||
} | ||
return nil | ||
|
||
} | ||
func NewNotifyService(runtimePath string) (NotifyService, error) { | ||
casaosAddressFile := filepath.Join(runtimePath, CasaOSURLFilename) | ||
|
||
buf, err := os.ReadFile(casaosAddressFile) | ||
if err != nil { | ||
return nil, err | ||
} | ||
|
||
address := string(buf) | ||
|
||
response, err := http.Get(address + "/ping") | ||
if err != nil { | ||
return nil, err | ||
} | ||
|
||
if response.StatusCode != 200 { | ||
return nil, errors.New("failed to ping casaos service") | ||
} | ||
|
||
return ¬ifyService{ | ||
address: address, | ||
}, nil | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,29 @@ | ||
package common | ||
|
||
import "testing" | ||
|
||
func TestSendNotify(t *testing.T) { | ||
notify, err := NewNotifyService("/var/run/casaos") | ||
if err != nil { | ||
t.Fatal(err) | ||
} | ||
err = notify.SendNotify("test", map[string]interface{}{ | ||
"test": "test", | ||
}) | ||
if err != nil { | ||
t.Fatal(err) | ||
} | ||
} | ||
|
||
func TestSendSystemStatusNotify(t *testing.T) { | ||
notify, err := NewNotifyService("/var/run/casaos") | ||
if err != nil { | ||
t.Fatal(err) | ||
} | ||
err = notify.SendSystemStatusNotify(map[string]interface{}{ | ||
"sys_usb": `[{"name": "sdc","size": 7747397632,"model": "DataTraveler_2.0","avail": 7714418688,"children": null}]`, | ||
}) | ||
if err != nil { | ||
t.Fatal(err) | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,78 @@ | ||
package common | ||
|
||
import ( | ||
"bytes" | ||
"encoding/json" | ||
"errors" | ||
"fmt" | ||
"net/http" | ||
"os" | ||
"path/filepath" | ||
"strings" | ||
) | ||
|
||
const ( | ||
APICasaOSShare = "/v1/samba/shares" | ||
) | ||
|
||
type ShareService interface { | ||
DeleteShare(id string) error | ||
} | ||
type shareService struct { | ||
address string | ||
} | ||
|
||
func (n *shareService) DeleteShare(id string) error { | ||
url := strings.TrimSuffix(n.address, "/") + APICasaOSShare + "/" + id | ||
fmt.Println(url) | ||
message := "{}" | ||
body, err := json.Marshal(message) | ||
if err != nil { | ||
return err | ||
} | ||
|
||
client := &http.Client{} | ||
|
||
// Create request | ||
req, err := http.NewRequest("DELETE", url, bytes.NewBuffer(body)) | ||
if err != nil { | ||
return err | ||
} | ||
|
||
// Fetch Request | ||
response, err := client.Do(req) | ||
if err != nil { | ||
return err | ||
} | ||
defer response.Body.Close() | ||
|
||
if response.StatusCode != http.StatusOK { | ||
return errors.New("failed to send share (status code: " + fmt.Sprint(response.StatusCode) + ")") | ||
} | ||
return nil | ||
|
||
} | ||
|
||
func NewShareService(runtimePath string) (ShareService, error) { | ||
casaosAddressFile := filepath.Join(runtimePath, CasaOSURLFilename) | ||
|
||
buf, err := os.ReadFile(casaosAddressFile) | ||
if err != nil { | ||
return nil, err | ||
} | ||
|
||
address := string(buf) | ||
|
||
response, err := http.Get(address + "/ping") | ||
if err != nil { | ||
return nil, err | ||
} | ||
|
||
if response.StatusCode != 200 { | ||
return nil, errors.New("failed to ping casaos service") | ||
} | ||
|
||
return &shareService{ | ||
address: address, | ||
}, nil | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,14 @@ | ||
package common | ||
|
||
import "testing" | ||
|
||
func TestDeleteShare(t *testing.T) { | ||
share, err := NewShareService("/var/run/casaos") | ||
if err != nil { | ||
t.Fatal(err) | ||
} | ||
err = share.DeleteShare("1") | ||
if err != nil { | ||
t.Fatal(err) | ||
} | ||
} |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,21 +1,65 @@ | ||
/* | ||
* @Author: LinkLeong [email protected] | ||
* @Date: 2022-11-15 15:51:44 | ||
* @LastEditors: LinkLeong | ||
* @LastEditTime: 2022-11-15 15:55:16 | ||
* @FilePath: /CasaOS/route/init.go | ||
* @Description: | ||
* @Website: https://www.casaos.io | ||
* Copyright (c) 2022 by icewhale, All Rights Reserved. | ||
*/ | ||
package route | ||
|
||
import ( | ||
"encoding/json" | ||
"fmt" | ||
"os" | ||
"strings" | ||
"time" | ||
|
||
"github.com/IceWhaleTech/CasaOS/model" | ||
"github.com/IceWhaleTech/CasaOS/pkg/config" | ||
"github.com/IceWhaleTech/CasaOS/pkg/samba" | ||
"github.com/IceWhaleTech/CasaOS/pkg/utils/encryption" | ||
"github.com/IceWhaleTech/CasaOS/pkg/utils/file" | ||
"github.com/IceWhaleTech/CasaOS/pkg/utils/loger" | ||
"github.com/IceWhaleTech/CasaOS/service" | ||
"github.com/IceWhaleTech/CasaOS/types" | ||
"go.uber.org/zap" | ||
) | ||
|
||
func InitFunction() { | ||
go InitNetworkMount() | ||
go InitInfo() | ||
} | ||
|
||
func InitInfo() { | ||
mb := model.BaseInfo{} | ||
if file.Exists(config.AppInfo.DBPath + "/baseinfo.conf") { | ||
err := json.Unmarshal(file.ReadFullFile(config.AppInfo.DBPath+"/baseinfo.conf"), &mb) | ||
if err != nil { | ||
loger.Error("baseinfo.conf", zap.String("error", err.Error())) | ||
} | ||
} | ||
if file.Exists("/etc/CHANNEL") { | ||
channel := file.ReadFullFile("/etc/CHANNEL") | ||
mb.Channel = string(channel) | ||
} | ||
mac, err := service.MyService.System().GetMacAddress() | ||
if err != nil { | ||
loger.Error("GetMacAddress", zap.String("error", err.Error())) | ||
} | ||
mb.Hash = encryption.GetMD5ByStr(mac) | ||
mb.Version = types.CURRENTVERSION | ||
os.Remove(config.AppInfo.DBPath + "/baseinfo.conf") | ||
by, err := json.Marshal(mb) | ||
if err != nil { | ||
loger.Error("init info err", zap.Any("err", err)) | ||
return | ||
} | ||
file.WriteToFullPath(by, config.AppInfo.DBPath+"/baseinfo.conf", 0o666) | ||
} | ||
|
||
func InitNetworkMount() { | ||
time.Sleep(time.Second * 10) | ||
connections := service.MyService.Connections().GetConnectionsList() | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,3 +1,13 @@ | ||
/* | ||
* @Author: LinkLeong [email protected] | ||
* @Date: 2022-11-15 15:51:44 | ||
* @LastEditors: LinkLeong | ||
* @LastEditTime: 2022-11-15 15:56:03 | ||
* @FilePath: /CasaOS/types/system.go | ||
* @Description: | ||
* @Website: https://www.casaos.io | ||
* Copyright (c) 2022 by icewhale, All Rights Reserved. | ||
*/ | ||
/*@Author: LinkLeong [email protected] | ||
*@Date: 2022-02-17 18:53:22 | ||
*@LastEditors: LinkLeong | ||
|