From 71d04509ca7f65f3e5243dc7cb2cb586bce1ba96 Mon Sep 17 00:00:00 2001
From: Krxk <1246924370@qq.com>
Date: Fri, 3 Nov 2023 19:35:21 +0800
Subject: [PATCH] Version 0.1.1
---
Output.txt | 1 +
SSHCommand/RunSSH.go | 216 +++++++++++++++++++
ScutLog.log | 0
config.json | 5 +
frontend/src/components/Demo.vue | 2 +-
frontend/src/components/Login.vue | 15 +-
frontend/src/views/ACInfo.vue | 5 +-
frontend/src/views/SyncTime.vue | 4 +-
frontend/src/views/WiredLogin.vue | 4 +-
frontend/src/views/WirelessLogin.vue | 4 +-
frontend/wailsjs/go/SSHCommand/SSHClass.d.ts | 29 +++
frontend/wailsjs/go/SSHCommand/SSHClass.js | 55 +++++
go.mod | 8 +-
go.sum | 1 +
main.go | 28 ++-
15 files changed, 363 insertions(+), 14 deletions(-)
create mode 100644 Output.txt
create mode 100644 SSHCommand/RunSSH.go
create mode 100644 ScutLog.log
create mode 100644 config.json
create mode 100644 frontend/wailsjs/go/SSHCommand/SSHClass.d.ts
create mode 100644 frontend/wailsjs/go/SSHCommand/SSHClass.js
diff --git a/Output.txt b/Output.txt
new file mode 100644
index 0000000..58104dd
--- /dev/null
+++ b/Output.txt
@@ -0,0 +1 @@
+Startup
\ No newline at end of file
diff --git a/SSHCommand/RunSSH.go b/SSHCommand/RunSSH.go
new file mode 100644
index 0000000..858b7a9
--- /dev/null
+++ b/SSHCommand/RunSSH.go
@@ -0,0 +1,216 @@
+package SSHCommand
+
+import (
+ "encoding/json"
+ "fmt"
+ "github.com/labstack/gommon/log"
+ "golang.org/x/crypto/ssh"
+ "io"
+ "os"
+)
+
+type ScutInfo struct {
+ Username, Password string
+ AcIP, AcName string
+}
+
+type SSHClass struct {
+ client *ssh.Client
+ session *ssh.Session
+ err error
+ status bool // 记录上一操作是否成功
+ out io.Writer // 用于保存输出SSH运行结果
+ ScutInfo // 保存信息
+}
+
+func (s *SSHClass) SetOut(out io.Writer) {
+ // 先于 NewSession 被调用
+ s.out = out
+}
+
+func (s *SSHClass) NewSession() {
+ // 每条命令都需要一个新 Session
+ if s.status != true {
+ log.Errorf("Dial Fails: %s", s.err.Error())
+ return
+ }
+ s.session, s.err = s.client.NewSession()
+ s.Assert()
+
+ // 设置输出
+ s.session.Stdout = s.out
+}
+
+func (s *SSHClass) Close() {
+ if s.session != nil {
+ s.session.Close()
+ s.session = nil
+ s.status = true
+ }
+}
+
+func (s *SSHClass) Assert() {
+ if s.err != nil {
+ log.Error(s.err.Error())
+ s.status = false
+ return
+ }
+ s.status = true
+}
+
+func (s *SSHClass) RunCommand(cmd string) {
+ s.NewSession()
+ defer s.Close()
+
+ if s.status != true {
+ log.Errorf("New Session Fails: %s", s.err.Error())
+ return
+ }
+ err := s.session.Run(cmd)
+ if err != nil {
+ log.Error(err.Error())
+ }
+}
+
+type accountInfo struct {
+ Username, Password string
+ AcIP, AcName string
+}
+
+func (s *SSHClass) preLoadInfo() {
+ // 遇加载信息
+ infoFile, err := os.ReadFile("./Account.json")
+ if err != nil {
+ log.Info(err.Error())
+ return
+ }
+ accInfo := new(accountInfo)
+ json.Unmarshal(infoFile, accInfo)
+
+ s.Username = accInfo.Username
+ s.Password = accInfo.Password
+ s.AcIP = accInfo.AcIP
+ s.AcName = accInfo.AcName
+}
+
+func (s *SSHClass) saveInfo() {
+ infoFile, err := os.Create("./Account.json")
+ if err != nil {
+ log.Error(err.Error())
+ }
+
+ accInfo := new(accountInfo)
+ accInfo.Username = s.Username
+ accInfo.Password = s.Password
+ accInfo.AcIP = s.AcIP
+ accInfo.AcName = s.AcName
+
+ b, err := json.Marshal(accInfo)
+ if err != nil {
+ log.Error(err.Error())
+ }
+ infoFile.Write(b)
+ infoFile.Close()
+}
+
+func (s *SSHClass) RunSetScutInfo(username, password string) {
+ log.Info("RunSetScutInfo is called")
+ s.preLoadInfo()
+
+ s.Username = username
+ s.Password = password
+ s.HelpSetInfo()
+}
+
+func (s *SSHClass) RunSetAcInfo(acIP, acName string) {
+ log.Info("RunSetAcInfo is called")
+ s.preLoadInfo()
+
+ s.AcIP = acIP
+ s.AcName = acName
+ s.HelpSetInfo()
+}
+
+func (s *SSHClass) RunWiredLogin() {
+ cmd := fmt.Sprintf("chmod +x scutlogin.sh && ./scutlogin.sh")
+ log.Info("RunWiredLogin is called")
+ s.RunCommand(cmd)
+}
+
+func (s *SSHClass) RunWirelessLogin() {
+ cmd := fmt.Sprintf("chmod +x wl_login.sh && ./wl_login.sh")
+ log.Info("RunWirelessLogin is called")
+ s.RunCommand(cmd)
+}
+
+func (s *SSHClass) HelpSetInfo() {
+ // 辅助函数
+ s.saveInfo()
+
+ cmd := fmt.Sprintf("echo %s > SCUT_info.txt", s.Username)
+ cmd += fmt.Sprintf(" && echo %s >> SCUT_info.txt", s.Password)
+ cmd += fmt.Sprintf(" && echo %s >> SCUT_info.txt", s.AcIP)
+ cmd += fmt.Sprintf(" && echo %s >> SCUT_info.txt", s.AcName)
+ s.RunCommand(cmd)
+}
+
+func (s *SSHClass) RunSyncTime() {
+ cmd := fmt.Sprintf("ntpd -n -q -p ntp.aliyun.com")
+
+ log.Info("RunSyncTime is called")
+ s.RunCommand(cmd)
+}
+
+func (s *SSHClass) RunAutoLogin() {
+ log.Info("RunAutoLogin is called")
+
+ cmd := fmt.Sprintf("crontab -l | sed '/.*\\/root\\/scutlogin.sh.*/d' > temp_cron")
+ cmd += fmt.Sprintf(" && echo \"5 6 * * 1-5 /root/scutlogin.sh\" >> temp_cron")
+ cmd += fmt.Sprintf(" && crontab temp_cron\nrm temp_cron")
+
+ s.RunCommand(cmd)
+}
+
+func (s *SSHClass) CancelAutoLogin() {
+ log.Info("CancelAutoLogin is called")
+
+ cmd := fmt.Sprintf("crontab -l | sed '/.*\\/root\\/scutlogin.sh.*/d' > temp_cron")
+ cmd += fmt.Sprintf(" && crontab temp_cron\nrm temp_cron")
+ s.RunCommand(cmd)
+}
+
+type routerConfig struct {
+ Addr string `json:"addr"`
+ Password string `json:"password"`
+ Port string `json:"port"`
+}
+
+func NewRunSSH() *SSHClass {
+ buf, err := os.ReadFile("./config.json")
+ if err != nil {
+ log.Error(err.Error())
+ f, _ := os.Create("./config.json")
+ f.Close()
+ return nil
+ }
+
+ rConfig := new(routerConfig)
+ err = json.Unmarshal(buf, rConfig)
+ if err != nil {
+ log.Error(err.Error())
+ return nil
+ }
+
+ config := ssh.ClientConfig{
+ Config: ssh.Config{},
+ User: "root",
+ Auth: []ssh.AuthMethod{ssh.Password(rConfig.Password)},
+ HostKeyCallback: ssh.InsecureIgnoreHostKey(),
+ }
+ SSHObj := new(SSHClass)
+ addrPort := fmt.Sprintf("%s:%s", rConfig.Addr, rConfig.Port)
+ SSHObj.client, SSHObj.err = ssh.Dial("tcp", addrPort, &config)
+ SSHObj.Assert()
+
+ return SSHObj
+}
diff --git a/ScutLog.log b/ScutLog.log
new file mode 100644
index 0000000..e69de29
diff --git a/config.json b/config.json
new file mode 100644
index 0000000..6dc2c2a
--- /dev/null
+++ b/config.json
@@ -0,0 +1,5 @@
+{
+ "addr":"192.168.1.101",
+ "password": "KrxkKrxk",
+ "port": "22"
+}
\ No newline at end of file
diff --git a/frontend/src/components/Demo.vue b/frontend/src/components/Demo.vue
index 77ffd0b..427e13c 100644
--- a/frontend/src/components/Demo.vue
+++ b/frontend/src/components/Demo.vue
@@ -69,7 +69,7 @@
- 自由定制，真正属于你的校园网路由器
+ 自由定制，打造您的专属校园网路由
