api/verify.go

package api

import (
	"adams549659584/go-proxy-bingai/common"
	"adams549659584/go-proxy-bingai/common/helper"
	"encoding/json"
	"net/http"
	"net/url"
	"strings"

	"github.com/Harry-zklcdc/bing-lib/lib/aes"

	binglib "github.com/Harry-zklcdc/bing-lib"
)

var removeCookieName = []string{common.USER_TOKEN_COOKIE_NAME, common.USER_KievRPSSecAuth_COOKIE_NAME, common.USER_RwBf_COOKIE_NAME, common.PASS_SERVER_COOKIE_NAME, common.RAND_COOKIE_INDEX_NAME}

func VerifyHandler(w http.ResponseWriter, r *http.Request) {
	if !helper.CheckAuth(r) {
		helper.UnauthorizedResult(w)
		return
	}

	if r.Method != "GET" {
		helper.CommonResult(w, http.StatusMethodNotAllowed, "Method Not Allowed", nil)
		return
	}

	queryRaw := r.URL.Query()
	IG, _ := url.QueryUnescape(queryRaw.Get("IG"))
	T, _ := url.QueryUnescape(r.URL.Query().Get("T"))
	token, err := aes.Decrypt(T, IG)
	if err != nil {
		helper.ErrorResult(w, http.StatusInternalServerError, "Server Error")
		common.Logger.Error("VerifyHandler Decrypt Error: %v", err)
		return
	}
	if token != common.AUTHOR {
		helper.ErrorResult(w, http.StatusUnavailableForLegalReasons, "T error")
		common.Logger.Error("VerifyHandler T error: %v", token)
		return
	}

	bypassServer := common.BypassServer

	header := http.Header{}
	header.Add("Cookie", r.Header.Get("Cookie"))
	req := &http.Request{
		Header: header,
	}
	if cookie, err := req.Cookie(common.PASS_SERVER_COOKIE_NAME); err == nil {
		if cookie.Value != "" {
			bypassServer = cookie.Value
		}
	}
	reqCookies := []string{}
	for _, cookie := range req.Cookies() {
		if !common.IsInArray(removeCookieName, cookie.Name) {
			reqCookies = append(reqCookies, cookie.String())
		}
	}

	iframeid, _ := url.QueryUnescape(queryRaw.Get("iframeid"))
	convId, _ := url.QueryUnescape(queryRaw.Get("convId"))
	rid, _ := url.QueryUnescape(queryRaw.Get("rid"))
	resp, status, err := binglib.Bypass(bypassServer, strings.Join(reqCookies, "; "), iframeid, IG, convId, rid, T, r.Host)
	if err != nil {
		helper.ErrorResult(w, http.StatusInternalServerError, err.Error())
		common.Logger.Error("VerifyHandler Bypass Error: %v", err)
		return
	}
	if status != http.StatusOK {
		respBytes, err := json.Marshal(resp)
		if err != nil {
			helper.ErrorResult(w, http.StatusInternalServerError, err.Error())
			common.Logger.Error("VerifyHandler Bypass Marshal Error: %v", err)
			return
		}
		helper.ErrorResult(w, status, string(respBytes))
		return
	}

	cookies := strings.Split(resp.Result.Cookies, "; ")
	for _, cookie := range cookies {
		if !common.IsInArray(removeCookieName, strings.Split(cookie, "=")[0]) {
			w.Header().Add("Set-Cookie", cookie+"; path=/")
		}
	}

	helper.CommonResult(w, http.StatusOK, "ok", resp)
}