mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2025-06-24 10:00:50 +00:00
2457f5ff22
- Massive replacement of changing `code.gitea.io/gitea` to `forgejo.org`. - Resolves forgejo/discussions#258 Reviewed-on: https://codeberg.org/forgejo/forgejo/pulls/7337 Reviewed-by: Earl Warren <earl-warren@noreply.codeberg.org> Reviewed-by: Michael Kriese <michael.kriese@gmx.de> Reviewed-by: Beowulf <beowulf@beocode.eu> Reviewed-by: Panagiotis "Ivory" Vasilopoulos <git@n0toose.net> Co-authored-by: Gusted <postmaster@gusted.xyz> Co-committed-by: Gusted <postmaster@gusted.xyz>
70 lines
1.7 KiB
Go
70 lines
1.7 KiB
Go
// Copyright 2023 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package payload
|
|
|
|
import (
|
|
"context"
|
|
|
|
issues_model "forgejo.org/models/issues"
|
|
"forgejo.org/modules/util"
|
|
)
|
|
|
|
const replyPayloadVersion1 byte = 1
|
|
|
|
type payloadReferenceType byte
|
|
|
|
const (
|
|
payloadReferenceIssue payloadReferenceType = iota
|
|
payloadReferenceComment
|
|
)
|
|
|
|
// CreateReferencePayload creates data which GetReferenceFromPayload resolves to the reference again.
|
|
func CreateReferencePayload(reference any) ([]byte, error) {
|
|
var refType payloadReferenceType
|
|
var refID int64
|
|
|
|
switch r := reference.(type) {
|
|
case *issues_model.Issue:
|
|
refType = payloadReferenceIssue
|
|
refID = r.ID
|
|
case *issues_model.Comment:
|
|
refType = payloadReferenceComment
|
|
refID = r.ID
|
|
default:
|
|
return nil, util.NewInvalidArgumentErrorf("unsupported reference type: %T", r)
|
|
}
|
|
|
|
payload, err := util.PackData(refType, refID)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return append([]byte{replyPayloadVersion1}, payload...), nil
|
|
}
|
|
|
|
// GetReferenceFromPayload resolves the reference from the payload
|
|
func GetReferenceFromPayload(ctx context.Context, payload []byte) (any, error) {
|
|
if len(payload) < 1 {
|
|
return nil, util.NewInvalidArgumentErrorf("payload to small")
|
|
}
|
|
|
|
if payload[0] != replyPayloadVersion1 {
|
|
return nil, util.NewInvalidArgumentErrorf("unsupported payload version")
|
|
}
|
|
|
|
var ref payloadReferenceType
|
|
var id int64
|
|
if err := util.UnpackData(payload[1:], &ref, &id); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
switch ref {
|
|
case payloadReferenceIssue:
|
|
return issues_model.GetIssueByID(ctx, id)
|
|
case payloadReferenceComment:
|
|
return issues_model.GetCommentByID(ctx, id)
|
|
default:
|
|
return nil, util.NewInvalidArgumentErrorf("unsupported reference type: %T", ref)
|
|
}
|
|
}
|