提交

internal/controller/auth/oauth.go

@@ -0,0 +1,100 @@
+package auth
+
+import (
+	"encoding/json"
+	"github.com/gin-gonic/gin"
+	"net/http"
+	"os"
+	"ripper/internal/app/github_auth"
+	"ripper/internal/cache"
+	"ripper/internal/middleware"
+	"ripper/internal/response"
+	jwtpkg "ripper/pkg/jwt"
+	"time"
+)
+
+type getLoginOauthAuthorizeRequest struct {
+	ClientId    string `json:"client_id" form:"client_id"`
+	Prompt      string `json:"prompt" form:"prompt"`
+	RedirectUri string `json:"redirect_uri" form:"redirect_uri"`
+	Scope       string `json:"scope" form:"scope"`
+	State       string `json:"state" form:"state"`
+}
+
+func getLoginOauthAuthorize(ctx *gin.Context) {
+	req := getLoginOauthAuthorizeRequest{}
+	err := ctx.BindQuery(&req)
+	if err != nil {
+		response.FailJson(ctx, response.FailStruct{
+			Code: -1,
+			Msg:  "Invalid request.",
+		}, false)
+		return
+	}
+
+	vsCopilotClientId := os.Getenv("VS_COPILOT_CLIENT_ID")
+	if req.ClientId != vsCopilotClientId {
+		response.FailJson(ctx, response.FailStruct{
+			Code: -1,
+			Msg:  "Invalid client id.",
+		}, false)
+		return
+	}
+
+	oauthCode := github_auth.GenDevicesCode(20)
+	cai := github_auth.ClientOAuthInfo{
+		ClientId: req.ClientId,
+		Code:     oauthCode,
+		Scope:    req.Scope,
+	}
+	cacheKey := "oauth2_authorize_" + req.ClientId
+	caiInfo, _ := json.Marshal(cai)
+	err = cache.Set(cacheKey, caiInfo, 300)
+	if err != nil {
+		response.FailJson(ctx, response.FailStruct{
+			Code: -1,
+			Msg:  "Internal error.",
+		}, false)
+		return
+	}
+
+	// Redirect to the client's redirect_uri
+	browserSessionId := github_auth.GenDevicesCode(64)
+	ctx.Redirect(302, req.RedirectUri+"?browserSessionId="+browserSessionId+"&code="+oauthCode+"&state="+req.State)
+}
+
+func postLoginOauthAccessTokenForVs2022(ctx *gin.Context) {
+	v, exists := ctx.Get("client_auth_info")
+	if !exists {
+		response.FailJson(ctx, response.FailStruct{
+			Code: -1,
+			Msg:  "Invalid client id.",
+		}, false)
+		return
+	}
+	cliAuthInfo := v.(*github_auth.ClientOAuthInfo)
+	t := time.Now()
+	t.Add(24 * 3 * time.Hour)
+	tk, _ := jwtpkg.CreateToken(&middleware.UserLoad{
+		CardCode:         cliAuthInfo.Code,
+		Client:           cliAuthInfo.ClientId,
+		RegisteredClaims: jwtpkg.CreateStandardClaims(t.Unix(), "user"),
+	})
+	ctx.JSON(http.StatusOK, gin.H{
+		"access_token": tk,
+		"scope":        cliAuthInfo.Scope,
+		"token_type":   "bearer",
+	})
+}
+
+func getSiteSha(ctx *gin.Context) {
+	ctx.Header("X-GitHub-Request-Id", "C0E1:6A1A:1A1F:2A1D:1A1F:1A1F:1A1F:1A1F")
+	ctx.JSON(http.StatusOK, gin.H{})
+}
+
+func getLoginConfig(ctx *gin.Context) {
+	loginPassword := os.Getenv("LOGIN_PASSWORD")
+	ctx.JSON(http.StatusOK, gin.H{
+		"is_login_password": loginPassword != "",
+	})
+}