50 lines
1.6 KiB
Go
50 lines
1.6 KiB
Go
// Copyright 2022 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package activitypub
|
|
|
|
import (
|
|
"context"
|
|
|
|
user_model "code.gitea.io/gitea/models/user"
|
|
"code.gitea.io/gitea/modules/util"
|
|
)
|
|
|
|
const rsaBits = 3072
|
|
|
|
// GetKeyPair function returns a user's private and public keys
|
|
func GetKeyPair(ctx context.Context, user *user_model.User) (pub, priv string, err error) {
|
|
var settings map[string]*user_model.Setting
|
|
settings, err = user_model.GetSettings(ctx, user.ID, []string{user_model.UserActivityPubPrivPem, user_model.UserActivityPubPubPem})
|
|
if err != nil {
|
|
return pub, priv, err
|
|
} else if len(settings) == 0 {
|
|
if priv, pub, err = util.GenerateKeyPair(rsaBits); err != nil {
|
|
return pub, priv, err
|
|
}
|
|
if err = user_model.SetUserSetting(ctx, user.ID, user_model.UserActivityPubPrivPem, priv); err != nil {
|
|
return pub, priv, err
|
|
}
|
|
if err = user_model.SetUserSetting(ctx, user.ID, user_model.UserActivityPubPubPem, pub); err != nil {
|
|
return pub, priv, err
|
|
}
|
|
return pub, priv, err
|
|
} else {
|
|
priv = settings[user_model.UserActivityPubPrivPem].SettingValue
|
|
pub = settings[user_model.UserActivityPubPubPem].SettingValue
|
|
return pub, priv, err
|
|
}
|
|
}
|
|
|
|
// GetPublicKey function returns a user's public key
|
|
func GetPublicKey(ctx context.Context, user *user_model.User) (pub string, err error) {
|
|
pub, _, err = GetKeyPair(ctx, user)
|
|
return pub, err
|
|
}
|
|
|
|
// GetPrivateKey function returns a user's private key
|
|
func GetPrivateKey(ctx context.Context, user *user_model.User) (priv string, err error) {
|
|
_, priv, err = GetKeyPair(ctx, user)
|
|
return priv, err
|
|
}
|