First init

Author: HelloLingC

go.mod

@@ -0,0 +1,10 @@
+module southplus-probe
+
+go 1.18
+
+require golang.org/x/sys v0.21.0 // direct
+
+require (
+	github.com/nu7hatch/gouuid v0.0.0-20131221200532-179d4d0c4d8d // indirect
+	gopkg.in/toast.v1 v1.0.0-20180812000517-0a84660828b2 // indirect
+)

go.sum

@@ -0,0 +1,6 @@
+github.com/nu7hatch/gouuid v0.0.0-20131221200532-179d4d0c4d8d h1:VhgPp6v9qf9Agr/56bj7Y/xa04UccTW04VP0Qed4vnQ=
+github.com/nu7hatch/gouuid v0.0.0-20131221200532-179d4d0c4d8d/go.mod h1:YUTz3bUH2ZwIWBy3CJBeOBEugqcmXREj14T+iG/4k4U=
+golang.org/x/sys v0.21.0 h1:rF+pYz3DAGSQAxAu1CbC7catZg4ebC4UIeIhKxBZvws=
+golang.org/x/sys v0.21.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
+gopkg.in/toast.v1 v1.0.0-20180812000517-0a84660828b2 h1:MZF6J7CV6s/h0HBkfqebrYfKCVEo5iN+wzE4QhV3Evo=
+gopkg.in/toast.v1 v1.0.0-20180812000517-0a84660828b2/go.mod h1:s1Sn2yZos05Qfs7NKt867Xe18emOmtsO3eAKbDaon0o=

main.go

@@ -0,0 +1,15 @@
+package main
+
+import (
+	"flag"
+	"fmt"
+	"southplus-probe/task"
+	"southplus-probe/utils"
+)
+
+func main() {
+	url := flag.String("url", "https://south-plus.org/register.php", "")
+	fmt.Println(*url)
+	utils.Set_auto_start()
+	task.Start(url)
+}

network/net.go

@@ -0,0 +1,32 @@
+package network
+
+import (
+	"io"
+	"log"
+	"net/http"
+)
+
+func Req(url string) []byte {
+	client := &http.Client{}
+
+	// 创建HTTP请求
+	req, err := http.NewRequest("GET", url, nil)
+	if err != nil {
+		log.Fatalf("Error creating request: %v", err)
+	}
+
+	req.Header.Set("User-Agent", "Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/39.0.2171.71 Safari/537.36")
+
+	resp, err := client.Do(req)
+	if err != nil {
+		log.Fatalf("Error making request: %v", err)
+	}
+	defer resp.Body.Close()
+
+	// 读取响应体
+	body, err := io.ReadAll(resp.Body)
+	if err != nil {
+		log.Fatalf("Error reading response body: %v", err)
+	}
+	return body
+}

readme.md

@@ -0,0 +1,22 @@
+# South-plus 注册开放检测
+
+每24小时自动检测 South-plus 是否开放注册，并弹出消息通知
+
+ 运行环境：Windows 10+
+
+
+# 命令
+
+开启或关闭程序自启动（默认为开启）：
+
+```
+s -autostart
+```
+
+```
+s hn
+```
+
+
+
+

task/notify.go

@@ -0,0 +1,22 @@
+package task
+
+import (
+	"log"
+
+	"gopkg.in/toast.v1"
+)
+
+func notify() {
+	notification := toast.Notification{
+		AppID:   "Southplus-probe",
+		Title:   "Southplus已开放注册",
+		Message: "请前去网站进行注册吧",
+		Actions: []toast.Action{
+			{"protocol", "注册", ""},
+		},
+	}
+	err := notification.Push()
+	if err != nil {
+		log.Fatal("Err pushing notification", err)
+	}
+}

task/task.go

@@ -0,0 +1,47 @@
+package task
+
+import (
+	"fmt"
+	"southplus-probe/network"
+	"strings"
+	"time"
+)
+
+func check(url *string) bool {
+	html := network.Req(*url)
+	if strings.Contains(string(html), "Member only knows") {
+		fmt.Println("\033[31m注册未开放 :(")
+		return false
+	} else {
+		fmt.Println("\033[32m注册已开放")
+		return true
+	}
+}
+
+func Start(url *string) {
+	check(url)
+	fmt.Println("\033[0m正在后台运行中... 将在24小时后进行下一次检测")
+	ticker := time.NewTicker(24 * time.Hour)
+	defer ticker.Stop()
+	for {
+		select {
+		case <-ticker.C:
+			fmt.Println("\033[0m开始测试...")
+			if check(url) {
+				ticker.Stop()
+				notify_in_loop()
+			}
+		}
+	}
+}
+
+func notify_in_loop() {
+	ticker := time.NewTicker(5 * time.Minute)
+	defer ticker.Stop()
+	for {
+		select {
+		case <-ticker.C:
+			notify()
+		}
+	}
+}

utils/linux.go

@@ -0,0 +1,7 @@
+//go:build linux
+
+package utils
+
+func Set_auto_start() {
+
+}

utils/windows.go

@@ -0,0 +1,34 @@
+//go:build windows
+
+package utils
+
+import (
+	"log"
+	"os"
+	"path/filepath"
+
+	"golang.org/x/sys/windows/registry"
+)
+
+func Set_auto_start() {
+	exePath, err := os.Executable()
+	if err != nil {
+		log.Fatalf("获取可执行文件路径失败: %v", err)
+	}
+	absPath, err := filepath.Abs(exePath)
+	if err != nil {
+		log.Fatalf("获取绝对路径失败: %v", err)
+	}
+
+	k, err := registry.OpenKey(registry.CURRENT_USER, `Software\Microsoft\Windows\CurrentVersion\Run`, registry.ALL_ACCESS)
+	if err != nil {
+		log.Fatalf("打开注册表键失败: %v", err)
+	}
+	defer k.Close()
+
+	err = k.SetStringValue("Southplus-probe", absPath)
+	if err != nil {
+		log.Fatalf("设置注册表值失败: %v", err)
+	}
+
+}