-
Notifications
You must be signed in to change notification settings - Fork 6
/
subinfo.go
168 lines (165 loc) · 6.96 KB
/
subinfo.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
package main
import (
"errors"
"fmt"
_ "fmt"
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
"math"
"net/http"
"regexp"
"strconv"
"strings"
"subinfobot/handler"
"subinfobot/utils"
"time"
)
type Subinfo struct {
Link string
ExpireTime string
TimeRemain string
Upload string
Download string
Used string
Total string
Expired int //0:not Expired,1:Expired,2:unknown
Available int //0:Available,1:unavailable,2:unknown
DataRemain string
}
func getSinf(link string) (error, Subinfo) {
httpClient := &http.Client{Timeout: 10 * time.Second}
req, err := http.NewRequest("GET", link, nil)
req.Header.Add("User-Agent", "ClashforWindows/0.19.21")
if err != nil {
return err, Subinfo{}
}
res, err := httpClient.Do(req)
if err != nil {
return err, Subinfo{}
}
if res.StatusCode >= 400 {
return errors.New(fmt.Sprintf("获取失败，服务器返回了代码%s", strconv.Itoa(res.StatusCode))), Subinfo{}
}
if sinfo := res.Header["Subscription-Userinfo"]; sinfo == nil {
return errors.New("未获取到订阅详细信息，该订阅可能已经到期或者已被删除"), Subinfo{}
} else {
sinf := Subinfo{Link: link}
sinfmap := make(map[string]int64)
parseExp := regexp.MustCompile("[A-Za-z]+=[0-9]+")
sslice := parseExp.FindAllStringSubmatch(sinfo[0], -1)
for _, val := range sslice {
kvslice := strings.Split(val[0], "=")
if len(kvslice) == 2 {
i, err := strconv.ParseInt(kvslice[1], 10, 64)
if err == nil {
sinfmap[kvslice[0]] = i
}
}
}
if upload, oku := sinfmap["upload"]; oku {
sinf.Upload = utils.FormatFileSize(upload)
} else {
sinf.Upload = "没有说明捏"
}
if download, okd := sinfmap["download"]; okd {
sinf.Download = utils.FormatFileSize(download)
} else {
sinf.Download = "没有说明捏"
}
if total, okt := sinfmap["total"]; okt {
sinf.Total = utils.FormatFileSize(total)
down, oka := sinfmap["download"]
up, okb := sinfmap["upload"]
if (oka) && (okb) {
sinf.Used = utils.FormatFileSize(up + down)
remain := total - (up + down)
if remain >= 0 {
if remain > 0 {
sinf.Available = 0
sinf.DataRemain = utils.FormatFileSize(remain)
} else {
sinf.Available = 1
sinf.DataRemain = utils.FormatFileSize(remain)
}
} else {
sinf.Available = 1
sinf.DataRemain = "逾量" + utils.FormatFileSize(int64(math.Abs(float64(remain))))
}
} else {
sinf.Available = 2
sinf.DataRemain = "不知道捏"
}
} else {
sinf.Available = 2
sinf.Total = "没有说明捏"
}
if exp, oke := sinfmap["expire"]; oke {
//get expire time and remain time
timeStamp := time.Now().Unix()
timeExp := time.Unix(exp, 0)
sinf.ExpireTime = timeExp.String()
if timeStamp >= exp {
sinf.Expired = 1
sinf.Available = 1
remain := timeExp.Sub(time.Now())
if remain.Hours() > 24 {
sinf.TimeRemain = "逾期<code>" + strconv.Itoa(int(math.Floor(remain.Hours()/24))) + "天" + strconv.Itoa(int(math.Floor(float64(int(math.Floor(remain.Hours()))%24)))) + "小时" + strconv.Itoa(int(math.Floor(float64(int(math.Floor(remain.Minutes()))%60)))) + "分" + strconv.Itoa(int(math.Floor(float64(int(math.Floor(remain.Seconds()))%60)))) + "秒" + "</code>"
} else if remain.Minutes() > 60 {
sinf.TimeRemain = "逾期<code>" + strconv.Itoa(int(math.Floor(float64(int(math.Floor(remain.Hours()))%24)))) + "小时" + strconv.Itoa(int(math.Floor(float64(int(math.Floor(remain.Minutes()))%60)))) + "分" + strconv.Itoa(int(math.Floor(float64(int(math.Floor(remain.Seconds()))%60)))) + "秒" + "</code>"
} else if remain.Seconds() > 60 {
sinf.TimeRemain = "逾期<code>" + strconv.Itoa(int(math.Floor(float64(int(math.Floor(remain.Minutes()))%60)))) + "分" + strconv.Itoa(int(math.Floor(float64(int(math.Floor(remain.Seconds()))%60)))) + "秒" + "</code>"
} else {
sinf.TimeRemain = "逾期<code>" + strconv.Itoa(int(math.Floor(remain.Seconds()))) + "秒" + "</code>"
}
} else {
sinf.Expired = 0
remain := timeExp.Sub(time.Now())
if remain.Hours() > 24 {
sinf.TimeRemain = "距离到期还有<code>" + strconv.Itoa(int(math.Floor(remain.Hours()/24))) + "天" + strconv.Itoa(int(math.Floor(float64(int(math.Floor(remain.Hours()))%24)))) + "小时" + strconv.Itoa(int(math.Floor(float64(int(math.Floor(remain.Minutes()))%60)))) + "分" + strconv.Itoa(int(math.Floor(float64(int(math.Floor(remain.Seconds()))%60)))) + "秒" + "</code>"
} else if remain.Minutes() > 60 {
sinf.TimeRemain = "距离到期还有<code>" + strconv.Itoa(int(math.Floor(float64(int(math.Floor(remain.Hours()))%24)))) + "小时" + strconv.Itoa(int(math.Floor(float64(int(math.Floor(remain.Minutes()))%60)))) + "分" + strconv.Itoa(int(math.Floor(float64(int(math.Floor(remain.Seconds()))%60)))) + "秒" + "</code>"
} else if remain.Seconds() > 60 {
sinf.TimeRemain = "距离到期还有<code>" + strconv.Itoa(int(math.Floor(float64(int(math.Floor(remain.Minutes()))%60)))) + "分" + strconv.Itoa(int(math.Floor(float64(int(math.Floor(remain.Seconds()))%60)))) + "秒" + "</code>"
} else {
sinf.TimeRemain = "距离到期还有<code>" + strconv.Itoa(int(math.Floor(remain.Seconds()))) + "秒" + "</code>"
}
}
} else {
sinf.ExpireTime = "未知"
sinf.TimeRemain = "可能是无限时长订阅或者服务器抽抽了呢"
}
return nil, sinf
}
}
func subInfoMsg(link string, update *tgbotapi.Update, bot *tgbotapi.BotAPI, msg *tgbotapi.MessageConfig) {
msg.Text = "🕰获取中..."
msg.ReplyToMessageID = update.Message.MessageID
sres, err := handler.SendMsg(bot, msg)
handler.HandleError(err)
if err == nil {
err, sinf := getSinf(link)
handler.HandleError(err)
if err != nil {
_, err := handler.EditMsg(fmt.Sprintf("<strong>❌获取失败</strong>\n\n获取订阅<code>%s</code>时发生错误:\n<code>%s</code>", sinf.Link, err), "html", bot, sres)
handler.HandleError(err)
if update.Message.Chat.Type == "group" || update.Message.Chat.Type == "supergroup" {
_, _ = handler.DelMsgWithTimeOut(10*time.Second, bot, sres)
}
} else {
var resMsg string
if sinf.Expired == 0 && sinf.Available == 0 {
resMsg = "✅该订阅有效"
}
if sinf.Expired == 2 || sinf.Available == 2 {
resMsg = "❓该订阅状态未知"
}
if sinf.Expired == 1 || sinf.Available == 1 {
resMsg = "❌该订阅不可用"
}
_, err = handler.EditMsg(fmt.Sprintf("<strong>%s</strong>\n<strong>订阅链接:</strong><code>%s</code>\n<strong>总流量:</strong><code>%s</code>\n<strong>剩余流量:</strong><code>%s</code>\n<strong>已上传:</strong><code>%s</code>\n<strong>已下载:</strong><code>%s</code>\n<strong>该订阅将于<code>%s</code>过期,%s</strong>", resMsg, sinf.Link, sinf.Total, sinf.DataRemain, sinf.Upload, sinf.Download, sinf.ExpireTime, sinf.TimeRemain), "html", bot, sres)
handler.HandleError(err)
if update.Message.Chat.Type == "group" || update.Message.Chat.Type == "supergroup" {
_, _ = handler.DelMsgWithTimeOut(10*time.Second, bot, sres)
}
}
}
}