48 lines · 1.5 KB · Go
RawDownload
// Copyright 2022 The Gitea Authors. All rights reserved.// SPDX-License-Identifier: MIT package activitypub import (	"context" 	user_model "forgejo.org/models/user"	"forgejo.org/modules/util") const rsaBits = 3072 // GetKeyPair function returns a user's private and public keysfunc 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	}	priv = settings[user_model.UserActivityPubPrivPem].SettingValue	pub = settings[user_model.UserActivityPubPubPem].SettingValue	return pub, priv, err} // GetPublicKey function returns a user's public keyfunc 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 keyfunc GetPrivateKey(ctx context.Context, user *user_model.User) (priv string, err error) {	_, priv, err = GetKeyPair(ctx, user)	return priv, err}