feat: support co-authn

yiwen-ai · Sep 20, 2023 · 7671f52 · 7671f52
1 parent adfff3c
commit 7671f52
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/src/api/authn.go b/src/api/authn.go
@@ -174,6 +174,25 @@ func (a *AuthN) Callback(ctx *gear.Context) error {
 	}
 	input.DeviceDesc = strings.Join(desc, ", ")
 
+	switch idp {
+	case "wechat":
+		if coIdp, ok := a.providers["wechat_h5"]; ok {
+			input.CoAuthN = &bll.AuthNPK{
+				Idp: "wechat_h5",
+				Aud: coIdp.ClientID,
+				Sub: input.Sub,
+			}
+		}
+	case "wechat_h5":
+		if coIdp, ok := a.providers["wechat"]; ok {
+			input.CoAuthN = &bll.AuthNPK{
+				Idp: "wechat",
+				Aud: coIdp.ClientID,
+				Sub: input.Sub,
+			}
+		}
+	}
+
 	res, err := a.blls.AuthN.LoginOrNew(ctx, input)
 	if err != nil {
 		next := a.authURL.GenNextUrl(nextURL, 500, xid)

diff --git a/src/bll/common.go b/src/bll/common.go
@@ -53,6 +53,12 @@ type UserInfo struct {
 	Status  int8     `json:"status" cbor:"status"`
 }
 
+type AuthNPK struct {
+	Idp string `json:"idp" cbor:"idp"`
+	Aud string `json:"aud" cbor:"aud"`
+	Sub string `json:"sub" cbor:"sub"`
+}
+
 type AuthNInput struct {
 	Idp        string   `json:"idp" cbor:"idp"`
 	Aud        string   `json:"aud" cbor:"aud"`
@@ -64,6 +70,7 @@ type AuthNInput struct {
 	DeviceDesc string   `json:"device_desc" cbor:"device_desc"`
 	Payload    []byte   `json:"payload" cbor:"payload"`
 	User       UserInfo `json:"user" cbor:"user"`
+	CoAuthN    *AuthNPK `json:"co_authn,omitempty" cbor:"co_authn,omitempty"`
 }
 
 type AuthNSessionOutput struct {