+
+
+
+
+
+
+ `
+
+func Count() string {
+ zs := 0
+ yx := 0
+ wx := 0
+ tl := 0
+ ts := 0
+ tc := 0
+ dt := Date()
+ cks := GetJdCookies()
+ for _, ck := range cks {
+ zs++
+ if ck.Available == True {
+ yx++
+ } else {
+ wx++
+ }
+ if ck.CreateAt == dt {
+ tc++
+ }
+ // if ck.ScanedAt == dt {
+ // ts++
+ // }
+ // if ck.LoseAt == dt {
+ // tl++
+ // }
+ }
+ return fmt.Sprintf("总数%d,有效%d,无效%d,今日失效%d,今日扫码%d,今日新增%d", zs, yx, wx, tc, ts, tl)
+}
diff --git a/models/asset.go b/models/asset.go
new file mode 100644
index 0000000..6cbc299
--- /dev/null
+++ b/models/asset.go
@@ -0,0 +1,143 @@
+package models
+
+import (
+ "encoding/json"
+ "fmt"
+ "net/url"
+ "strconv"
+ "strings"
+ "time"
+
+ "github.com/beego/beego/v2/client/httplib"
+)
+
+type Asset struct {
+ Nickname string
+ Bean struct {
+ Total int
+ TodayIn int
+ TodayOut int
+ YestodayIn int
+ YestodayOut int
+ MonthIn int
+ MonthOut int
+ ToExpire []int
+ }
+ RedPacket struct {
+ Total float64
+ ToExpire float64
+ ToExpireJd float64
+ ToExpireJx float64
+ ToExpireJs float64
+ }
+}
+
+var Int = func(s string) int {
+ i, _ := strconv.Atoi(s)
+ return i
+}
+
+func DailyAssetsPush() {
+ for _, ck := range GetJdCookies() {
+ msg := ck.Query()
+ if ck.QQ != 0 && Config.QQID != 0 && SendQQ != nil {
+ SendQQ(int64(ck.QQ), msg)
+ }
+ if ck.PushPlus != "" {
+ pushPlus(ck.PushPlus, msg)
+ }
+ }
+}
+
+func (ck *JdCookie) Query() string {
+ msgs := []string{}
+ asset := Asset{}
+ if CookieOK(ck) {
+ today := time.Now().Local().Format("2006-01-02")
+ yestoday := time.Now().Local().Add(-time.Hour * 24).Format("2006-01-02")
+ month := time.Now().Local().Format("2006-01")
+ page := 1
+ end := false
+ for {
+ if end {
+ break
+ }
+ bds := getJingBeanBalanceDetail(page, fmt.Sprintf("pt_key=%s;pt_pin=%s;", ck.PtKey, ck.PtPin))
+ if bds == nil {
+ end = true
+ break
+ }
+ for _, bd := range bds {
+ amount := Int(bd.Amount)
+ if !strings.Contains(bd.Date, month) {
+ end = true
+ break
+ }
+ if amount > 0 {
+ asset.Bean.MonthIn += amount
+ } else {
+ asset.Bean.MonthOut += -amount
+ }
+ if strings.Contains(bd.Date, today) {
+ if amount > 0 {
+ asset.Bean.TodayIn += amount
+ } else {
+ asset.Bean.TodayOut += -amount
+ }
+ }
+ if strings.Contains(bd.Date, yestoday) {
+ if amount > 0 {
+ asset.Bean.YestodayIn += amount
+ } else {
+ asset.Bean.YestodayOut += -amount
+ }
+ }
+ }
+ page++
+ }
+ msgs = append(msgs, []string{
+ fmt.Sprintf("当月收入：%d京豆", asset.Bean.MonthIn),
+ fmt.Sprintf("当月支出：%d京豆", asset.Bean.MonthOut),
+ fmt.Sprintf("昨日收入：%d京豆", asset.Bean.YestodayIn),
+ fmt.Sprintf("昨日支出：%d京豆", asset.Bean.YestodayOut),
+ fmt.Sprintf("今日收入：%d京豆", asset.Bean.TodayIn),
+ fmt.Sprintf("今日支出：%d京豆", asset.Bean.TodayOut),
+ }...)
+ } else {
+ msgs = append(msgs, []string{
+ "提醒：该账号已过期，请重新登录",
+ }...)
+ }
+ ck.PtPin, _ = url.QueryUnescape(ck.PtPin)
+ return strings.Join(append([]string{
+ fmt.Sprintf("账号：%s", ck.PtPin),
+ fmt.Sprintf("昵称：%s", ck.Nickname),
+ fmt.Sprintf("备注：%s", ck.Note),
+ }, append(msgs, fmt.Sprintf("当前京豆：%v京豆", ck.BeanNum))...), "\n")
+}
+
+type BeanDetail struct {
+ Date string `json:"date"`
+ Amount string `json:"amount"`
+ EventMassage string `json:"eventMassage"`
+}
+
+func getJingBeanBalanceDetail(page int, cookie string) []BeanDetail {
+ type AutoGenerated struct {
+ Code string `json:"code"`
+ DetailList []BeanDetail `json:"detailList"`
+ }
+ a := AutoGenerated{}
+ req := httplib.Post(`https://api.m.jd.com/client.action?functionId=getJingBeanBalanceDetail`)
+ req.Header("User-Agent", ua)
+ req.Header("Host", "api.m.jd.com")
+ req.Header("Content-Type", "application/x-www-form-urlencoded")
+ req.Header("Cookie", cookie)
+ req.Body(fmt.Sprintf(`body={"pageSize": "20", "page": "%d"}&appid=ld`, page))
+ data, err := req.Bytes()
+ if err != nil {
+ return nil
+ }
+ json.Unmarshal(data, &a)
+ return a.DetailList
+}
diff --git a/models/available.go b/models/available.go
new file mode 100644
index 0000000..654d969
--- /dev/null
+++ b/models/available.go
@@ -0,0 +1,178 @@
+package models
+
+import (
+ "encoding/json"
+ "fmt"
+
+ "github.com/beego/beego/v2/client/httplib"
+)
+
+type UserInfoResult struct {
+ Data struct {
+ JdVvipCocoonInfo struct {
+ JdVvipCocoon struct {
+ DisplayType int `json:"displayType"`
+ HitTypeList []int `json:"hitTypeList"`
+ Link string `json:"link"`
+ Price string `json:"price"`
+ Qualification int `json:"qualification"`
+ SellingPoints string `json:"sellingPoints"`
+ } `json:"JdVvipCocoon"`
+ JdVvipCocoonStatus string `json:"JdVvipCocoonStatus"`
+ } `json:"JdVvipCocoonInfo"`
+ JdVvipInfo struct {
+ JdVvipStatus string `json:"jdVvipStatus"`
+ } `json:"JdVvipInfo"`
+ AssetInfo struct {
+ AccountBalance string `json:"accountBalance"`
+ BaitiaoInfo struct {
+ AvailableLimit string `json:"availableLimit"`
+ BaiTiaoStatus string `json:"baiTiaoStatus"`
+ Bill string `json:"bill"`
+ BillOverStatus string `json:"billOverStatus"`
+ Outstanding7Amount string `json:"outstanding7Amount"`
+ OverDueAmount string `json:"overDueAmount"`
+ OverDueCount string `json:"overDueCount"`
+ UnpaidForAll string `json:"unpaidForAll"`
+ UnpaidForMonth string `json:"unpaidForMonth"`
+ } `json:"baitiaoInfo"`
+ BeanNum string `json:"beanNum"`
+ CouponNum string `json:"couponNum"`
+ CouponRed string `json:"couponRed"`
+ RedBalance string `json:"redBalance"`
+ } `json:"assetInfo"`
+ FavInfo struct {
+ FavDpNum string `json:"favDpNum"`
+ FavGoodsNum string `json:"favGoodsNum"`
+ FavShopNum string `json:"favShopNum"`
+ FootNum string `json:"footNum"`
+ IsGoodsRed string `json:"isGoodsRed"`
+ IsShopRed string `json:"isShopRed"`
+ } `json:"favInfo"`
+ GrowHelperCoupon struct {
+ AddDays int `json:"addDays"`
+ BatchID int `json:"batchId"`
+ CouponKind int `json:"couponKind"`
+ CouponModel int `json:"couponModel"`
+ CouponStyle int `json:"couponStyle"`
+ CouponType int `json:"couponType"`
+ Discount float64 `json:"discount"`
+ LimitType int `json:"limitType"`
+ MsgType int `json:"msgType"`
+ Quota float64 `json:"quota"`
+ RoleID int `json:"roleId"`
+ State int `json:"state"`
+ Status int `json:"status"`
+ } `json:"growHelperCoupon"`
+ KplInfo struct {
+ KplInfoStatus string `json:"kplInfoStatus"`
+ Mopenbp17 string `json:"mopenbp17"`
+ Mopenbp22 string `json:"mopenbp22"`
+ } `json:"kplInfo"`
+ OrderInfo struct {
+ CommentCount string `json:"commentCount"`
+ Logistics []interface{} `json:"logistics"`
+ OrderCountStatus string `json:"orderCountStatus"`
+ ReceiveCount string `json:"receiveCount"`
+ WaitPayCount string `json:"waitPayCount"`
+ } `json:"orderInfo"`
+ PlusPromotion struct {
+ Status int `json:"status"`
+ } `json:"plusPromotion"`
+ UserInfo struct {
+ BaseInfo struct {
+ AccountType string `json:"accountType"`
+ BaseInfoStatus string `json:"baseInfoStatus"`
+ CurPin string `json:"curPin"`
+ DefinePin string `json:"definePin"`
+ HeadImageURL string `json:"headImageUrl"`
+ LevelName string `json:"levelName"`
+ Nickname string `json:"nickname"`
+ Pinlist string `json:"pinlist"`
+ UserLevel string `json:"userLevel"`
+ } `json:"baseInfo"`
+ IsHideNavi string `json:"isHideNavi"`
+ IsHomeWhite string `json:"isHomeWhite"`
+ IsJTH string `json:"isJTH"`
+ IsKaiPu string `json:"isKaiPu"`
+ IsPlusVip string `json:"isPlusVip"`
+ IsQQFans string `json:"isQQFans"`
+ IsRealNameAuth string `json:"isRealNameAuth"`
+ IsWxFans string `json:"isWxFans"`
+ Jvalue string `json:"jvalue"`
+ OrderFlag string `json:"orderFlag"`
+ PlusInfo struct {
+ } `json:"plusInfo"`
+ XbScore string `json:"xbScore"`
+ } `json:"userInfo"`
+ UserLifeCycle struct {
+ IdentityID string `json:"identityId"`
+ LifeCycleStatus string `json:"lifeCycleStatus"`
+ TrackID string `json:"trackId"`
+ } `json:"userLifeCycle"`
+ } `json:"data"`
+ Msg string `json:"msg"`
+ Retcode string `json:"retcode"`
+ Timestamp int64 `json:"timestamp"`
+}
+
+func initCookie() {
+ cks := GetJdCookies()
+ l := len(cks)
+ for i := 0; i < l-1; i++ {
+ if cks[i].Available == True && !CookieOK(&cks[i]) {
+ if pt_key, err := cks[i].OutPool(); err == nil && pt_key != "" {
+ i--
+ }
+ }
+ }
+ go func() {
+ Save <- &JdCookie{}
+ }()
+
+}
+
+func CookieOK(ck *JdCookie) bool {
+ if ck == nil {
+ return true
+ }
+ req := httplib.Get("https://me-api.jd.com/user_new/info/GetJDUserInfoUnion")
+ req.Header("Cookie", fmt.Sprintf("pt_key=%s;pt_pin=%s;", ck.PtKey, ck.PtPin))
+ req.Header("Accept", "*/*")
+ req.Header("Accept-Language", "zh-cn,")
+ req.Header("Connection", "keep-alive,")
+ req.Header("Referer", "https://home.m.jd.com/myJd/newhome.action?sceneval=2&ufc=&")
+ req.Header("Host", "me-api.jd.com")
+ req.Header("User-Agent", "jdapp;iPhone;9.4.4;14.3;network/4g;Mozilla/5.0 (iPhone; CPU iPhone OS 14_3 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Mobile/15E148;supportJDSHWK/1")
+ data, err := req.Bytes()
+ if err != nil {
+ return true
+ }
+ ui := &UserInfoResult{}
+ if nil != json.Unmarshal(data, ui) {
+ return true
+ }
+ switch ui.Retcode {
+ case "1001": //ck.BeanNum
+ if ui.Msg == "not login" {
+ if ck.Available == True {
+ ck.Push(fmt.Sprintf("失效账号，%s", ck.PtPin))
+ }
+ return false
+ }
+ case "0":
+ if ui.Data.UserInfo.BaseInfo.Nickname != ck.Nickname || ui.Data.AssetInfo.BeanNum != ck.BeanNum {
+ ck.Updates(JdCookie{
+ Nickname: ui.Data.UserInfo.BaseInfo.Nickname,
+ BeanNum: ui.Data.AssetInfo.BeanNum,
+ Available: True,
+ })
+ ck.Nickname = ui.Data.UserInfo.BaseInfo.Nickname
+ ck.BeanNum = ui.Data.AssetInfo.BeanNum
+
+ }
+ default:
+
+ }
+ return true
+}
diff --git a/models/bot.go b/models/bot.go
new file mode 100644
index 0000000..99ada34
--- /dev/null
+++ b/models/bot.go
@@ -0,0 +1,393 @@
+package models
+
+import (
+ "fmt"
+ "io/ioutil"
+ "net/url"
+ "os"
+ "os/exec"
+ "regexp"
+ "strconv"
+ "strings"
+
+ "github.com/beego/beego/v2/client/httplib"
+ "github.com/beego/beego/v2/core/logs"
+ "github.com/beego/beego/v2/server/web"
+)
+
+var SendQQ func(int64, interface{})
+var SendQQGroup func(int64, int64, interface{})
+var ListenQQPrivateMessage = func(uid int64, msg string) {
+ SendQQ(uid, handleMessage(msg, "qq", int(uid)))
+}
+
+var ListenQQGroupMessage = func(gid int64, uid int64, msg string) {
+ if gid == Config.QQGroupID {
+ if Config.QbotPublicMode {
+ SendQQGroup(gid, uid, handleMessage(msg, "qqg", int(gid), int(uid)))
+ } else {
+ SendQQ(uid, handleMessage(msg, "qq", int(uid)))
+ }
+ }
+}
+
+var replies = map[string]string{}
+
+func InitReplies() {
+ f, err := os.Open(ExecPath + "/conf/reply.php")
+ if err == nil {
+ defer f.Close()
+ data, _ := ioutil.ReadAll(f)
+ ss := regexp.MustCompile("`([^`]+)`\\s*=>\\s*`([^`]+)`").FindAllStringSubmatch(string(data), -1)
+ for _, s := range ss {
+ replies[s[1]] = s[2]
+ }
+ }
+ if _, ok := replies["壁纸"]; !ok {
+ replies["壁纸"] = "https://acg.toubiec.cn/random.php"
+ }
+}
+
+var sendMessagee = func(msg string, msgs ...interface{}) {
+ tp := msgs[1].(string)
+ id := msgs[2].(int)
+ switch tp {
+ case "tg":
+ SendTgMsg(id, msg)
+ case "qq":
+ SendQQ(int64(id), msg)
+ case "qqg":
+ SendQQGroup(int64(id), int64(msgs[3].(int)), msg)
+ }
+}
+
+var sendAdminMessagee = func(msg string, msgs ...interface{}) {
+ tp := msgs[1].(string)
+ id := msgs[2].(int)
+ switch tp {
+ case "tg":
+ if Config.TelegramUserID == id {
+ SendTgMsg(id, msg)
+ }
+ case "qq":
+ if int(Config.QQID) == id {
+ SendQQ(int64(id), msg)
+ }
+ case "qqg":
+ uid := msgs[3].(int)
+ if int(Config.QQID) == uid {
+ SendQQGroup(int64(id), int64(uid), msg)
+ }
+ }
+}
+
+var isAdmin = func(msgs ...interface{}) bool {
+ tp := msgs[1].(string)
+ id := msgs[2].(int)
+ switch tp {
+ case "tg":
+ if Config.TelegramUserID == id {
+ return true
+ }
+ case "qq":
+ if int(Config.QQID) == id {
+ return true
+ }
+ case "qqg":
+ uid := msgs[3].(int)
+ if int(Config.QQID) == uid {
+ return true
+ }
+ }
+ return false
+}
+
+var handleMessage = func(msgs ...interface{}) interface{} {
+ msg := msgs[0].(string)
+ tp := msgs[1].(string)
+ id := msgs[2].(int)
+ switch msg {
+ case "status", "状态":
+ if !isAdmin(msgs...) {
+ return "你没有权限操作"
+ }
+ return Count()
+ case "qrcode", "扫码", "二维码":
+ url := ""
+ if tp == "qqg" {
+ url = fmt.Sprintf("http://127.0.0.1:%d/api/login/qrcode.png?%vid=%v&qqguid=%v", web.BConfig.Listen.HTTPPort, tp, id, msgs[3].(int))
+ } else {
+ url = fmt.Sprintf("http://127.0.0.1:%d/api/login/qrcode.png?%vid=%v", web.BConfig.Listen.HTTPPort, tp, id)
+ }
+ rsp, err := httplib.Get(url).Response()
+ if err != nil {
+ return nil
+ }
+ return rsp
+ case "升级":
+ if !isAdmin(msgs...) { //
+ return "你没有权限操作"
+ }
+ sendMessagee("小滴滴开始拉取代码", msgs...)
+ rtn, err := exec.Command("sh", "-c", "cd "+ExecPath+" && git pull").Output()
+ if err != nil {
+ return err.Error()
+ }
+ t := string(rtn)
+ if !strings.Contains(t, "changed") {
+ if strings.Contains(t, "Already") || strings.Contains(t, "已经是最新") {
+ sendMessagee("小滴滴已是最新版啦", msgs...)
+ } else {
+ sendMessagee("小滴滴拉取代失败：", msgs...)
+ }
+ return nil
+ } else {
+ sendMessagee("小滴滴拉取代码成功", msgs...)
+ }
+ sendMessagee("小滴滴正在编译程序", msgs...)
+ rtn, err = exec.Command("sh", "-c", "cd "+ExecPath+" && go build -o "+pname).Output()
+ if err != nil {
+ sendMessagee("小滴滴编译失败：", msgs...)
+ return nil
+ } else {
+ sendAdminMessagee("小滴滴编译成功", msgs...)
+ }
+ fallthrough
+ case "重启":
+ if !isAdmin(msgs...) {
+ return "你没有权限操作"
+ }
+ sendAdminMessagee("小滴滴重启程序", msgs...)
+ Daemon()
+ return nil
+ case "查询", "query":
+ cks := GetJdCookies()
+ for _, ck := range cks {
+ if tp == "qq" {
+ if ck.QQ == id {
+ SendQQ(int64(id), ck.Query())
+ }
+ } else if tp == "qqg" {
+ if ck.QQ == msgs[3].(int) {
+ SendQQGroup(int64(id), int64(msgs[3].(int)), ck.Query())
+ }
+ }
+
+ }
+ return nil
+ default:
+ { //
+ ss := regexp.MustCompile(`pt_key=([^;=\s]+);pt_pin=([^;=\s]+)`).FindAllStringSubmatch(msg, -1)
+ if len(ss) > 0 {
+ xyb := 0
+ for _, s := range ss {
+ ck := JdCookie{
+ PtKey: s[1],
+ PtPin: s[2],
+ }
+ if CookieOK(&ck) {
+ xyb++
+ if tp == "qq" {
+ ck.QQ = id
+
+ } else if tp == "tg" {
+ ck.Telegram = id
+ } else if tp == "qqg" {
+ ck.QQ = msgs[3].(int)
+ }
+ if nck, err := GetJdCookie(ck.PtPin); err == nil {
+ nck.InPool(ck.PtKey)
+ msg := fmt.Sprintf("更新账号，%s", ck.PtPin)
+ (&JdCookie{}).Push(msg)
+ sendMessagee("许愿币+1", msgs...)
+ logs.Info(msg)
+ } else {
+ NewJdCookie(&ck)
+ msg := fmt.Sprintf("添加账号，%s", ck.PtPin)
+ (&JdCookie{}).Push(msg)
+ sendMessagee("许愿币+1", msgs...)
+ logs.Info(msg)
+ }
+ }
+ }
+ go func() {
+ Save <- &JdCookie{}
+ }()
+ return nil
+ }
+ }
+ {
+ s := regexp.MustCompile(`([^\s]+)\s+(.*)`).FindStringSubmatch(msg)
+ if len(s) > 0 {
+ v := s[2]
+ switch s[1] {
+ case "查询", "query":
+ if !isAdmin(msgs...) {
+ return "你没有权限操作"
+ }
+ cks := GetJdCookies()
+ a := s[2]
+ {
+ if s := strings.Split(a, "-"); len(s) == 2 {
+ for i, ck := range cks {
+ if i+1 >= Int(s[0]) && i+1 <= Int(s[1]) {
+ switch tp {
+ case "tg":
+ tgBotNotify(ck.Query())
+ case "qq":
+ if id == ck.QQ {
+ SendQQ(int64(id), ck.Query())
+ } else {
+ SendQQ(Config.QQID, ck.Query())
+ }
+ case "qqg":
+ uid := msgs[3].(int)
+ if uid == ck.QQ || uid == int(Config.QQID) {
+ SendQQGroup(int64(id), int64(msgs[3].(int)), ck.Query())
+ }
+ }
+ }
+ }
+ return nil
+ }
+ }
+ {
+ if x := regexp.MustCompile(`^[\s\d,]+$`).FindString(a); x != "" {
+ xx := regexp.MustCompile(`(\d+)`).FindAllStringSubmatch(a, -1)
+ for i, ck := range cks {
+ for _, x := range xx {
+ if fmt.Sprint(i+1) == x[1] {
+ switch tp {
+ case "tg":
+ tgBotNotify(ck.Query())
+ case "qq":
+ if id == ck.QQ {
+ SendQQ(int64(id), ck.Query())
+ } else {
+ SendQQ(Config.QQID, ck.Query())
+ }
+ case "qqg":
+ uid := msgs[3].(int)
+ if uid == ck.QQ || uid == int(Config.QQID) {
+ SendQQGroup(int64(id), int64(msgs[3].(int)), ck.Query())
+ }
+ }
+ }
+ }
+
+ }
+ return nil
+ }
+ }
+ {
+ a = strings.Replace(a, " ", "", -1)
+ for _, ck := range cks {
+ if strings.Contains(ck.Note, a) || strings.Contains(ck.Nickname, a) || strings.Contains(ck.PtPin, a) {
+ switch tp {
+ case "tg":
+ tgBotNotify(ck.Query())
+ case "qq":
+ if id == ck.QQ {
+ SendQQ(int64(id), ck.Query())
+ } else {
+ SendQQ(Config.QQID, ck.Query())
+ }
+ case "qqg":
+ uid := msgs[3].(int)
+ if uid == ck.QQ || uid == int(Config.QQID) {
+ SendQQGroup(int64(id), int64(msgs[3].(int)), ck.Query())
+ }
+ }
+ }
+ }
+ return nil
+ }
+ case "许愿":
+ if tp == "qqg" {
+ id = msgs[3].(int)
+ }
+ b := 0
+ for _, ck := range GetJdCookies() {
+ if id == ck.QQ || id == ck.Telegram {
+ b++
+ }
+ }
+ if b <= 0 {
+ return "许愿币不足"
+ } else {
+ (&JdCookie{}).Push(fmt.Sprintf("%d许愿%s，许愿币余额%d。", id, v, b))
+ return "收到许愿"
+ }
+ case "扣除许愿币":
+ id, _ := strconv.Atoi(v)
+ b := 0
+ k := 0
+ for _, ck := range GetJdCookies() {
+ if id == ck.QQ || id == ck.Telegram {
+ if k <= 5 {
+ ck.Updates(map[string]interface{}{
+ QQ: 0,
+ Telegram: 0,
+ })
+ k++
+ } else {
+ b++
+ }
+ }
+ }
+ return fmt.Sprintf("操作成功，%d剩余许愿币%d", id, b)
+ }
+
+ }
+ }
+ {
+ o := false
+ for _, v := range regexp.MustCompile(`京东账号\d*（(.*)）(.*)】(.*)`).FindAllStringSubmatch(msg, -1) {
+ if !strings.Contains(v[3], "种子") && !strings.Contains(v[3], "undefined") {
+ pt_pin := url.QueryEscape(v[1])
+ for key, ss := range map[string][]string{
+ "Fruit": {"京东农场", "东东农场"},
+ "Pet": {"京东萌宠"},
+ "Bean": {"种豆得豆"},
+ "JdFactory": {"东东工厂"},
+ "DreamFactory": {"京喜工厂"},
+ "Jxnc": {"京喜农场"},
+ "Jdzz": {"京东赚赚"},
+ "Joy": {"crazyJoy"},
+ "Sgmh": {"闪购盲盒"},
+ "Cfd": {"财富岛"},
+ "Cash": {"签到领现金"},
+ } {
+ for _, s := range ss {
+ if strings.Contains(v[2], s) && v[3] != "" {
+ if ck, err := GetJdCookie(pt_pin); err == nil {
+ ck.Update(key, v[3])
+ }
+ if !o {
+ o = true
+ }
+ }
+ }
+ }
+ }
+ }
+ if o {
+ return "导入互助码成功"
+ }
+ }
+ for k, v := range replies {
+ if regexp.MustCompile(k).FindString(msg) != "" {
+ if regexp.MustCompile(`^https{0,1}://[^\x{4e00}-\x{9fa5}\n\r\s]{3,}$`).FindString(v) != "" {
+ url := v
+ rsp, err := httplib.Get(url).Response()
+ if err != nil {
+ return nil
+ }
+ return rsp
+ }
+ return v
+ }
+ }
+ }
+ return nil
+}
diff --git a/models/config.go b/models/config.go
new file mode 100644
index 0000000..59a59b2
--- /dev/null
+++ b/models/config.go
@@ -0,0 +1,109 @@
+package models
+
+import (
+ "bufio"
+ "io"
+ "io/ioutil"
+ "os"
+
+ "github.com/beego/beego/v2/client/httplib"
+ "github.com/beego/beego/v2/core/logs"
+ "gopkg.in/yaml.v2"
+)
+
+type Container struct {
+ Type string
+ Name string
+ Default bool
+ Address string
+ Username string
+ Password string
+ Path string
+ Version string
+ Token string
+ Available bool
+ Delete []string
+ Weigth int
+ Mode string
+ Reader *bufio.Reader
+ Config string
+ Limit int
+}
+type Yaml struct {
+ Containers []Container
+ Qrcode string
+ Master string
+ Mode string
+ Static string
+ Database string
+ QywxKey string `yaml:"qywx_key"`
+ Resident string
+ UserAgent string `yaml:"user_agent"`
+ Theme string
+ TelegramBotToken string `yaml:"telegram_bot_token"`
+ TelegramUserID int `yaml:"telegram_user_id"`
+ QQID int64 `yaml:"qquid"`
+ QQGroupID int64 `yaml:"qqgid"`
+ DefaultPriority int `yaml:"default_priority"`
+ NoGhproxy bool `yaml:"no_ghproxy"`
+ QbotPublicMode bool `yaml:"qbot_public_mode"`
+ DailyAssetPushCron string `yaml:"daily_asset_push_cron"`
+ Version string `yaml:"version"`
+}
+
+var Balance = "balance"
+var Parallel = "parallel"
+var GhProxy = "https://ghproxy.com/"
+var Cdle = false
+
+var Config Yaml
+
+func initConfig() {
+ confDir := ExecPath + "/conf"
+ if _, err := os.Stat(confDir); err != nil {
+ os.MkdirAll(confDir, os.ModePerm)
+ }
+ for _, name := range []string{"app.conf", "config.yaml"} {
+ f, err := os.OpenFile(ExecPath+"/conf/"+name, os.O_RDWR|os.O_CREATE, 0777)
+ if err != nil {
+ logs.Warn(err)
+ }
+ s, _ := ioutil.ReadAll(f)
+ if len(s) == 0 {
+ logs.Info("下载配置%s", name)
+ r, err := httplib.Get(GhProxy + "https://raw.githubusercontent.com/cdle/xdd/main/conf/" + name).Response()
+ if err == nil {
+ io.Copy(f, r.Body)
+ }
+ }
+ f.Close()
+ }
+ content, err := ioutil.ReadFile(ExecPath + "/conf/config.yaml")
+ if err != nil {
+ logs.Warn("解析config.yaml读取错误: %v", err)
+ }
+ if yaml.Unmarshal(content, &Config) != nil {
+ logs.Warn("解析config.yaml出错: %v", err)
+ }
+ if Config.Master == "" {
+ Config.Master = "xxxx"
+ }
+ if Config.Mode != Parallel {
+ Config.Mode = Balance
+ }
+ if Config.Qrcode != "" {
+ Config.Theme = Config.Qrcode
+ }
+ //测试
+ if ExecPath == "/Users/cdle/Desktop/xdd" {
+ Cdle = true
+ Config.QQID = 17745270
+ Config.QQGroupID = 610790654
+ }
+ if Config.NoGhproxy {
+ GhProxy = ""
+ }
+ if Config.Database == "" {
+ Config.Database = ExecPath + "/.xdd.db"
+ }
+}
diff --git a/models/container.go b/models/container.go
new file mode 100644
index 0000000..f8f5690
--- /dev/null
+++ b/models/container.go
@@ -0,0 +1,491 @@
+package models
+
+import (
+ "bufio"
+ "encoding/json"
+ "errors"
+ "fmt"
+ "io"
+ "io/ioutil"
+ "os"
+ "regexp"
+ "strings"
+
+ "github.com/beego/beego/v2/client/httplib"
+ "github.com/beego/beego/v2/core/logs"
+ "github.com/buger/jsonparser"
+)
+
+const (
+ QL = "ql"
+ V4 = "v4"
+ LI = "li"
+)
+
+func initContainer() {
+ for i := range Config.Containers {
+ if Config.Containers[i].Weigth == 0 {
+ Config.Containers[i].Weigth = 1
+ }
+ Config.Containers[i].Type = ""
+ if Config.Containers[i].Address != "" {
+ vv := regexp.MustCompile(`^(https?://[\.\w]+:?\d*)`).FindStringSubmatch(Config.Containers[i].Address)
+ if len(vv) == 2 {
+ Config.Containers[i].Address = vv[1]
+ } else {
+ logs.Warn("%s地址错误", Config.Containers[i].Type)
+ }
+ version, err := GetQlVersion(Config.Containers[i].Address)
+ if err == nil {
+ if Config.Containers[i].getToken() == nil {
+ logs.Info("青龙" + version + "登录成功")
+ } else {
+ logs.Warn("青龙" + version + "登录失败")
+ }
+ Config.Containers[i].Type = "ql"
+ Config.Containers[i].Version = version
+ } else {
+ if err := Config.Containers[i].getSession(); err == nil {
+ logs.Info("v系登录成功")
+ } else {
+ logs.Info("v系登录失败")
+ }
+ Config.Containers[i].Type = "v4"
+ }
+ } else if Config.Containers[i].Path != "" {
+ f, err := os.Open(Config.Containers[i].Path)
+ if err != nil {
+ logs.Warn("无法打开%s，请检查路径是否正确", Config.Containers[i].Path)
+ } else {
+ rd := bufio.NewReader(f)
+ for {
+ line, err := rd.ReadString('\n') //以'\n'为结束符读入一行
+ if err != nil || io.EOF == err {
+ break
+ }
+ if pt := regexp.MustCompile(`^pt_key=`).FindString(line); pt != "" {
+ Config.Containers[i].Type = "li"
+ break
+ }
+ if pt := regexp.MustCompile(`^Cookie\d+`).FindString(line); pt != "" {
+ Config.Containers[i].Type = "v4"
+ break
+ }
+ if strings.Contains(line, "TempBlockCookie") {
+ Config.Containers[i].Type = "v4"
+ break
+ }
+ if strings.Contains(line, "QYWX_KEY") {
+ Config.Containers[i].Type = "v4"
+ break
+ }
+ }
+ if Config.Containers[i].Type == "" {
+ Config.Containers[i].Type = "li"
+ }
+ f.Close()
+ logs.Info(Config.Containers[i].Type + "配置文件正确")
+ }
+ }
+ }
+
+}
+
+func (c *Container) write(cks []JdCookie) error {
+ switch c.Type {
+ case "ql":
+
+ if c.Version == "2.8" {
+ if len(c.Delete) > 0 {
+ c.request("/api/envs", DELETE, fmt.Sprintf(`[%s]`, strings.Join(c.Delete, ",")))
+ }
+ d := ""
+ if len(cks) != 0 {
+ for _, ck := range cks {
+ if ck.Available == True {
+ d += fmt.Sprintf("pt_key=%s;pt_pin=%s;\\n", ck.PtKey, ck.PtPin)
+ }
+ }
+ c.request("/api/envs", POST, `{"name":"JD_COOKIE","value":"`+d+`"}`)
+
+ type AutoGenerated struct {
+ Code int `json:"code"`
+ Data []struct {
+ Value string `json:"value"`
+ ID string `json:"_id"`
+ Created int64 `json:"created"`
+ Status int `json:"status"`
+ Timestamp string `json:"timestamp"`
+ Position float64 `json:"position"`
+ Name string `json:"name"`
+ Remarks string `json:"remarks,omitempty"`
+ } `json:"data"`
+ }
+ help := getQLHelp(len(cks))
+ for k := range help {
+ var data, err = c.request("/api/envs?searchValue=" + k)
+ a := AutoGenerated{}
+ err = json.Unmarshal(data, &a)
+ if err != nil {
+ continue
+ }
+ toDelete := []string{}
+ for _, env := range a.Data {
+ toDelete = append(toDelete, fmt.Sprintf("\"%s\"", env.ID))
+ }
+ if len(toDelete) > 0 {
+ c.request("/api/envs", DELETE, fmt.Sprintf(`[%s]`, strings.Join(toDelete, ",")))
+ }
+ }
+ for k, v := range help {
+ if v == "" {
+ v = "&"
+ }
+ r := map[string]string{
+ "name": k,
+ "value": v,
+ }
+ d, _ := json.Marshal(r)
+ c.request("/api/envs", POST, string(d))
+ }
+
+ }
+ } else {
+ if len(c.Delete) > 0 {
+ c.request("/api/cookies", DELETE, fmt.Sprintf(`[%s]`, strings.Join(c.Delete, ",")))
+ }
+ d := []string{}
+ for _, ck := range cks {
+ if ck.Available == True {
+ d = append(d, fmt.Sprintf("\"pt_key=%s;pt_pin=%s;\"", ck.PtKey, ck.PtPin))
+ }
+ }
+ if len(d) != 0 {
+ c.request("/api/cookies", POST, fmt.Sprintf(`[%s]`, strings.Join(d, ",")))
+ }
+ }
+ case "v4":
+ return c.postConfig(func(config string) string {
+ TempBlockCookie := ""
+ cookies := ""
+ for i, ck := range cks {
+ if ck.PtPin == "" || ck.PtKey == "" {
+ continue
+ }
+ if ck.Available == False {
+ TempBlockCookie += fmt.Sprintf("%d ", i+1)
+ }
+ cookies += fmt.Sprintf("Cookie%d=\"pt_key=%s;pt_pin=%s;\"\n", i+1, ck.PtKey, ck.PtPin)
+ }
+ config = fmt.Sprintf(`TempBlockCookie="%s"`, TempBlockCookie) + "\n" + cookies + getVhelpRule(len(cks)) + config
+ return config
+ })
+ case "li":
+ config := ""
+ f, err := os.OpenFile(c.Path, os.O_RDWR|os.O_CREATE, 0777) //打开文件 |os.O_RDWR
+ if err != nil {
+ return err
+ }
+ defer f.Close()
+ rd := bufio.NewReader(f)
+ for {
+ line, err := rd.ReadString('\n') //以'\n'为结束符读入一行
+ if err != nil || io.EOF == err {
+ break
+ }
+ if pt := regexp.MustCompile(`^pt_key=(.*);pt_pin=([^'";\s]+);?`).FindStringSubmatch(line); len(pt) != 0 {
+ continue
+ }
+ if pt := regexp.MustCompile(`^pt_key=(.*)`).FindStringSubmatch(line); len(pt) != 0 {
+ continue
+ }
+ config += line
+ }
+ for _, ck := range cks {
+ if ck.PtPin == "" || ck.PtKey == "" {
+ continue
+ }
+ if ck.Available == True {
+ config += fmt.Sprintf("pt_key=%s;pt_pin=%s\n", ck.PtKey, ck.PtPin)
+ }
+ }
+ f.Truncate(0)
+ f.Seek(0, 0)
+ if _, err := io.WriteString(f, config); err != nil {
+ return err
+ }
+ return nil
+ }
+ return nil
+}
+
+func (c *Container) read() error {
+ c.Available = true
+ switch c.Type {
+ case "ql":
+ if c.Version == "2.8" {
+ type AutoGenerated struct {
+ Code int `json:"code"`
+ Data []struct {
+ Value string `json:"value"`
+ ID string `json:"_id"`
+ Created int64 `json:"created"`
+ Status int `json:"status"`
+ Timestamp string `json:"timestamp"`
+ Position float64 `json:"position"`
+ Name string `json:"name"`
+ Remarks string `json:"remarks,omitempty"`
+ } `json:"data"`
+ }
+ var data, err = c.request("/api/envs?searchValue=JD_COOKIE")
+ a := AutoGenerated{}
+ err = json.Unmarshal(data, &a)
+ if err != nil {
+ c.Available = false
+ return err
+ }
+ c.Delete = []string{}
+
+ for _, env := range a.Data {
+ c.Delete = append(c.Delete, fmt.Sprintf("\"%s\"", env.ID))
+ res := regexp.MustCompile(`pt_key=(\S+);pt_pin=([^\s;]+);?`).FindAllStringSubmatch(env.Value, -1)
+ for _, v := range res {
+ CheckIn(v[2], v[1])
+ }
+ }
+ return nil
+ } else {
+ var data, err = c.request("/api/cookies")
+ if err != nil {
+ c.Available = false
+ return err
+ }
+ type AutoGenerated struct {
+ Code int `json:"code"`
+ Data []struct {
+ Value string `json:"value"`
+ ID string `json:"_id"`
+ Created int64 `json:"created"`
+ Status int `json:"status"`
+ Timestamp string `json:"timestamp"`
+ Position float64 `json:"position"`
+ Nickname string `json:"nickname"`
+ } `json:"data"`
+ }
+ var a = AutoGenerated{}
+ c.Delete = []string{}
+ json.Unmarshal(data, &a)
+ for _, vv := range a.Data {
+ c.Delete = append(c.Delete, fmt.Sprintf("\"%s\"", vv.ID))
+ res := regexp.MustCompile(`pt_key=(\S+);pt_pin=([^\s;]+);?`).FindStringSubmatch(vv.Value)
+ if len(res) == 3 {
+ CheckIn(res[2], res[1])
+ }
+ }
+ }
+ case "v4":
+ return c.getConfig(func(rd *bufio.Reader) string {
+ config := ""
+ for {
+ line, err := rd.ReadString('\n') //以'\n'为结束符读入一行
+ if err != nil || io.EOF == err {
+ config += line
+ break
+ }
+ if pt := regexp.MustCompile(`^#?\s?Cookie(\d+)=\S+pt_key=(.+);pt_pin=([^'";\s]+);?`).FindStringSubmatch(line); len(pt) != 0 {
+ CheckIn(pt[3], pt[2])
+ continue
+ }
+ if pt := regexp.MustCompile(`^ForOther`).FindString(line); pt != "" {
+ continue
+ }
+ if pt := regexp.MustCompile(`^My.*\d+=`).FindString(line); pt != "" {
+ continue
+ }
+ if pt := regexp.MustCompile(`^Cookie\d+`).FindString(line); pt != "" {
+ continue
+ }
+ if pt := regexp.MustCompile(`^TempBlockCookie`).FindString(line); pt != "" {
+ continue
+ }
+ config += line
+ }
+ return config
+ })
+ case "li":
+ f, err := os.OpenFile(c.Path, os.O_RDWR|os.O_CREATE, 0777) //打开文件 |os.O_RDWR
+ if err != nil {
+ c.Available = false
+ return err
+ }
+ defer f.Close()
+ rd := bufio.NewReader(f)
+ for {
+ line, err := rd.ReadString('\n') //以'\n'为结束符读入一行
+ if err != nil || io.EOF == err {
+ break
+ }
+ if pt := regexp.MustCompile(`^pt_key=(.+);pt_pin=([^'";\s]+);?`).FindStringSubmatch(line); len(pt) != 0 {
+ CheckIn(pt[2], pt[1])
+ continue
+ }
+ }
+ }
+ return nil
+}
+
+func (c *Container) getToken() error {
+ req := httplib.Post(c.Address + "/api/login")
+ req.Header("Content-Type", "application/json;charset=UTF-8")
+ req.Body(fmt.Sprintf(`{"username":"%s","password":"%s"}`, c.Username, c.Password))
+ if rsp, err := req.Response(); err == nil {
+ data, err := ioutil.ReadAll(rsp.Body)
+ if err != nil {
+ return err
+ }
+ c.Token, _ = jsonparser.GetString(data, "token")
+ } else {
+ return err
+ }
+ return nil
+}
+
+func (c *Container) request(ss ...string) ([]byte, error) {
+ var api, method, body string
+ for _, s := range ss {
+ if s == GET || s == POST || s == PUT || s == DELETE {
+ method = s
+ } else if strings.Contains(s, "/api/") {
+ api = s
+ } else {
+ body = s
+ }
+ }
+ var req *httplib.BeegoHTTPRequest
+ var i = 0
+ for {
+ i++
+ switch method {
+ case POST:
+ req = httplib.Post(c.Address + api)
+ case PUT:
+ req = httplib.Put(c.Address + api)
+ case DELETE:
+ req = httplib.Delete(c.Address + api)
+ default:
+ req = httplib.Get(c.Address + api)
+ }
+ req.Header("Authorization", "Bearer "+c.Token)
+ if body != "" {
+ req.Header("Content-Type", "application/json;charset=UTF-8")
+ req.Body(body)
+ }
+ if data, err := req.Bytes(); err == nil {
+ code, _ := jsonparser.GetInt(data, "code")
+ if code == 200 {
+ return data, nil
+ } else {
+ logs.Warn(string(data))
+ if i >= 5 {
+ return nil, errors.New("异常")
+ }
+ c.getToken()
+ }
+ }
+ }
+ return []byte{}, nil
+}
+
+func GetQlVersion(address string) (string, error) {
+ data, err := httplib.Get(address).String()
+ if err != nil {
+ return "", err
+ }
+ js := regexp.MustCompile(`/umi\.\w+\.js`).FindString(data)
+ if js == "" {
+ return "", errors.New("好像不是青龙面板")
+ }
+ data, err = httplib.Get(address + js).String()
+ if err != nil {
+ return "", err
+ }
+ v := ""
+ if strings.Contains(data, "v2.8") {
+ v = "2.8"
+ } else if strings.Contains(data, "v2.2") {
+ v = "2.2"
+ }
+ return v, nil
+}
+
+const (
+ GET = "GET"
+ POST = "POST"
+ PUT = "PUT"
+ DELETE = "DELETE"
+)
+
+func (c *Container) getConfig(handle func(*bufio.Reader) string) error {
+ if c.Address == "" {
+ f, err := os.OpenFile(c.Path, os.O_RDWR|os.O_CREATE, 0777) //打开文件 |os.O_RDWR
+ if err != nil {
+ return err
+ }
+ defer f.Close()
+ c.Config = handle(bufio.NewReader(f))
+ } else {
+ err := c.getSession()
+ if err != nil {
+ return err
+ }
+ req := httplib.Get(c.Address + "/api/config/config")
+ req.Header("Cookie", c.Token)
+ rsp, err := req.Response()
+ if err != nil {
+ return err
+ }
+ c.Config = handle(bufio.NewReader(rsp.Body))
+ }
+ return nil
+}
+
+func (c *Container) postConfig(handle func(config string) string) error {
+ if c.Address == "" {
+ f, err := os.OpenFile(c.Path, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0777)
+ if err != nil {
+ return err
+ }
+ defer f.Close()
+ f.WriteString(handle(c.Config))
+ } else {
+ req := httplib.Post(c.Address + "/api/save")
+ req.Header("Cookie", c.Token)
+ req.Param("content", handle(c.Config))
+ req.Param("name", "config.sh")
+ _, err := req.Bytes()
+ if err != nil {
+ return err
+ }
+ }
+ return nil
+}
+
+func (c *Container) getSession() error {
+ req := httplib.Post(c.Address + "/auth")
+ req.Param("username", c.Username)
+ req.Param("password", c.Password)
+ rsp, err := req.Response()
+ if err != nil {
+ return err
+ }
+ c.Token = rsp.Header.Get("Set-Cookie")
+ if data, err := ioutil.ReadAll(rsp.Body); err != nil {
+ return err
+ } else {
+ err, _ := jsonparser.GetInt(data, "err")
+ if err != 0 {
+ return errors.New(string(data))
+ }
+ }
+ return nil
+}
diff --git a/models/cron.go b/models/cron.go
new file mode 100644
index 0000000..65ea210
--- /dev/null
+++ b/models/cron.go
@@ -0,0 +1,22 @@
+package models
+
+import (
+ "github.com/beego/beego/v2/adapter/logs"
+ "github.com/robfig/cron/v3"
+)
+
+var c *cron.Cron
+
+func initCron() {
+ c = cron.New()
+ if Config.DailyAssetPushCron != "" {
+ _, err := c.AddFunc(Config.DailyAssetPushCron, DailyAssetsPush)
+ if err != nil {
+ logs.Warn("资产推送任务失败：%v", err)
+ } else {
+ logs.Info("资产推送任务就绪")
+ }
+ c.AddFunc("3 */1 * * *", initVersion)
+ }
+ c.Start()
+}
diff --git a/models/daily_push.go b/models/daily_push.go
new file mode 100644
index 0000000..2640e7f
--- /dev/null
+++ b/models/daily_push.go
@@ -0,0 +1 @@
+package models
diff --git a/models/db.go b/models/db.go
new file mode 100644
index 0000000..f2b1278
--- /dev/null
+++ b/models/db.go
@@ -0,0 +1,255 @@
+package models
+
+import (
+ "fmt"
+ "strings"
+ "time"
+
+ "gorm.io/driver/mysql"
+ "gorm.io/driver/postgres"
+ "gorm.io/driver/sqlite"
+ "gorm.io/gorm"
+ "gorm.io/gorm/logger"
+)
+
+var db *gorm.DB
+var JD_COOKIE = "JD_COOKIE"
+var RECORD = "RECORD"
+var ENV = "env"
+var TASK = "TASK"
+var keys map[string]bool
+var pins map[string]bool
+
+func initDB() {
+ var err error
+ var c = &gorm.Config{
+ Logger: logger.Default.LogMode(logger.Info),
+ }
+ if strings.Contains(Config.Database, "@tcp(") {
+ db, err = gorm.Open(mysql.Open(Config.Database), c)
+ } else if strings.Contains(Config.Database, "dbname=") {
+ db, err = gorm.Open(postgres.Open(Config.Database), c)
+ } else {
+ db, err = gorm.Open(sqlite.Open(Config.Database), c)
+ }
+ if err != nil {
+ panic(err)
+ }
+ db.AutoMigrate(
+ &JdCookie{},
+ &JdCookiePool{},
+ )
+ keys = make(map[string]bool)
+ pins = make(map[string]bool)
+ jps := []JdCookiePool{}
+ db.Find(&jps)
+ for _, jp := range jps {
+ keys[jp.PtKey] = true
+ pins[jp.PtPin] = true
+ }
+}
+
+func HasPin(pin string) bool {
+ if _, ok := pins[pin]; ok {
+ return ok
+ }
+ pins[pin] = true
+ return false
+}
+
+func HasKey(key string) bool {
+ if _, ok := keys[key]; ok {
+ return ok
+ }
+ keys[key] = true
+ return false
+}
+
+type JdCookie struct {
+ ID int `gorm:"column:ID;primaryKey"`
+ Priority int `gorm:"column:Priority;default:1"`
+ CreateAt string `gorm:"column:CreateAt"`
+ PtKey string `gorm:"column:PtKey"`
+ PtPin string `gorm:"column:PtPin;unique"`
+ Note string `gorm:"column:Note"`
+ Available string `gorm:"column:Available;default:true" validate:"oneof=true false"`
+ Nickname string `gorm:"column:Nickname"`
+ BeanNum string `gorm:"column:BeanNum"`
+ QQ int `gorm:"column:QQ"`
+ PushPlus string `gorm:"column:PushPlus"`
+ Telegram int `gorm:"column:Telegram"`
+ Fruit string `gorm:"column:Fruit"`
+ Pet string `gorm:"column:Pet"`
+ Bean string `gorm:"column:Bean"`
+ JdFactory string `gorm:"column:JdFactory"`
+ DreamFactory string `gorm:"column:DreamFactory"`
+ Jxnc string `gorm:"column:Jxnc"`
+ Jdzz string `gorm:"column:Jdzz"`
+ Joy string `gorm:"column:Joy"`
+ Sgmh string `gorm:"column:Sgmh"`
+ Cfd string `gorm:"column:Cfd"`
+ Cash string `gorm:"column:Cash"`
+ Help string `gorm:"column:Help;default:false" validate:"oneof=true false"`
+ Pool string `gorm:"-"`
+}
+
+type JdCookiePool struct {
+ ID int `gorm:"column:ID;primaryKey"`
+ PtKey string `gorm:"column:PtKey;unique"`
+ PtPin string `gorm:"column:PtPin"`
+ LoseAt string `gorm:"column:LoseAt"`
+ CreateAt string `gorm:"column:CreateAt"`
+}
+
+var ScanedAt = "ScanedAt"
+var LoseAt = "LoseAt"
+var CreateAt = "CreateAt"
+var Note = "Note"
+var Available = "Available"
+var UnAvailable = "UnAvailable"
+var PtKey = "PtKey"
+var PtPin = "PtPin"
+var Priority = "Priority"
+var Nickname = "Nickname"
+var BeanNum = "BeanNum"
+var Pool = "Pool"
+var True = "true"
+var False = "false"
+var QQ = "QQ"
+var PushPlus = "PushPlus"
+var Save chan *JdCookie
+var ExecPath string
+var Telegram = "Telegram"
+
+const (
+ Fruit = "Fruit"
+ Pet = "Pet"
+ Bean = "Bean"
+ JdFactory = "JdFactory"
+ DreamFactory = "DreamFactory"
+ Jxnc = "Jxnc"
+ Jdzz = "Jdzz"
+ Joy = "Joy"
+ Sgmh = "Sgmh"
+ Cfd = "Cfd"
+ Cash = "Cash"
+ Help = "Help"
+)
+
+func Date() string {
+ return time.Now().Local().Format("2006-01-02")
+}
+
+func GetJdCookies() []JdCookie {
+ cks := []JdCookie{}
+ db.Order("priority desc").Find(&cks)
+ return cks
+}
+
+func GetJdCookie(pin string) (*JdCookie, error) {
+ ck := &JdCookie{}
+ return ck, db.Where(PtPin+" = ?", pin).First(ck).Error
+}
+
+func (ck *JdCookie) Updates(values interface{}) {
+ if ck.ID != 0 {
+ db.Model(ck).Updates(values)
+ }
+ if ck.PtPin != "" {
+ db.Model(ck).Where(PtPin+" = ?", ck.PtPin).Updates(values)
+ }
+}
+func (ck *JdCookie) Update(column string, value interface{}) {
+ if ck.ID != 0 {
+ db.Model(ck).Update(column, value)
+ }
+ if ck.PtPin != "" {
+ db.Model(ck).Where(PtPin+" = ?", ck.PtPin).Update(column, value)
+ }
+}
+
+func (ck *JdCookie) InPool(pt_key string) error {
+ if ck.ID != 0 {
+ date := Date()
+ tx := db.Begin()
+ jp := &JdCookiePool{}
+ if tx.Where(fmt.Sprintf("%s = '%s' and %s = '%s'", PtPin, ck.PtPin, PtKey, pt_key)).First(jp).Error == nil {
+ return tx.Rollback().Error
+ }
+ if err := tx.Create(&JdCookiePool{
+ PtPin: ck.PtPin,
+ PtKey: pt_key,
+ CreateAt: date,
+ }).Error; err != nil {
+ tx.Rollback()
+ return err
+ }
+ tx.Model(ck).Where(fmt.Sprintf("%s = '%s'", Available, False)).Updates(map[string]interface{}{
+ Available: True,
+ PtKey: pt_key,
+ })
+ return tx.Commit().Error
+ }
+ return nil
+}
+
+func (ck *JdCookie) OutPool() (string, error) {
+ if ck.ID != 0 {
+ date := Date()
+ tx := db.Begin()
+ jp := &JdCookiePool{}
+ tx.Model(jp).Where(fmt.Sprintf("%s = '%s' and %s = '%s'", PtPin, ck.PtPin, PtKey, ck.PtKey)).Update(LoseAt, date)
+ us := map[string]interface{}{}
+ if tx.Where(fmt.Sprintf("%s = '%s' and %s = '%s'", PtPin, ck.PtPin, LoseAt, "")).First(jp).Error != nil {
+ us[Available] = False
+ us[PtKey] = ""
+ } else {
+ us[Available] = True
+ us[PtKey] = jp.PtKey
+ }
+ e := tx.Where(fmt.Sprintf("%s = '%s'", Available, True)).Model(ck).Updates(us).RowsAffected
+ if e == 0 {
+ tx.Rollback()
+ return "", nil
+ }
+ ck.Available = True
+ ck.PtKey = jp.PtKey
+ return jp.PtKey, tx.Commit().Error
+ }
+ return "", nil
+}
+
+func NewJdCookie(ck *JdCookie) error {
+ ck.Priority = Config.DefaultPriority
+ date := Date()
+ ck.CreateAt = date
+ tx := db.Begin()
+ if err := tx.Create(ck).Error; err != nil {
+ tx.Rollback()
+ return err
+ }
+ if err := tx.Create(&JdCookiePool{
+ PtPin: ck.PtPin,
+ PtKey: ck.PtKey,
+ CreateAt: date,
+ }).Error; err != nil {
+ tx.Rollback()
+ return err
+ }
+ return tx.Commit().Error
+}
+
+func CheckIn(pin, key string) int {
+ if !HasPin(pin) {
+ NewJdCookie(&JdCookie{
+ PtKey: key,
+ PtPin: pin,
+ })
+ return 0
+ } else if !HasKey(key) {
+ ck, _ := GetJdCookie(pin)
+ ck.InPool(key)
+ return 1
+ }
+ return 2
+}
diff --git a/models/deamon.go b/models/deamon.go
new file mode 100644
index 0000000..0a6910c
--- /dev/null
+++ b/models/deamon.go
@@ -0,0 +1,60 @@
+package models
+
+import (
+ "fmt"
+ "os"
+ "os/exec"
+ "regexp"
+ "strings"
+
+ "github.com/beego/beego/v2/adapter/logs"
+)
+
+func Daemon() {
+ args := os.Args[1:]
+ execArgs := make([]string, 0)
+ l := len(args)
+ for i := 0; i < l; i++ {
+ if strings.Index(args[i], "-d") == 0 {
+ continue
+ }
+
+ execArgs = append(execArgs, args[i])
+ }
+ proc := exec.Command(os.Args[0], execArgs...)
+ err := proc.Start()
+ if err != nil {
+ panic(err)
+ }
+ logs.Info("小滴滴运行于deamon模式")
+ os.Exit(0)
+}
+
+func killp() {
+ pids, err := ppid()
+ if err == nil {
+ if len(pids) == 0 {
+ return
+ } else {
+ exec.Command("sh", "-c", "kill -9 "+strings.Join(pids, " ")).Output()
+ }
+ } else {
+ return
+ }
+}
+
+func ppid() ([]string, error) {
+ pid := fmt.Sprint(os.Getpid())
+ pids := []string{}
+ rtn, err := exec.Command("sh", "-c", "pidof "+pname).Output()
+ if err != nil {
+ return pids, err
+ }
+ re := regexp.MustCompile(`[\d]+`)
+ for _, v := range re.FindAll(rtn, -1) {
+ if string(v) != pid {
+ pids = append(pids, string(v))
+ }
+ }
+ return pids, nil
+}
diff --git a/models/handle.go b/models/handle.go
new file mode 100644
index 0000000..b27b52e
--- /dev/null
+++ b/models/handle.go
@@ -0,0 +1,139 @@
+package models
+
+import (
+ "math"
+ "strings"
+ "time"
+)
+
+func initHandle() {
+ //获取路径
+ Save = make(chan *JdCookie)
+ go func() {
+ init := true
+ for {
+ get := <-Save
+ if get.Pool == "s" {
+ initCookie()
+ continue
+ }
+ cks := GetJdCookies()
+ tmp := []JdCookie{}
+ for _, ck := range cks {
+ if ck.Priority >= 0 {
+ tmp = append(tmp, ck)
+ }
+ }
+ cks = tmp
+ if Config.Mode == Parallel {
+ for i := range Config.Containers {
+ (&Config.Containers[i]).read()
+ }
+ for i := range Config.Containers {
+ (&Config.Containers[i]).write(cks)
+ }
+ } else {
+ resident := []JdCookie{}
+ if Config.Resident != "" {
+ tmp := cks
+ cks = []JdCookie{}
+ for _, ck := range tmp {
+ if strings.Contains(Config.Resident, ck.PtPin) {
+ resident = append(resident, ck)
+ } else {
+ cks = append(cks, ck)
+ }
+ }
+ }
+ type balance struct {
+ Container Container
+ Weigth float64
+ Ready []JdCookie
+ Should int
+ }
+ availables := []Container{}
+ parallels := []Container{}
+ bs := []balance{}
+ for i := range Config.Containers {
+ (&Config.Containers[i]).read()
+ if Config.Containers[i].Available {
+ if Config.Containers[i].Mode == Parallel {
+ parallels = append(parallels, Config.Containers[i])
+ } else {
+ availables = append(availables, Config.Containers[i])
+ bs = append(bs, balance{
+ Container: Config.Containers[i],
+ Weigth: float64(Config.Containers[i].Weigth),
+ })
+ }
+ }
+ }
+ bat := cks
+ for {
+ left := []JdCookie{}
+ l := len(cks)
+ total := 0.0
+ for i := range bs {
+ total += float64(bs[i].Weigth)
+ }
+ for i := range bs {
+ if bs[i].Weigth == 0 {
+ bs[i].Should = 0
+ } else {
+ bs[i].Should = int(math.Ceil(bs[i].Weigth / total * float64(l)))
+ }
+
+ }
+ a := 0
+ for i := range bs {
+ j := bs[i].Should
+ if j == 0 {
+ continue
+ }
+ s := 0
+ if bs[i].Container.Limit > 0 && j > bs[i].Container.Limit {
+ s = a + bs[i].Container.Limit
+ left = append(left, cks[s:a+j]...)
+ bs[i].Weigth = 0
+ } else {
+ s = a + j
+ }
+ if s > l {
+ s = l
+ }
+ bs[i].Ready = append(bs[i].Ready, cks[a:s]...)
+ a += j
+ if a >= l-1 {
+ break
+ }
+ }
+ if len(left) != 0 {
+ cks = left
+ continue
+ }
+ break
+ }
+
+ for i := range bs {
+ bs[i].Container.write(append(resident, bs[i].Ready...))
+ }
+ for i := range parallels {
+ parallels[i].write(append(resident, bat...))
+ }
+ }
+ if init {
+ go func() {
+ for {
+ Save <- &JdCookie{
+ Pool: "s",
+ }
+
+ time.Sleep(time.Minute * 30)
+ // time.Sleep(time.Second * 1)
+ }
+ }()
+ init = false
+ }
+ }
+ }()
+}
diff --git a/models/help.go b/models/help.go
new file mode 100644
index 0000000..9865f57
--- /dev/null
+++ b/models/help.go
@@ -0,0 +1,182 @@
+package models
+
+import (
+ "fmt"
+ "strings"
+)
+
+func getVhelpRule(num int) string {
+ rules := ""
+ var codes = map[string][]string{
+ "Fruit": {},
+ "Pet": {},
+ "Bean": {},
+ "JdFactory": {},
+ "DreamFactory": {},
+ "Jxnc": {},
+ "Jdzz": {},
+ "Joy": {},
+ "Sgmh": {},
+ "Cfd": {},
+ "Cash": {},
+ }
+ cks := GetJdCookies()
+ for _, ck := range cks {
+ if ck.Help == True || Cdle {
+ for k := range codes {
+ switch k {
+ case "Fruit":
+ codes[k] = append(codes[k], ck.Fruit)
+ case "Pet":
+ codes[k] = append(codes[k], ck.Pet)
+ case "Bean":
+ codes[k] = append(codes[k], ck.Bean)
+ case "JdFactory":
+ codes[k] = append(codes[k], ck.JdFactory)
+ case "DreamFactory":
+ codes[k] = append(codes[k], ck.DreamFactory)
+ case "Jxnc":
+ codes[k] = append(codes[k], ck.Jxnc)
+ case "Jdzz":
+ codes[k] = append(codes[k], ck.Jdzz)
+ case "Joy":
+ codes[k] = append(codes[k], ck.Joy)
+ case "Sgmh":
+ codes[k] = append(codes[k], ck.Sgmh)
+ case "Cfd":
+ codes[k] = append(codes[k], ck.Cfd)
+ case "Cash":
+ codes[k] = append(codes[k], ck.Cash)
+ }
+ if len := len(codes[k]); len != 0 {
+ if codes[k][len-1] == "undefined" || codes[k][len-1] == "" || codes[k][len-1] == "--" {
+ codes[k] = codes[k][:len-1]
+ }
+ }
+ }
+ }
+ }
+ for k := range codes {
+ for i, code := range codes[k] {
+ code = strings.Replace(code, `"`, `\"`, -1)
+ rules += fmt.Sprintf("My%s%d=\"%s\"\n", k, i+1, code)
+ codes[k][i] = fmt.Sprintf("${My%s%d}", k, i+1)
+ }
+ }
+ for k := range codes {
+ for i := 0; i < num; i++ {
+ if len(codes[k]) > 0 {
+ rules += fmt.Sprintf("ForOther"+k+"%d=\"%s\"\n", i+1, strings.Join(codes[k], "@"))
+ }
+ }
+ }
+ return rules
+}
+
+func getQLHelp(num int) map[string]string {
+ var codes = map[string][]string{
+ "Fruit": {},
+ "Pet": {},
+ "Bean": {},
+ "JdFactory": {},
+ "DreamFactory": {},
+ "Jxnc": {},
+ "Jdzz": {},
+ "Joy": {},
+ "Sgmh": {},
+ "Cfd": {},
+ "Cash": {},
+ }
+ cks := GetJdCookies()
+ for _, ck := range cks {
+ if ck.Help == True || Cdle {
+ for k := range codes {
+ switch k {
+ case "Fruit":
+ codes[k] = append(codes[k], ck.Fruit)
+ case "Pet":
+ codes[k] = append(codes[k], ck.Pet)
+ case "Bean":
+ codes[k] = append(codes[k], ck.Bean)
+ case "JdFactory":
+ codes[k] = append(codes[k], ck.JdFactory)
+ case "DreamFactory":
+ codes[k] = append(codes[k], ck.DreamFactory)
+ case "Jxnc":
+ codes[k] = append(codes[k], ck.Jxnc)
+ case "Jdzz":
+ codes[k] = append(codes[k], ck.Jdzz)
+ case "Joy":
+ codes[k] = append(codes[k], ck.Joy)
+ case "Sgmh":
+ codes[k] = append(codes[k], ck.Sgmh)
+ case "Cfd":
+ codes[k] = append(codes[k], ck.Cfd)
+ case "Cash":
+ codes[k] = append(codes[k], ck.Cash)
+ }
+ if len := len(codes[k]); len != 0 {
+ if codes[k][len-1] == "undefined" || codes[k][len-1] == "" || codes[k][len-1] == "--" {
+ codes[k] = codes[k][:len-1]
+ }
+ }
+ }
+ }
+ }
+ var e = map[string]string{
+ "Fruit": "",
+ "Pet": "",
+ "Bean": "",
+ "JdFactory": "",
+ "DreamFactory": "",
+ "Jxnc": "",
+ "Jdzz": "",
+ "Joy": "",
+ "Sgmh": "",
+ "Cfd": "",
+ "Cash": "",
+ }
+ for k := range codes {
+ vv := codes[k]
+ for i := range vv {
+ vv[i] = strings.Replace(vv[i], `"`, `\"`, -1)
+
+ }
+ e[k] += strings.Join(vv, "@")
+ }
+ for k := range e {
+ n := []string{}
+ for i := 0; i < num; i++ {
+ n = append(n, e[k])
+ }
+ e[k] = strings.Join(n, "&")
+ }
+ var f = map[string]string{}
+ for k := range e {
+ switch k {
+ case "Fruit":
+ f["FRUITSHARECODES"] = e[k]
+ case "Pet":
+ f["PETSHARECODES"] = e[k]
+ case "Bean":
+ f["PLANT_BEAN_SHARECODES"] = e[k]
+ case "JdFactory":
+ f["DDFACTORY_SHARECODES"] = e[k]
+ case "DreamFactory":
+ f["DREAM_FACTORY_SHARE_CODES"] = e[k]
+ case "Jxnc":
+ f["JXNC_SHARECODES"] = e[k]
+ // case "Jdzz":
+ // f[k] = e[k]
+ // case "Joy":
+ // f[k] = e[k]
+ case "Sgmh":
+ f["JDSGMH_SHARECODES"] = e[k]
+ // case "Cfd":
+ // f[k] = e[k]
+ case "Cash":
+ f["JD_CASH_SHARECODES"] = e[k]
+ }
+ }
+ return f
+}
diff --git a/models/init.go b/models/init.go
new file mode 100644
index 0000000..7218325
--- /dev/null
+++ b/models/init.go
@@ -0,0 +1,28 @@
+package models
+
+import (
+ "os"
+ "path/filepath"
+
+ "github.com/beego/beego/v2/core/logs"
+)
+
+func init() {
+ killp()
+ for _, arg := range os.Args {
+ if arg == "-d" {
+ Daemon()
+ }
+ }
+ ExecPath, _ = filepath.Abs(filepath.Dir(os.Args[0]))
+ logs.Info("当前%s", ExecPath)
+ initConfig()
+ initDB()
+ go initVersion()
+ go initUserAgent()
+ initContainer()
+ initHandle()
+ initCron()
+ go initTgBot()
+ InitReplies()
+}
diff --git a/models/push.go b/models/push.go
new file mode 100644
index 0000000..bb3b96a
--- /dev/null
+++ b/models/push.go
@@ -0,0 +1,17 @@
+package models
+
+func (ck *JdCookie) Push(msg string) {
+ if Config.QywxKey != "" {
+ go qywxNotify(&QywxConfig{Content: msg})
+ }
+ if Config.TelegramBotToken != "" {
+ go tgBotNotify(msg)
+ }
+ if Config.QQID != 0 {
+ go SendQQ(Config.QQID, msg)
+ go SendQQ(int64(ck.QQ), msg)
+ }
+ if ck.PushPlus != "" {
+ go pushPlus(ck.PushPlus, msg)
+ }
+}
diff --git a/models/pushplus.go b/models/pushplus.go
new file mode 100644
index 0000000..390d628
--- /dev/null
+++ b/models/pushplus.go
@@ -0,0 +1,21 @@
+package models
+
+import (
+ "encoding/json"
+
+ "github.com/beego/beego/v2/client/httplib"
+)
+
+func pushPlus(token string, content string) {
+ req := httplib.Post("http://pushplus.hxtrip.com/send")
+ req.Header("Content-Type", "application/json")
+ data, _ := json.Marshal(struct {
+ Token string `json:"token"`
+ Content string `json:"Content"`
+ }{
+ Token: token,
+ Content: content,
+ })
+ req.Body(data)
+ req.Response()
+}
diff --git a/models/qywx.go b/models/qywx.go
new file mode 100644
index 0000000..9ddf129
--- /dev/null
+++ b/models/qywx.go
@@ -0,0 +1,40 @@
+package models
+
+import (
+ "time"
+
+ "github.com/beego/beego/v2/client/httplib"
+)
+
+type QywxConfig struct {
+ QywxKey string
+ Content string
+}
+
+type QywxNotifyMessage struct {
+ Msgtype string `json:"msgtype"`
+ Text struct {
+ Content string `json:"content"`
+ } `json:"text"`
+}
+
+func qywxNotify(c *QywxConfig) {
+ // if Config.TelegramBotToken == "" {
+
+ // }
+ if c.QywxKey == "" {
+ c.QywxKey = Config.QywxKey
+ }
+ if c.QywxKey == "" {
+ return
+ }
+ wx := QywxNotifyMessage{
+ Msgtype: "text",
+ }
+ wx.Text.Content = c.Content
+ req := httplib.Post("https://qyapi.weixin.qq.com/cgi-bin/webhook/send?key=" + c.QywxKey)
+ req.Header("Content-Type", "application/json")
+ req, _ = req.JSONBody(wx)
+ req.SetTimeout(time.Second*2, time.Second*2)
+ req.Response()
+}
diff --git a/models/sf.go b/models/sf.go
new file mode 100644
index 0000000..de8d93d
--- /dev/null
+++ b/models/sf.go
@@ -0,0 +1,54 @@
+package models
+
+// import (
+// "reflect"
+// "strconv"
+// "strings"
+
+// "github.com/boltdb/bolt"
+// )
+
+// type Env struct {
+// ID int
+// Name string
+// Value string
+// }
+
+// type Bd struct {
+// name string
+// model interface{}
+// }
+
+// func (sf *Bd) Name(name string) error {
+// sf.name = name
+// return nil
+// }
+
+// func (sf *Bd) GetAll(is []interface{}) error {
+// db.View(func(tx *bolt.Tx) error {
+// b := tx.Bucket([]byte(sf.name))
+// b.ForEach(func(_, v []byte) error {
+// i := is[0]
+// var _v = reflect.ValueOf(&ck).Elem()
+// for _, vv := range strings.Split(string(v), ";") {
+// v := strings.Split(vv, "=")
+// if len(v) == 2 {
+// t := _v.FieldByName(v[0])
+// if t.CanSet() {
+// switch t.Kind() {
+// case reflect.Int:
+// i, _ := strconv.Atoi(v[1])
+// t.SetInt(int64(i))
+// case reflect.String:
+// t.SetString(v[1])
+// }
+// }
+// }
+// }
+// is = append(is, i)
+// return nil
+// })
+// return nil
+// })
+// return nil
+// }
diff --git a/models/task.go b/models/task.go
new file mode 100644
index 0000000..2640e7f
--- /dev/null
+++ b/models/task.go
@@ -0,0 +1 @@
+package models
diff --git a/models/tbot.go b/models/tbot.go
new file mode 100644
index 0000000..97a99da
--- /dev/null
+++ b/models/tbot.go
@@ -0,0 +1,57 @@
+package models
+
+import (
+ "net/http"
+ "time"
+
+ "github.com/beego/beego/v2/core/logs"
+ tb "gopkg.in/tucnak/telebot.v2"
+)
+
+var b *tb.Bot
+
+func initTgBot() {
+ go func() {
+ if Config.TelegramBotToken == "" {
+ return
+ }
+ var err error
+ b, err = tb.NewBot(tb.Settings{
+ Token: Config.TelegramBotToken,
+ Poller: &tb.LongPoller{Timeout: 10 * time.Second},
+ })
+ if err != nil {
+ logs.Warn("监听tgbot失败")
+ return
+ }
+ b.Handle(tb.OnText, func(m *tb.Message) {
+ rt := handleMessage(m.Text, "tg", m.Sender.ID)
+ switch rt.(type) {
+ case string:
+ b.Send(m.Sender, rt.(string))
+ case *http.Response:
+ b.SendAlbum(m.Sender, tb.Album{&tb.Photo{File: tb.FromReader(rt.(*http.Response).Body)}})
+ }
+ })
+ logs.Info("监听tgbot")
+ b.Start()
+ }()
+}
+
+func tgBotNotify(msg string) {
+ if b == nil {
+ return
+ }
+ if Config.TelegramUserID == 0 {
+ logs.Warn("tgbot未绑定用id")
+ return
+ }
+ b.Send(&tb.User{ID: Config.TelegramUserID}, msg)
+}
+
+func SendTgMsg(id int, msg string) {
+ if b == nil || id == 0 {
+ return
+ }
+ b.Send(&tb.User{ID: id}, msg)
+}
diff --git a/models/ua.go b/models/ua.go
new file mode 100644
index 0000000..d32dddb
--- /dev/null
+++ b/models/ua.go
@@ -0,0 +1,27 @@
+package models
+
+import (
+ "github.com/beego/beego/v2/adapter/logs"
+ "github.com/beego/beego/v2/client/httplib"
+)
+
+var ua = "Mozilla/5.0 (iPhone; CPU iPhone OS 13_3_1 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Mobile/15E148 SP-engine/2.14.0 main%2F1.0 baiduboxapp/11.18.0.16 (Baidu; P2 13.3.1) NABar/0.0"
+
+func initUserAgent() {
+ if Config.UserAgent != "" {
+ logs.Info("使用自定义User-Agent")
+ ua = Config.UserAgent
+ } else {
+ logs.Info("更新User-Agent")
+ var err error
+ ua, err = httplib.Get(GhProxy + "https://raw.githubusercontent.com/cdle/xdd/main/ua.txt").String()
+ if err != nil {
+ logs.Info("更新User-Agent失败")
+ }
+ }
+
+}
+
+func GetUserAgent() string {
+ return ua
+}
diff --git a/models/version.go b/models/version.go
new file mode 100644
index 0000000..3e68a77
--- /dev/null
+++ b/models/version.go
@@ -0,0 +1,54 @@
+package models
+
+import (
+ "io"
+ "os"
+ "regexp"
+ "runtime"
+
+ "github.com/beego/beego/v2/client/httplib"
+ "github.com/beego/beego/v2/core/logs"
+)
+
+var version = "2021081301"
+var AppName = "xdd"
+var pname = regexp.MustCompile(`/([^/\s]+)`).FindStringSubmatch(os.Args[0])[1]
+
+func initVersion() {
+ if Config.Version != "" {
+ version = Config.Version
+ }
+ logs.Info("检查更新" + version)
+ value, err := httplib.Get(GhProxy + "https://raw.githubusercontent.com/cdle/xdd/main/models/version.go").String()
+ if err != nil {
+ logs.Info("更新User-Agent失败")
+ } else {
+ name := AppName + "_" + runtime.GOOS + "_" + runtime.GOARCH
+ if match := regexp.MustCompile(`var version = "(\d{10})"`).FindStringSubmatch(value); len(match) != 0 {
+ if match[1] > version {
+ logs.Warn("版本过低，下载更新")
+ rsp, err := httplib.Get(GhProxy + "https://github.com/cdle/jd_study/releases/download/main/" + name).Response()
+ if err != nil {
+ logs.Warn("无法下载更新")
+ return
+ }
+ filename := ExecPath + "/." + pname
+ f, err := os.OpenFile(filename, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0777)
+ if err != nil {
+ logs.Warn("无法创建更新临时文件：%v"+filename, err)
+ return
+ }
+ _, err = io.Copy(f, rsp.Body)
+ f.Close()
+ if err != nil {
+ logs.Warn("下载更新失败")
+ return
+ }
+ if err := os.Rename(filename, ExecPath+"/"+pname); err != nil {
+ logs.Warn("移动临时更新文件失败")
+ }
+ Daemon()
+ }
+ }
+ }
+}
diff --git a/qbot/login.go b/qbot/login.go
new file mode 100644
index 0000000..2783738
--- /dev/null
+++ b/qbot/login.go
@@ -0,0 +1,194 @@
+package qbot
+
+import (
+ "bufio"
+ "bytes"
+ "os"
+ "strings"
+ "time"
+
+ qrcodeTerminal "github.com/Baozisoftware/qrcode-terminal-go"
+ "github.com/Mrs4s/MiraiGo/client"
+ "github.com/pkg/errors"
+ log "github.com/sirupsen/logrus"
+ "github.com/tuotoo/qrcode"
+
+ "github.com/Mrs4s/go-cqhttp/global"
+)
+
+var console = bufio.NewReader(os.Stdin)
+
+var readLine = func() (str string) {
+ str, _ = console.ReadString('\n')
+ str = strings.TrimSpace(str)
+ return
+}
+
+var readLineTimeout = func(t time.Duration, de string) (str string) {
+ r := make(chan string)
+ go func() {
+ select {
+ case r <- readLine():
+ case <-time.After(t):
+ }
+ }()
+ str = de
+ select {
+ case str = <-r:
+ case <-time.After(t):
+ }
+ return
+}
+
+var cli *client.QQClient
+
+// ErrSMSRequestError SMS请求出错
+var ErrSMSRequestError = errors.New("sms request error")
+
+func commonLogin() error {
+ res, err := cli.Login()
+ if err != nil {
+ return err
+ }
+ return loginResponseProcessor(res)
+}
+
+func qrcodeLogin() error {
+ rsp, err := cli.FetchQRCode()
+ if err != nil {
+ return err
+ }
+ fi, err := qrcode.Decode(bytes.NewReader(rsp.ImageData))
+ if err != nil {
+ return err
+ }
+ _ = os.WriteFile("qrcode.png", rsp.ImageData, 0o644)
+ defer func() { _ = os.Remove("qrcode.png") }()
+ if cli.Uin != 0 {
+ log.Infof("请使用账号 %v 登录手机QQ扫描二维码 (qrcode.png) : ", cli.Uin)
+ } else {
+ log.Infof("请使用手机QQ扫描二维码 (qrcode.png) : ")
+ }
+ time.Sleep(time.Second)
+ qrcodeTerminal.New().Get(fi.Content).Print()
+ s, err := cli.QueryQRCodeStatus(rsp.Sig)
+ if err != nil {
+ return err
+ }
+ prevState := s.State
+ for {
+ time.Sleep(time.Second)
+ s, _ = cli.QueryQRCodeStatus(rsp.Sig)
+ if s == nil {
+ continue
+ }
+ if prevState == s.State {
+ continue
+ }
+ prevState = s.State
+ switch s.State {
+ case client.QRCodeCanceled:
+ log.Fatalf("扫码被用户取消.")
+ case client.QRCodeTimeout:
+ log.Fatalf("二维码过期")
+ case client.QRCodeWaitingForConfirm:
+ log.Infof("扫码成功, 请在手机端确认登录.")
+ case client.QRCodeConfirmed:
+ res, err := cli.QRCodeLogin(s.LoginInfo)
+ if err != nil {
+ return err
+ }
+ return loginResponseProcessor(res)
+ case client.QRCodeImageFetch, client.QRCodeWaitingForScan:
+ // ignore
+ }
+ }
+}
+
+func loginResponseProcessor(res *client.LoginResponse) error {
+ var err error
+ for {
+ if err != nil {
+ return err
+ }
+ if res.Success {
+ return nil
+ }
+ var text string
+ switch res.Error {
+ case client.SliderNeededError:
+ log.Warnf("登录需要滑条验证码. ")
+ log.Warnf("请参考文档 -> https://docs.go-cqhttp.org/faq/slider.html <- 进行处理")
+ log.Warnf("1. 自行抓包并获取 Ticket 输入.")
+ log.Warnf("2. 使用手机QQ扫描二维码登入. (推荐)")
+ log.Warn("请输入(1 - 2) (将在10秒后自动选择2)：")
+ text = readLineTimeout(time.Second*10, "2")
+ if strings.Contains(text, "1") {
+ println()
+ log.Warnf("请用浏览器打开 -> %v <- 并获取Ticket.", res.VerifyUrl)
+ println()
+ log.Warn("请输入Ticket： (Enter 提交)")
+ text = readLine()
+ res, err = cli.SubmitTicket(text)
+ continue
+ }
+ cli.Disconnect()
+ cli.Release()
+ cli = client.NewClientEmpty()
+ return qrcodeLogin()
+ case client.NeedCaptcha:
+ log.Warnf("登录需要验证码.")
+ _ = os.WriteFile("captcha.jpg", res.CaptchaImage, 0o644)
+ log.Warnf("请输入验证码 (captcha.jpg)： (Enter 提交)")
+ text = readLine()
+ global.DelFile("captcha.jpg")
+ res, err = cli.SubmitCaptcha(text, res.CaptchaSign)
+ continue
+ case client.SMSNeededError:
+ log.Warnf("账号已开启设备锁, 按 Enter 向手机 %v 发送短信验证码.", res.SMSPhone)
+ readLine()
+ if !cli.RequestSMS() {
+ log.Warnf("发送验证码失败，可能是请求过于频繁.")
+ return errors.WithStack(ErrSMSRequestError)
+ }
+ log.Warn("请输入短信验证码： (Enter 提交)")
+ text = readLine()
+ res, err = cli.SubmitSMS(text)
+ continue
+ case client.SMSOrVerifyNeededError:
+ log.Warnf("账号已开启设备锁，请选择验证方式:")
+ log.Warnf("1. 向手机 %v 发送短信验证码", res.SMSPhone)
+ log.Warnf("2. 使用手机QQ扫码验证.")
+ log.Warn("请输入(1 - 2) (将在10秒后自动选择2)：")
+ text = readLineTimeout(time.Second*10, "2")
+ if strings.Contains(text, "1") {
+ if !cli.RequestSMS() {
+ log.Warnf("发送验证码失败，可能是请求过于频繁.")
+ return errors.WithStack(ErrSMSRequestError)
+ }
+ log.Warn("请输入短信验证码： (Enter 提交)")
+ text = readLine()
+ res, err = cli.SubmitSMS(text)
+ continue
+ }
+ fallthrough
+ case client.UnsafeDeviceError:
+ log.Warnf("账号已开启设备锁，请前往 -> %v <- 验证后重启Bot.", res.VerifyUrl)
+ log.Infof("按 Enter 或等待 5s 后继续....")
+ readLineTimeout(time.Second*5, "")
+ os.Exit(0)
+ case client.OtherLoginError, client.UnknownLoginError, client.TooManySMSRequestError:
+ msg := res.ErrorMessage
+ if strings.Contains(msg, "版本") {
+ msg = "密码错误或账号被冻结"
+ }
+ if strings.Contains(msg, "冻结") {
+ log.Fatalf("账号被冻结")
+ }
+ log.Warnf("登录失败: %v", msg)
+ log.Infof("按 Enter 或等待 5s 后继续....")
+ readLineTimeout(time.Second*5, "")
+ os.Exit(0)
+ }
+ }
+}
diff --git a/qbot/main.go b/qbot/main.go
new file mode 100644
index 0000000..88afa2f
--- /dev/null
+++ b/qbot/main.go
@@ -0,0 +1,687 @@
+package qbot
+
+import (
+ "bufio"
+ "bytes"
+ "crypto/aes"
+ "crypto/md5"
+ "crypto/sha1"
+ "encoding/hex"
+ "flag"
+ "fmt"
+ "io/ioutil"
+ "net/http"
+ "os"
+ "os/exec"
+ "path"
+ "path/filepath"
+ "runtime"
+ "strings"
+ "sync"
+ "time"
+
+ "github.com/Mrs4s/go-cqhttp/coolq"
+ "github.com/Mrs4s/go-cqhttp/global"
+ "github.com/Mrs4s/go-cqhttp/global/config"
+ "github.com/cdle/xdd/models"
+
+ // "github.com/Mrs4s/go-cqhttp/global/terminal"
+ "github.com/Mrs4s/go-cqhttp/global/update"
+ "github.com/Mrs4s/go-cqhttp/server"
+
+ "github.com/Mrs4s/MiraiGo/binary"
+ "github.com/Mrs4s/MiraiGo/client"
+ "github.com/Mrs4s/MiraiGo/message"
+ "github.com/guonaihong/gout"
+ rotatelogs "github.com/lestrrat-go/file-rotatelogs"
+ log "github.com/sirupsen/logrus"
+ easy "github.com/t-tomalak/logrus-easy-formatter"
+ "github.com/tidwall/gjson"
+ "golang.org/x/crypto/pbkdf2"
+ "golang.org/x/term"
+)
+
+var (
+ conf *config.Config
+ isFastStart = false
+ // PasswordHash 存储QQ密码哈希供登录使用
+ PasswordHash [16]byte
+
+ // AccountToken 存储AccountToken供登录使用
+ AccountToken []byte
+
+ // 允许通过配置文件设置的状态列表
+ allowStatus = [...]client.UserOnlineStatus{
+ client.StatusOnline, client.StatusAway, client.StatusInvisible, client.StatusBusy,
+ client.StatusListening, client.StatusConstellation, client.StatusWeather, client.StatusMeetSpring,
+ client.StatusTimi, client.StatusEatChicken, client.StatusLoving, client.StatusWangWang, client.StatusCookedRice,
+ client.StatusStudy, client.StatusStayUp, client.StatusPlayBall, client.StatusSignal, client.StatusStudyOnline,
+ client.StatusGaming, client.StatusVacationing, client.StatusWatchingTV, client.StatusFitness,
+ }
+)
+
+var bot *coolq.CQBot
+
+func Main() {
+ time.Sleep(time.Second)
+ models.SendQQ = func(uid int64, msg interface{}) {
+ if bot == nil {
+ return
+ }
+ if uid == 0 {
+ return
+ }
+ switch msg.(type) {
+ case string:
+ if bot != nil {
+ bot.SendPrivateMessage(uid, models.Config.QQGroupID, &message.SendingMessage{Elements: []message.IMessageElement{&message.TextElement{Content: msg.(string)}}})
+ }
+ case *http.Response:
+ data, _ := ioutil.ReadAll(msg.(*http.Response).Body)
+ bot.SendPrivateMessage(uid, models.Config.QQGroupID, &message.SendingMessage{Elements: []message.IMessageElement{&coolq.LocalImageElement{Stream: bytes.NewReader(data)}}})
+ }
+ }
+ models.SendQQGroup = func(gid int64, uid int64, msg interface{}) {
+ if bot == nil {
+ return
+ }
+ switch msg.(type) {
+ case string:
+ if bot != nil {
+ bot.SendGroupMessage(gid, &message.SendingMessage{Elements: []message.IMessageElement{&message.AtElement{Target: uid}, &message.TextElement{Content: msg.(string)}}})
+ }
+ case *http.Response:
+ data, _ := ioutil.ReadAll(msg.(*http.Response).Body)
+ bot.SendGroupMessage(gid, &message.SendingMessage{Elements: []message.IMessageElement{&message.AtElement{Target: uid}, &message.TextElement{Content: "\n"}, &coolq.LocalImageElement{Stream: bytes.NewReader(data)}}})
+ }
+ }
+ coolq.PrivateMessageEventCallback = models.ListenQQPrivateMessage
+ coolq.GroupMessageEventCallback = models.ListenQQGroupMessage
+ // c := flag.String("c", config.DefaultConfigFile, "configuration filename default is config.hjson")
+ // d := flag.Bool("d", false, "running as a daemon")
+ // h := flag.Bool("h", false, "this help")
+ // wd := flag.String("w", "", "cover the working directory")
+ // debug := flag.Bool("D", false, "debug mode")
+ // flag.Parse()
+
+ // switch {
+ // case *h:
+ // help()
+ // case *d:xs
+ // server.Daemon()
+ // case *wd != "":
+ // resetWorkDir(*wd)
+ // }
+
+ // 通过-c 参数替换 配置文件路径
+ // config.DefaultConfigFile = models.ExecPath + "/qbot"
+ // conf = config.Get()
+
+ conf = &config.Config{}
+ // if *debug {
+ // conf.Output.Debug = true
+ // }
+ if conf.Output.Debug {
+ log.SetReportCaller(true)
+ }
+
+ logFormatter := &easy.Formatter{
+ TimestampFormat: "2006-01-02 15:04:05",
+ LogFormat: "[%time%] [%lvl%]: %msg% \n",
+ }
+ rotateOptions := []rotatelogs.Option{
+ rotatelogs.WithRotationTime(time.Hour * 24),
+ }
+
+ if conf.Output.LogAging > 0 {
+ rotateOptions = append(rotateOptions, rotatelogs.WithMaxAge(time.Hour*24*time.Duration(conf.Output.LogAging)))
+ }
+ if conf.Output.LogForceNew {
+ rotateOptions = append(rotateOptions, rotatelogs.ForceNewFile())
+ }
+
+ w, err := rotatelogs.New(path.Join("logs", "%Y-%m-%d.log"), rotateOptions...)
+ if err != nil {
+ log.Errorf("rotatelogs init err: %v", err)
+ panic(err)
+ }
+
+ log.AddHook(global.NewLocalHook(w, logFormatter, global.GetLogLevel(conf.Output.LogLevel)...))
+
+ mkCacheDir := func(path string, _type string) {
+ if !global.PathExists(path) {
+ if err := os.MkdirAll(path, 0o755); err != nil {
+ log.Fatalf("创建%s缓存文件夹失败: %v", _type, err)
+ }
+ }
+ }
+ mkCacheDir(global.ImagePath, "图片")
+ mkCacheDir(global.VoicePath, "语音")
+ mkCacheDir(global.VideoPath, "视频")
+ mkCacheDir(global.CachePath, "发送图片")
+
+ var byteKey []byte
+ arg := os.Args
+ if len(arg) > 1 {
+ for i := range arg {
+ switch arg[i] {
+ case "update":
+ if len(arg) > i+1 {
+ selfUpdate(arg[i+1])
+ } else {
+ selfUpdate("")
+ }
+ case "key":
+ if len(arg) > i+1 {
+ byteKey = []byte(arg[i+1])
+ }
+ case "faststart":
+ isFastStart = true
+ }
+ }
+ }
+ // if terminal.RunningByDoubleClick() && !isFastStart {
+ // log.Warning("警告: 强烈不推荐通过双击直接运行本程序, 这将导致一些非预料的后果.")
+ // log.Warning("将等待10s后启动")
+ // time.Sleep(time.Second * 10)
+ // }
+
+ if (conf.Account.Uin == 0 || (conf.Account.Password == "" && !conf.Account.Encrypt)) && !global.PathExists("session.token") {
+ // log.Warn("账号密码未配置, 将使用二维码登录.")
+ // if !isFastStart {
+ // log.Warn("将在 5秒 后继续.")
+ // time.Sleep(time.Second * 5)
+ // }
+ }
+
+ // log.Info("当前版本:", coolq.Version)
+ if conf.Output.Debug {
+ log.SetLevel(log.DebugLevel)
+ log.Warnf("已开启Debug模式.")
+ // log.Debugf("开发交流群: 192548878")
+ }
+ // log.Info("用户交流群: 721829413")
+ if !global.PathExists("device.json") {
+ log.Warn("虚拟设备信息不存在, 将自动生成随机设备.")
+ client.GenRandomDevice()
+ _ = os.WriteFile("device.json", client.SystemDeviceInfo.ToJson(), 0o644)
+ log.Info("已生成设备信息并保存到 device.json 文件.")
+ } else {
+ log.Info("将使用 device.json 内的设备信息运行Bot.")
+ if err := client.SystemDeviceInfo.ReadJson([]byte(global.ReadAllText("device.json"))); err != nil {
+ log.Fatalf("加载设备信息失败: %v", err)
+ }
+ }
+ if conf.Account.Encrypt {
+ if !global.PathExists("password.encrypt") {
+ if conf.Account.Password == "" {
+ log.Error("无法进行加密，请在配置文件中的添加密码后重新启动.")
+ readLine()
+ os.Exit(0)
+ }
+ log.Infof("密码加密已启用, 请输入Key对密码进行加密: (Enter 提交)")
+ byteKey, _ = term.ReadPassword(int(os.Stdin.Fd()))
+ PasswordHash = md5.Sum([]byte(conf.Account.Password))
+ _ = os.WriteFile("password.encrypt", []byte(PasswordHashEncrypt(PasswordHash[:], byteKey)), 0o644)
+ log.Info("密码已加密，为了您的账号安全，请删除配置文件中的密码后重新启动.")
+ readLine()
+ os.Exit(0)
+ } else {
+ if conf.Account.Password != "" {
+ log.Error("密码已加密，为了您的账号安全，请删除配置文件中的密码后重新启动.")
+ readLine()
+ os.Exit(0)
+ }
+
+ if len(byteKey) == 0 {
+ log.Infof("密码加密已启用, 请输入Key对密码进行解密以继续: (Enter 提交)")
+ cancel := make(chan struct{}, 1)
+ state, _ := term.GetState(int(os.Stdin.Fd()))
+ go func() {
+ select {
+ case <-cancel:
+ return
+ case <-time.After(time.Second * 45):
+ log.Infof("解密key输入超时")
+ time.Sleep(3 * time.Second)
+ _ = term.Restore(int(os.Stdin.Fd()), state)
+ os.Exit(0)
+ }
+ }()
+ byteKey, _ = term.ReadPassword(int(os.Stdin.Fd()))
+ cancel <- struct{}{}
+ } else {
+ log.Infof("密码加密已启用, 使用运行时传递的参数进行解密，按 Ctrl+C 取消.")
+ }
+
+ encrypt, _ := os.ReadFile("password.encrypt")
+ ph, err := PasswordHashDecrypt(string(encrypt), byteKey)
+ if err != nil {
+ log.Fatalf("加密存储的密码损坏，请尝试重新配置密码")
+ }
+ copy(PasswordHash[:], ph)
+ }
+ } else {
+ PasswordHash = md5.Sum([]byte(conf.Account.Password))
+ }
+ // if !isFastStart {
+ // log.Info("Bot将在5秒后登录并开始信息处理, 按 Ctrl+C 取消.")
+ // time.Sleep(time.Second * 5)
+ // }
+ log.Info("开始尝试登录并同步消息...")
+ log.Infof("使用协议: %v", func() string {
+ switch client.SystemDeviceInfo.Protocol {
+ case client.IPad:
+ return "iPad"
+ case client.AndroidPhone:
+ return "Android Phone"
+ case client.AndroidWatch:
+ return "Android Watch"
+ case client.MacOS:
+ return "MacOS"
+ case client.QiDian:
+ return "企点"
+ }
+ return "未知"
+ }())
+ cli = newClient()
+ global.Proxy = conf.Message.ProxyRewrite
+ isQRCodeLogin := (conf.Account.Uin == 0 || len(conf.Account.Password) == 0) && !conf.Account.Encrypt
+ isTokenLogin := false
+ saveToken := func() {
+ AccountToken = cli.GenToken()
+ _ = os.WriteFile("session.token", AccountToken, 0o644)
+ }
+ if global.PathExists("session.token") {
+ token, err := os.ReadFile("session.token")
+ if err == nil {
+ if conf.Account.Uin != 0 {
+ r := binary.NewReader(token)
+ cu := r.ReadInt64()
+ if cu != conf.Account.Uin {
+ log.Warnf("警告: 配置文件内的QQ号 (%v) 与缓存内的QQ号 (%v) 不相同", conf.Account.Uin, cu)
+ log.Warnf("1. 使用会话缓存继续.")
+ log.Warnf("2. 删除会话缓存并重启.")
+ log.Warnf("请选择: (5秒后自动选1)")
+ text := readLineTimeout(time.Second*5, "1")
+ if text == "2" {
+ _ = os.Remove("session.token")
+ os.Exit(0)
+ }
+ }
+ }
+ if err = cli.TokenLogin(token); err != nil {
+ _ = os.Remove("session.token")
+ log.Warnf("恢复会话失败: %v , 尝试使用正常流程登录.", err)
+ time.Sleep(time.Second)
+ cli.Disconnect()
+ cli.Release()
+ cli = newClient()
+ } else {
+ isTokenLogin = true
+ }
+ }
+ }
+ if conf.Account.Uin != 0 && PasswordHash != [16]byte{} {
+ cli.Uin = conf.Account.Uin
+ cli.PasswordMd5 = PasswordHash
+ }
+ if !isTokenLogin {
+ if !isQRCodeLogin {
+ if err := commonLogin(); err != nil {
+ log.Fatalf("登录时发生致命错误: %v", err)
+ }
+ } else {
+ if err := qrcodeLogin(); err != nil {
+ log.Fatalf("登录时发生致命错误: %v", err)
+ }
+ }
+ }
+ var times uint = 1 // 重试次数
+ var reLoginLock sync.Mutex
+ cli.OnDisconnected(func(q *client.QQClient, e *client.ClientDisconnectedEvent) {
+ reLoginLock.Lock()
+ defer reLoginLock.Unlock()
+ times = 1
+ if cli.Online {
+ return
+ }
+ log.Warnf("Bot已离线: %v", e.Message)
+ time.Sleep(time.Second * time.Duration(conf.Account.ReLogin.Delay))
+ for {
+ if conf.Account.ReLogin.Disabled {
+ os.Exit(1)
+ }
+ if times > conf.Account.ReLogin.MaxTimes && conf.Account.ReLogin.MaxTimes != 0 {
+ log.Fatalf("Bot重连次数超过限制, 停止")
+ }
+ times++
+ if conf.Account.ReLogin.Interval > 0 {
+ log.Warnf("将在 %v 秒后尝试重连. 重连次数：%v/%v", conf.Account.ReLogin.Interval, times, conf.Account.ReLogin.MaxTimes)
+ time.Sleep(time.Second * time.Duration(conf.Account.ReLogin.Interval))
+ } else {
+ time.Sleep(time.Second)
+ }
+ log.Warnf("尝试重连...")
+ err := cli.TokenLogin(AccountToken)
+ if err == nil {
+ saveToken()
+ return
+ }
+ log.Warnf("快速重连失败: %v", err)
+ if isQRCodeLogin {
+ log.Fatalf("快速重连失败, 扫码登录无法恢复会话.")
+ }
+ log.Warnf("快速重连失败, 尝试普通登录. 这可能是因为其他端强行T下线导致的.")
+ time.Sleep(time.Second)
+ if err := commonLogin(); err != nil {
+ log.Errorf("登录时发生致命错误: %v", err)
+ } else {
+ saveToken()
+ break
+ }
+ }
+ })
+ saveToken()
+ cli.AllowSlider = true
+ log.Infof("登录成功 欢迎使用: %v", cli.Nickname)
+ log.Info("开始加载好友列表...")
+ global.Check(cli.ReloadFriendList(), true)
+ log.Infof("共加载 %v 个好友.", len(cli.FriendList))
+ log.Infof("开始加载群列表...")
+ global.Check(cli.ReloadGroupList(), true)
+ log.Infof("共加载 %v 个群.", len(cli.GroupList))
+ if conf.Account.Status >= int32(len(allowStatus)) || conf.Account.Status < 0 {
+ conf.Account.Status = 0
+ }
+ cli.SetOnlineStatus(allowStatus[int(conf.Account.Status)])
+ bot = coolq.NewQQBot(cli, conf)
+ _ = bot.Client
+ if conf.Message.PostFormat != "string" && conf.Message.PostFormat != "array" {
+ log.Warnf("post-format 配置错误, 将自动使用 string")
+ coolq.SetMessageFormat("string")
+ } else {
+ coolq.SetMessageFormat(conf.Message.PostFormat)
+ }
+ coolq.IgnoreInvalidCQCode = conf.Message.IgnoreInvalidCQCode
+ coolq.SplitURL = conf.Message.FixURL
+ coolq.ForceFragmented = conf.Message.ForceFragment
+ coolq.RemoveReplyAt = conf.Message.RemoveReplyAt
+ coolq.ExtraReplyData = conf.Message.ExtraReplyData
+ coolq.SkipMimeScan = conf.Message.SkipMimeScan
+ for _, m := range conf.Servers {
+ if h, ok := m["http"]; ok {
+ hc := new(config.HTTPServer)
+ if err := h.Decode(hc); err != nil {
+ log.Warn("读取http配置失败 :", err)
+ } else {
+ go server.RunHTTPServerAndClients(bot, hc)
+ }
+ }
+ if s, ok := m["ws"]; ok {
+ sc := new(config.WebsocketServer)
+ if err := s.Decode(sc); err != nil {
+ log.Warn("读取正向Websocket配置失败 :", err)
+ } else {
+ go server.RunWebSocketServer(bot, sc)
+ }
+ }
+ if c, ok := m["ws-reverse"]; ok {
+ rc := new(config.WebsocketReverse)
+ if err := c.Decode(rc); err != nil {
+ log.Warn("读取反向Websocket配置失败 :", err)
+ } else {
+ go server.RunWebSocketClient(bot, rc)
+ }
+ }
+ if p, ok := m["pprof"]; ok {
+ pc := new(config.PprofServer)
+ if err := p.Decode(pc); err != nil {
+ log.Warn("读取pprof配置失败 :", err)
+ } else {
+ go server.RunPprofServer(pc)
+ }
+ }
+ if p, ok := m["lambda"]; ok {
+ lc := new(config.LambdaServer)
+ if err := p.Decode(lc); err != nil {
+ log.Warn("读取pprof配置失败 :", err)
+ } else {
+ go server.RunLambdaClient(bot, lc)
+ }
+ }
+ }
+ log.Info("资源初始化完成, 开始处理信息.")
+ log.Info("アトリは、高性能ですから!")
+
+ // <-global.SetupMainSignalHandler()
+}
+
+// PasswordHashEncrypt 使用key加密给定passwordHash
+func PasswordHashEncrypt(passwordHash []byte, key []byte) string {
+ if len(passwordHash) != 16 {
+ panic("密码加密参数错误")
+ }
+ key = pbkdf2.Key(key, key, 114514, 32, sha1.New)
+
+ cipher, _ := aes.NewCipher(key)
+ result := make([]byte, 16)
+ cipher.Encrypt(result, passwordHash)
+
+ return hex.EncodeToString(result)
+}
+
+// PasswordHashDecrypt 使用key解密给定passwordHash
+func PasswordHashDecrypt(encryptedPasswordHash string, key []byte) ([]byte, error) {
+ ciphertext, err := hex.DecodeString(encryptedPasswordHash)
+ if err != nil {
+ return nil, err
+ }
+
+ key = pbkdf2.Key(key, key, 114514, 32, sha1.New)
+
+ cipher, _ := aes.NewCipher(key)
+ result := make([]byte, 16)
+ cipher.Decrypt(result, ciphertext)
+
+ return result, nil
+}
+
+func checkUpdate() {
+ log.Infof("正在检查更新.")
+ if coolq.Version == "(devel)" {
+ log.Warnf("检查更新失败: 使用的 Actions 测试版或自编译版本.")
+ return
+ }
+ var res string
+ if err := gout.GET("https://api.github.com/repos/Mrs4s/go-cqhttp/releases/latest").BindBody(&res).Do(); err != nil {
+ log.Warnf("检查更新失败: %v", err)
+ return
+ }
+ info := gjson.Parse(res)
+ if global.VersionNameCompare(coolq.Version, info.Get("tag_name").Str) {
+ log.Infof("当前有更新的 go-cqhttp 可供更新, 请前往 https://github.com/Mrs4s/go-cqhttp/releases 下载.")
+ log.Infof("当前版本: %v 最新版本: %v", coolq.Version, info.Get("tag_name").Str)
+ return
+ }
+ log.Infof("检查更新完成. 当前已运行最新版本.")
+}
+
+func selfUpdate(imageURL string) {
+ log.Infof("正在检查更新.")
+ var res, r string
+ if err := gout.GET("https://api.github.com/repos/Mrs4s/go-cqhttp/releases/latest").BindBody(&res).Do(); err != nil {
+ log.Warnf("检查更新失败: %v", err)
+ return
+ }
+ info := gjson.Parse(res)
+ version := info.Get("tag_name").Str
+ if coolq.Version == version {
+ log.Info("当前版本已经是最新版本!")
+ goto wait
+ }
+ log.Info("当前最新版本为 ", version)
+ log.Warn("是否更新(y/N): ")
+ r = strings.TrimSpace(readLine())
+ if r != "y" && r != "Y" {
+ log.Warn("已取消更新！")
+ } else {
+ log.Info("正在更新,请稍等...")
+ sumURL := fmt.Sprintf("%v/Mrs4s/go-cqhttp/releases/download/%v/go-cqhttp_checksums.txt",
+ func() string {
+ if imageURL != "" {
+ return imageURL
+ }
+ return "https://github.com"
+ }(), version)
+ closer, err := global.HTTPGetReadCloser(sumURL)
+ if err != nil {
+ log.Error("更新失败: ", err)
+ goto wait
+ }
+ rd := bufio.NewReader(closer)
+ binaryName := fmt.Sprintf("go-cqhttp_%v_%v.%v", runtime.GOOS, func() string {
+ if runtime.GOARCH == "arm" {
+ return "armv7"
+ }
+ return runtime.GOARCH
+ }(), func() string {
+ if runtime.GOOS == "windows" {
+ return "zip"
+ }
+ return "tar.gz"
+ }())
+ var sum []byte
+ for {
+ str, err := rd.ReadString('\n')
+ if err != nil {
+ break
+ }
+ str = strings.TrimSpace(str)
+ if strings.HasSuffix(str, binaryName) {
+ sum, _ = hex.DecodeString(strings.TrimSuffix(str, " "+binaryName))
+ break
+ }
+ }
+ url := fmt.Sprintf("%v/Mrs4s/go-cqhttp/releases/download/%v/%v",
+ func() string {
+ if imageURL != "" {
+ return imageURL
+ }
+ return "https://github.com"
+ }(), version, binaryName)
+
+ err = update.Update(url, sum)
+ if err != nil {
+ log.Error("更新失败: ", err)
+ } else {
+ log.Info("更新成功!")
+ }
+ }
+wait:
+ log.Info("按 Enter 继续....")
+ readLine()
+ os.Exit(0)
+}
+
+/*
+func restart(args []string) {
+ var cmd *exec.Cmd
+ if runtime.GOOS == "windows" {
+ file, err := exec.LookPath(args[0])
+ if err != nil {
+ log.Errorf("重启失败:%s", err.Error())
+ return
+ }
+ path, err := filepath.Abs(file)
+ if err != nil {
+ log.Errorf("重启失败:%s", err.Error())
+ }
+ args = append([]string{"/c", "start ", path, "faststart"}, args[1:]...)
+ cmd = &exec.Cmd{
+ Path: "cmd.exe",
+ Args: args,
+ Stderr: os.Stderr,
+ Stdout: os.Stdout,
+ }
+ } else {
+ args = append(args, "faststart")
+ cmd = &exec.Cmd{
+ Path: args[0],
+ Args: args,
+ Stderr: os.Stderr,
+ Stdout: os.Stdout,
+ }
+ }
+ _ = cmd.Start()
+}
+*/
+
+// help cli命令行-h的帮助提示
+func help() {
+ fmt.Printf(`go-cqhttp service
+version: %s
+
+Usage:
+
+server [OPTIONS]
+
+Options:
+`, coolq.Version)
+
+ flag.PrintDefaults()
+ os.Exit(0)
+}
+
+func resetWorkDir(wd string) {
+ args := make([]string, 0, len(os.Args))
+ for i := 1; i < len(os.Args); i++ {
+ if os.Args[i] == "-w" {
+ i++ // skip value field
+ } else if !strings.HasPrefix(os.Args[i], "-w") {
+ args = append(args, os.Args[i])
+ }
+ }
+ p, _ := filepath.Abs(os.Args[0])
+ proc := exec.Command(p, args...)
+ proc.Stdin = os.Stdin
+ proc.Stdout = os.Stdout
+ proc.Stderr = os.Stderr
+ proc.Dir = wd
+ err := proc.Run()
+ if err != nil {
+ panic(err)
+ }
+ os.Exit(0)
+}
+
+func newClient() *client.QQClient {
+ c := client.NewClientEmpty()
+ c.OnServerUpdated(func(bot *client.QQClient, e *client.ServerUpdatedEvent) bool {
+ if !conf.Account.UseSSOAddress {
+ log.Infof("收到服务器地址更新通知, 根据配置文件已忽略.")
+ return false
+ }
+ log.Infof("收到服务器地址更新通知, 将在下一次重连时应用. ")
+ return true
+ })
+ if global.PathExists("address.txt") {
+ log.Infof("检测到 address.txt 文件. 将覆盖目标IP.")
+ addr := global.ReadAddrFile("address.txt")
+ if len(addr) > 0 {
+ cli.SetCustomServer(addr)
+ }
+ log.Infof("读取到 %v 个自定义地址.", len(addr))
+ }
+ c.OnLog(func(c *client.QQClient, e *client.LogEvent) {
+ switch e.Type {
+ case "INFO":
+ log.Info("Protocol -> " + e.Message)
+ case "ERROR":
+ log.Error("Protocol -> " + e.Message)
+ case "DEBUG":
+ log.Debug("Protocol -> " + e.Message)
+ }
+ })
+ return c
+}
diff --git a/run.sh b/run.sh
new file mode 100644
index 0000000..b0d1d32
--- /dev/null
+++ b/run.sh
@@ -0,0 +1,22 @@
+#!/bin/bash
+arch=`uname -m`
+case $arch in
+x86_64)
+ arch="amd64"
+ ;;
+aarch64)
+ arch="arm64"
+ ;;
+*)
+ arch="arm"
+ ;;
+esac
+filename="xdd_linux_${arch}"
+url="https://ghproxy.com/https://github.com/cdle/jd_study/releases/download/main/${filename}"
+dirname="xdd"
+cd $HOME
+if [ ! -d dirname ];then
+ mkdir dirname
+fi
+cd xdd
+curl -L $url -O $filename
diff --git a/static/2233login.af9c53d.png b/static/2233login.af9c53d.png
new file mode 100644
index 0000000..3d9ff56
Binary files /dev/null and b/static/2233login.af9c53d.png differ
diff --git a/static/loadTV.99606e2.gif b/static/loadTV.99606e2.gif
new file mode 100644
index 0000000..33ebcd8
Binary files /dev/null and b/static/loadTV.99606e2.gif differ
diff --git a/static/ok.svg b/static/ok.svg
new file mode 100644
index 0000000..0a116ee
--- /dev/null
+++ b/static/ok.svg
@@ -0,0 +1 @@
+
\ No newline at end of file
diff --git a/static/test.txt b/static/test.txt
new file mode 100644
index 0000000..6769dd6
--- /dev/null
+++ b/static/test.txt
@@ -0,0 +1 @@
+Hello world!
\ No newline at end of file
diff --git a/theme/bidong.html b/theme/bidong.html
new file mode 100644
index 0000000..0eb23fe
--- /dev/null
+++ b/theme/bidong.html
@@ -0,0 +1,161 @@
+
+
+
+
+
+
+
哔咚哔咚
+
+
+
+
+
+
+
+
+
+
+
+
扫描二维码登录
+
+
跳转登录
+
或扫码下载APP
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/theme/kuduan.html b/theme/kuduan.html
new file mode 100644
index 0000000..f80b0d7
--- /dev/null
+++ b/theme/kuduan.html
@@ -0,0 +1,187 @@
+
+
+
+
+
+
+
何必烦忧
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/theme/noodin.html b/theme/noodin.html
new file mode 100644
index 0000000..9d62253
--- /dev/null
+++ b/theme/noodin.html
@@ -0,0 +1,130 @@
+
+
+
+
+
+
+
吃面啦
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/theme/random.html b/theme/random.html
new file mode 100644
index 0000000..e20f79b
--- /dev/null
+++ b/theme/random.html
@@ -0,0 +1,147 @@
+
+
+
+
+
+ JD — 任务助手
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/theme/summer.html b/theme/summer.html
new file mode 100644
index 0000000..9ba8708
--- /dev/null
+++ b/theme/summer.html
@@ -0,0 +1,171 @@
+
+
+
+
+
+
+
JD自动挂机
+
+
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/theme/survey.html b/theme/survey.html
new file mode 100644
index 0000000..c02f063
--- /dev/null
+++ b/theme/survey.html
@@ -0,0 +1,156 @@
+
+
+
+
+
+
问卷调查
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/theme/wave.html b/theme/wave.html
new file mode 100644
index 0000000..d0d8312
--- /dev/null
+++ b/theme/wave.html
@@ -0,0 +1,496 @@
+
+
+
+
+
+
+
+
+
东东邀约
+
+
+
+
+
+
+
+
+
+
+
登录成功
+
tips:
+
+ 京东👉我的👉更多👉(东东农场、东东萌宠、东东工厂、宠汪汪、金榜创造营、领现金、健康社区、种豆得豆)。
+
+ 下载【京喜App】👉我的👉(京喜农场、京喜工厂、京喜财富岛)。
+
+ 下载【京东金融】👉我的👉(种摇钱树、天天提鹅、养猪猪)。
+
每天可领取100+京豆和若干现金红包、优惠券等。
+
京东极速版APP-每天固定1.5汇率左右金币，可兑换红包。
+
一定要微信扫码订阅通知，不然东西好了你也不知道！过期不候
+
我知道了
+
+
+
+ 请使用京东 扫码
+
+
+
+
+
一键登录
+
+
+
+
+
+ 请使用微信 扫码
+
+
+
+
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/ua.txt b/ua.txt
new file mode 100644
index 0000000..62254e2
--- /dev/null
+++ b/ua.txt
@@ -0,0 +1 @@
+Mozilla/5.0 (iPhone; CPU iPhone OS 13_3_1 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Mobile/15E148 SP-engine/2.14.0 main%2F1.0 baiduboxapp/11.18.0.16 (Baidu; P2 13.3.1) NABar/0.0
\ No newline at end of file
+ 
+ 
+ 
欢迎登陆
+