mirror of
https://github.com/pocketbase/pocketbase.git
synced 2025-03-22 06:51:56 +02:00
95 lines
2.3 KiB
Go
95 lines
2.3 KiB
Go
package auth
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/pocketbase/pocketbase/tools/types"
|
|
"golang.org/x/oauth2"
|
|
"golang.org/x/oauth2/vk"
|
|
)
|
|
|
|
func init() {
|
|
Providers[NameVK] = wrapFactory(NewVKProvider)
|
|
}
|
|
|
|
var _ Provider = (*VK)(nil)
|
|
|
|
// NameVK is the unique name of the VK provider.
|
|
const NameVK string = "vk"
|
|
|
|
// VK allows authentication via VK OAuth2.
|
|
type VK struct {
|
|
BaseProvider
|
|
}
|
|
|
|
// NewVKProvider creates new VK provider instance with some defaults.
|
|
//
|
|
// Docs: https://dev.vk.com/api/oauth-parameters
|
|
func NewVKProvider() *VK {
|
|
return &VK{BaseProvider{
|
|
ctx: context.Background(),
|
|
displayName: "ВКонтакте",
|
|
pkce: false, // VK currently doesn't support PKCE and throws an error if PKCE params are send
|
|
scopes: []string{"email"},
|
|
authURL: vk.Endpoint.AuthURL,
|
|
tokenURL: vk.Endpoint.TokenURL,
|
|
userInfoURL: "https://api.vk.com/method/users.get?fields=photo_max,screen_name&v=5.131",
|
|
}}
|
|
}
|
|
|
|
// FetchAuthUser returns an AuthUser instance based on VK's user api.
|
|
//
|
|
// API reference: https://dev.vk.com/method/users.get
|
|
func (p *VK) FetchAuthUser(token *oauth2.Token) (*AuthUser, error) {
|
|
data, err := p.FetchRawUserInfo(token)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
rawUser := map[string]any{}
|
|
if err := json.Unmarshal(data, &rawUser); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
extracted := struct {
|
|
Response []struct {
|
|
Id int64 `json:"id"`
|
|
FirstName string `json:"first_name"`
|
|
LastName string `json:"last_name"`
|
|
Username string `json:"screen_name"`
|
|
AvatarURL string `json:"photo_max"`
|
|
} `json:"response"`
|
|
}{}
|
|
|
|
if err := json.Unmarshal(data, &extracted); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if len(extracted.Response) == 0 {
|
|
return nil, errors.New("missing response entry")
|
|
}
|
|
|
|
user := &AuthUser{
|
|
Id: strconv.FormatInt(extracted.Response[0].Id, 10),
|
|
Name: strings.TrimSpace(extracted.Response[0].FirstName + " " + extracted.Response[0].LastName),
|
|
Username: extracted.Response[0].Username,
|
|
AvatarURL: extracted.Response[0].AvatarURL,
|
|
RawUser: rawUser,
|
|
AccessToken: token.AccessToken,
|
|
RefreshToken: token.RefreshToken,
|
|
}
|
|
|
|
user.Expiry, _ = types.ParseDateTime(token.Expiry)
|
|
|
|
if email := token.Extra("email"); email != nil {
|
|
user.Email = fmt.Sprint(email)
|
|
}
|
|
|
|
return user, nil
|
|
}
|