授权验证

controller/login.go

@@ -3,6 +3,8 @@ package controller
 import (
 	"github.com/gin-gonic/gin"
 	"github.com/taoshihan1991/imaptool/tools"
+	"log"
+	"net/url"
 	"time"
 )
 
@@ -53,3 +55,20 @@ func LoginCheckPass(c *gin.Context) {
 		},
 	})
 }
+//远程请求
+func PostBindOfficial(c *gin.Context) {
+	api := "https://gofly.v1kf.com/2/officialBindIp"
+
+	phone := c.PostForm("phone")
+	password := c.PostForm("password")
+	host := c.Request.Host
+	data := url.Values{}
+	data.Set("phone", phone)
+	data.Set("password", password)
+	data.Set("host", host)
+	res, err := tools.PostForm(api, data)
+	if err != nil {
+		log.Println("绑定官网账户发送认证连接错误")
+	}
+	c.Writer.Write([]byte(res))
+}

go.mod

@@ -19,5 +19,6 @@ require (
 	github.com/satori/go.uuid v1.2.0
 	github.com/sirupsen/logrus v1.4.2
 	github.com/spf13/cobra v0.0.5
+	github.com/tidwall/gjson v1.14.1
 	github.com/zh-five/xdaemon v0.1.1
 )

go.sum

@@ -146,6 +146,12 @@ github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXf
 github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
 github.com/stretchr/testify v1.4.0 h1:2E4SXV/wtOkTonXsotYi4li6zVWxYlZuYNCXe9XRJyk=
 github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
+github.com/tidwall/gjson v1.14.1 h1:iymTbGkQBhveq21bEvAQ81I0LEBork8BFe1CUZXdyuo=
+github.com/tidwall/gjson v1.14.1/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk=
+github.com/tidwall/match v1.1.1 h1:+Ho715JplO36QYgwN9PGYNhgZvoUSc9X2c80KVTi+GA=
+github.com/tidwall/match v1.1.1/go.mod h1:eRSPERbgtNPcGhD8UCthc6PmLEQXEWd3PRB5JTxsfmM=
+github.com/tidwall/pretty v1.2.0 h1:RWIZEg2iJ8/g6fDDYzMpobmaoGh5OLl4AXtGUGPcqCs=
+github.com/tidwall/pretty v1.2.0/go.mod h1:ITEVvHYasfjBbM0u2Pg8T2nJnzm8xPwvNhhsoaGGjNU=
 github.com/ugorji/go v1.1.7 h1:/68gy2h+1mWMrwZFeD1kQialdSzAb432dtpeJ42ovdo=
 github.com/ugorji/go v1.1.7/go.mod h1:kZn38zHttfInRq0xu/PH0az30d+z6vm202qpg1oXVMw=
 github.com/ugorji/go/codec v0.0.0-20181204163529-d75b2dcb6bc8/go.mod h1:VFNgLljTbGfSG7qAOspJ7OScBnGdDN/yBr0sguwnwf0=

middleware/domain_limit.go

@@ -0,0 +1,35 @@
+package middleware
+
+import (
+	"github.com/gin-gonic/gin"
+	"github.com/taoshihan1991/imaptool/tools"
+	"log"
+)
+
+/**
+域名中间件
+*/
+func DomainLimitMiddleware(c *gin.Context) {
+	//离线或者远程
+	if !CheckBindOffcial(c) {
+		c.Abort()
+		return
+	}
+
+}
+
+
+//绑定官网账户
+func CheckBindOffcial(c *gin.Context) bool {
+	res, err := tools.HTTPGet("https://gofly.v1kf.com/2/isBindOfficial")
+	if err != nil {
+		log.Println("离线授权码失败,认证连接失败")
+		c.Redirect(302, "/bind")
+		c.Abort()
+	}
+	if string(res) != "success" {
+		c.Redirect(302, "/bind")
+		c.Abort()
+	}
+	return true
+}

router/api.go

@@ -17,6 +17,8 @@ func InitApiRouter(engine *gin.Engine) {
 		v2.POST("/message", middleware.Ipblack, controller.SendMessageV2)
 		//关闭连接
 		v2.GET("/message_close", controller.SendCloseMessageV2)
+		//绑定
+		v2.POST("/bindOfficial", controller.PostBindOfficial)
 	}
 	engine.GET("/captcha", controller.GetCaptcha)
 	engine.POST("/check", controller.LoginCheckPass)

router/view.go

@@ -12,12 +12,11 @@ func InitViewRouter(engine *gin.Engine) {
 	engine.GET("/install", tmpl.PageInstall)
 	engine.GET("/detail_:page",tmpl.PageDetail)
 	engine.GET("/login", tmpl.PageLogin)
-	engine.GET("/chat_page",tmpl.PageChat)
+	engine.GET("/bind", tmpl.PageBind)
-	engine.GET("/chatIndex",tmpl.PageChat)
+	engine.GET("/chatIndex",middleware.DomainLimitMiddleware,tmpl.PageChat)
-	engine.GET("/chatKfIndex", tmpl.PageKfChat)
 	engine.GET("/main", middleware.JwtPageMiddleware, tmpl.PageMain)
-	engine.GET("/chat_main", middleware.JwtPageMiddleware, tmpl.PageChatMain)
+	engine.GET("/chat_main", middleware.JwtPageMiddleware,middleware.DomainLimitMiddleware, tmpl.PageChatMain)
-	engine.GET("/setting", tmpl.PageSetting)
+	engine.GET("/setting",middleware.DomainLimitMiddleware, tmpl.PageSetting)
 	engine.GET("/setting_statistics", tmpl.PageSettingStatis)
 	engine.GET("/setting_indexpage", tmpl.PageSettingIndexPage)
 	engine.GET("/setting_indexpages", tmpl.PageSettingIndexPages)

static/templates/bind.html

@@ -0,0 +1,134 @@
+<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
+<html lang="cn">
+<head>
+    <meta charset="utf-8">
+    <meta name="renderer" content="webkit">
+    <meta name="viewport" content="width=device-width, initial-scale=1, maximum-scale=1, user-scalable=no">
+    <title></title>
+    <link rel="stylesheet" href="https://cdn.staticfile.org/element-ui/2.15.1/theme-chalk/index.min.css">
+    <script src="https://cdn.staticfile.org/vue/2.6.11/vue.min.js"></script>
+    <script src="https://cdn.staticfile.org/element-ui/2.15.1/index.js"></script>
+    <script src="https://cdn.staticfile.org/jquery/3.6.0/jquery.min.js"></script>
+
+    <style>
+        body {
+            background-color: #2d3a4b;
+            margin: 0;
+            padding: 0;
+        }
+        .clear{clear: both;}
+        .signin {
+            width: 430px;
+            padding: 25px;
+            background: #fff;
+            margin:0 auto;
+            margin-top: 160px;
+            box-shadow: 0 25px 30px -13px rgba(40,40,40,.4);
+            border-radius: 10px;
+        }
+        .signin h1,.signin h2,.signin .copyright{
+            font-weight: normal;
+            color: #4d627b;
+            text-align: center;
+        }
+        .signin .loginTitle{
+            font-size: 24px;
+            margin: 10px 0px;
+        }
+        .signin .loginDesc{
+            font-size: 14px;
+            margin-bottom: 15px;
+        }
+        .signin .loginDesc .regBtn{
+            float: left;
+        }
+        .signin .loginDesc a{
+            text-decoration: none;
+            cursor: pointer;
+            color: #409EFF;
+        }
+        .signin .copyright{
+            font-size: 12px;
+        }
+        @media (max-width: 768px) {
+            body{
+                background-color: #fff;
+            }
+            .signin{
+                width: 90%;
+                margin:100px auto;
+                background-color: #fff;
+                box-shadow:none;
+            }
+        }
+    </style>
+    <link rel="icon" href="/static/images/favicon.ico">
+</head>
+<body>
+<div id="app" class="signin">
+    <template>
+        <el-alert
+                title="请在官网注册账户并绑定手机，在此进行授权验证！"
+                type="warning">
+        </el-alert>
+        <div class="loginHtml">
+            <h1 class="loginTitle">官网账户验证</h1>
+            <el-form :model="kefuForm">
+                <el-form-item  prop="phone">
+                    <el-input v-model="kefuForm.phone" placeholder="手机号"></el-input>
+                </el-form-item>
+                <el-form-item  prop="password">
+                    <el-input show-password v-model="kefuForm.password" placeholder="密码"></el-input>
+                </el-form-item>
+                <el-form-item>
+                    <el-button type="success" style="width: 100%" :loading="loading" @click="kefuLogin('kefuForm')">登录</el-button>
+                </el-form-item>
+            </el-form>
+        </div>
+    </template>
+</div>
+</body>
+<script>
+    new Vue({
+        el: '#app',
+        delimiters:["<{","}>"],
+        data: {
+            window:window,
+            loading:false,
+            kefuForm:{
+                phone:"",
+                password:""
+            }
+        },
+        methods: {
+            kefuLogin(){
+                var _this=this;
+                var data={
+                    phone:this.kefuForm.phone,
+                    password:this.kefuForm.password,
+                };
+                $.post("/2/bindOfficial", data, function (data) {
+                    if (data == "success") {
+                        _this.$message({
+                            message: "认证成功！",
+                            type: 'success'
+                        });
+                    } else {
+                        _this.$message({
+                            message: "认证失败！",
+                            type: 'error'
+                        });
+                    }
+                    _this.loading = false;
+                });
+            }
+        },
+        created: function () {
+
+        }
+    })
+
+</script>
+
+
+</html>

tmpl/login.go

@@ -13,3 +13,7 @@ func PageLogin(c *gin.Context) {
 	}
 	c.HTML(http.StatusOK, "login.html", nil)
 }
+//绑定界面
+func PageBind(c *gin.Context) {
+	c.HTML(http.StatusOK, "bind.html", gin.H{})
+}

tools/http.go

@@ -3,6 +3,7 @@ package tools
 import (
 	"io/ioutil"
 	"net/http"
+	"net/url"
 	"regexp"
 	"strings"
 )
@@ -59,3 +60,16 @@ func IsMobile(userAgent string) bool {
 	}
 	return false
 }
+//发送http post请求数据为form
+func PostForm(url string, data url.Values) (string, error) {
+	resp, err := http.PostForm(url, data)
+	if err != nil {
+		return "", err
+	}
+	defer resp.Body.Close()
+	content, err := ioutil.ReadAll(resp.Body)
+	if err != nil {
+		return "", err
+	}
+	return string(content), nil
+}

tools/http_tool.go

@@ -0,0 +1,198 @@
+package tools
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"encoding/xml"
+	"fmt"
+	"io"
+	"io/ioutil"
+	"mime/multipart"
+	"net/http"
+	"os"
+)
+
+// HTTPGet get 请求
+func HTTPGet(uri string) ([]byte, error) {
+	return HTTPGetContext(context.Background(), uri)
+}
+
+// HTTPGetContext get 请求
+func HTTPGetContext(ctx context.Context, uri string) ([]byte, error) {
+	request, err := http.NewRequestWithContext(ctx, http.MethodGet, uri, nil)
+	if err != nil {
+		return nil, err
+	}
+	response, err := http.DefaultClient.Do(request)
+	if err != nil {
+		return nil, err
+	}
+
+	defer response.Body.Close()
+	if response.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("http get error : uri=%v , statusCode=%v", uri, response.StatusCode)
+	}
+	return ioutil.ReadAll(response.Body)
+}
+
+// HTTPPost post 请求
+func HTTPPost(uri string, data string) ([]byte, error) {
+	return HTTPPostContext(context.Background(), uri, data)
+}
+
+// HTTPPostContext post 请求
+func HTTPPostContext(ctx context.Context, uri string, data string) ([]byte, error) {
+	body := bytes.NewBuffer([]byte(data))
+	request, err := http.NewRequestWithContext(ctx, http.MethodPost, uri, body)
+	if err != nil {
+		return nil, err
+	}
+	response, err := http.DefaultClient.Do(request)
+	if err != nil {
+		return nil, err
+	}
+
+	defer response.Body.Close()
+	if response.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("http get error : uri=%v , statusCode=%v", uri, response.StatusCode)
+	}
+	return ioutil.ReadAll(response.Body)
+}
+
+// PostJSON post json 数据请求
+func PostJSON(uri string, obj interface{}) ([]byte, error) {
+	jsonBuf := new(bytes.Buffer)
+	enc := json.NewEncoder(jsonBuf)
+	enc.SetEscapeHTML(false)
+	err := enc.Encode(obj)
+	if err != nil {
+		return nil, err
+	}
+	response, err := http.Post(uri, "application/json;charset=utf-8", jsonBuf)
+	if err != nil {
+		return nil, err
+	}
+	defer response.Body.Close()
+
+	if response.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("http get error : uri=%v , statusCode=%v", uri, response.StatusCode)
+	}
+	return ioutil.ReadAll(response.Body)
+}
+
+// PostJSONWithRespContentType post json数据请求，且返回数据类型
+func PostJSONWithRespContentType(uri string, obj interface{}) ([]byte, string, error) {
+	jsonBuf := new(bytes.Buffer)
+	enc := json.NewEncoder(jsonBuf)
+	enc.SetEscapeHTML(false)
+	err := enc.Encode(obj)
+	if err != nil {
+		return nil, "", err
+	}
+
+	response, err := http.Post(uri, "application/json;charset=utf-8", jsonBuf)
+	if err != nil {
+		return nil, "", err
+	}
+	defer response.Body.Close()
+
+	if response.StatusCode != http.StatusOK {
+		return nil, "", fmt.Errorf("http get error : uri=%v , statusCode=%v", uri, response.StatusCode)
+	}
+	responseData, err := ioutil.ReadAll(response.Body)
+	contentType := response.Header.Get("Content-Type")
+	return responseData, contentType, err
+}
+
+// PostFile 上传文件
+func PostFile(fieldname, filename, uri string) ([]byte, error) {
+	fields := []MultipartFormField{
+		{
+			IsFile:    true,
+			Fieldname: fieldname,
+			Filename:  filename,
+		},
+	}
+	return PostMultipartForm(fields, uri)
+}
+
+// MultipartFormField 保存文件或其他字段信息
+type MultipartFormField struct {
+	IsFile    bool
+	Fieldname string
+	Value     []byte
+	Filename  string
+}
+
+// PostMultipartForm 上传文件或其他多个字段
+func PostMultipartForm(fields []MultipartFormField, uri string) (respBody []byte, err error) {
+	bodyBuf := &bytes.Buffer{}
+	bodyWriter := multipart.NewWriter(bodyBuf)
+
+	for _, field := range fields {
+		if field.IsFile {
+			fileWriter, e := bodyWriter.CreateFormFile(field.Fieldname, field.Filename)
+			if e != nil {
+				err = fmt.Errorf("error writing to buffer , err=%v", e)
+				return
+			}
+
+			fh, e := os.Open(field.Filename)
+			if e != nil {
+				err = fmt.Errorf("error opening file , err=%v", e)
+				return
+			}
+			defer fh.Close()
+
+			if _, err = io.Copy(fileWriter, fh); err != nil {
+				return
+			}
+		} else {
+			partWriter, e := bodyWriter.CreateFormField(field.Fieldname)
+			if e != nil {
+				err = e
+				return
+			}
+			valueReader := bytes.NewReader(field.Value)
+			if _, err = io.Copy(partWriter, valueReader); err != nil {
+				return
+			}
+		}
+	}
+
+	contentType := bodyWriter.FormDataContentType()
+	bodyWriter.Close()
+
+	resp, e := http.Post(uri, contentType, bodyBuf)
+	if e != nil {
+		err = e
+		return
+	}
+	defer resp.Body.Close()
+	if resp.StatusCode != http.StatusOK {
+		return nil, err
+	}
+	respBody, err = ioutil.ReadAll(resp.Body)
+	return
+}
+
+// PostXML perform a HTTP/POST request with XML body
+func PostXML(uri string, obj interface{}) ([]byte, error) {
+	xmlData, err := xml.Marshal(obj)
+	if err != nil {
+		return nil, err
+	}
+
+	body := bytes.NewBuffer(xmlData)
+	response, err := http.Post(uri, "application/xml;charset=utf-8", body)
+	if err != nil {
+		return nil, err
+	}
+	defer response.Body.Close()
+
+	if response.StatusCode != http.StatusOK {
+		return nil, fmt.Errorf("http code error : uri=%v , statusCode=%v", uri, response.StatusCode)
+	}
+	return ioutil.ReadAll(response.Body)
+}