f6a5b783d2
## Checklist The [contributor guide](https://forgejo.org/docs/next/contributor/) contains information that will be helpful to first time contributors. There also are a few [conditions for merging Pull Requests in Forgejo repositories](https://codeberg.org/forgejo/governance/src/branch/main/PullRequestsAgreement.md). You are also welcome to join the [Forgejo development chatroom](https://matrix.to/#/#forgejo-development:matrix.org). ### Tests - I added test coverage for Go changes... - [x] in their respective `*_test.go` for unit tests. - [ ] in the `tests/integration` directory if it involves interactions with a live Forgejo server. ### Documentation - [ ] I created a pull request [to the documentation](https://codeberg.org/forgejo/docs) to explain to Forgejo users how to use this change. - [ ] I did not document these changes and I do not expect someone else to do it. ### Release notes - [ ] I do not want this change to show in the release notes. - [ ] I want the title to show in the release notes with a link to this pull request. - [ ] I want the content of the `release-notes/<pull request number>.md` to be be used for the release notes instead of the title. Co-authored-by: Michael Jerger <michael.jerger@meissa-gmbh.de> Reviewed-on: https://codeberg.org/forgejo/forgejo/pulls/7203 Reviewed-by: Earl Warren <earl-warren@noreply.codeberg.org> Co-authored-by: zam <mirco.zachmann@meissa.de> Co-committed-by: zam <mirco.zachmann@meissa.de>
78 lines
1.9 KiB
Go
78 lines
1.9 KiB
Go
// Copyright 2024 The Forgejo Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package forgefed
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"forgejo.org/models/db"
|
|
"forgejo.org/modules/validation"
|
|
)
|
|
|
|
func init() {
|
|
db.RegisterModel(new(FederationHost))
|
|
}
|
|
|
|
func GetFederationHost(ctx context.Context, ID int64) (*FederationHost, error) {
|
|
host := new(FederationHost)
|
|
has, err := db.GetEngine(ctx).Where("id=?", ID).Get(host)
|
|
if err != nil {
|
|
return nil, err
|
|
} else if !has {
|
|
return nil, fmt.Errorf("FederationInfo record %v does not exist", ID)
|
|
}
|
|
if res, err := validation.IsValid(host); !res {
|
|
return nil, err
|
|
}
|
|
return host, nil
|
|
}
|
|
|
|
func findFederationHostFromDB(ctx context.Context, searchKey, searchValue string) (*FederationHost, error) {
|
|
host := new(FederationHost)
|
|
has, err := db.GetEngine(ctx).Where(searchKey, searchValue).Get(host)
|
|
if err != nil {
|
|
return nil, err
|
|
} else if !has {
|
|
return nil, nil
|
|
}
|
|
if res, err := validation.IsValid(host); !res {
|
|
return nil, err
|
|
}
|
|
return host, nil
|
|
}
|
|
|
|
func FindFederationHostByFqdnAndPort(ctx context.Context, fqdn string, port uint16) (*FederationHost, error) {
|
|
host := new(FederationHost)
|
|
has, err := db.GetEngine(ctx).Where("host_fqdn=? AND host_port=?", fqdn, port).Get(host)
|
|
if err != nil {
|
|
return nil, err
|
|
} else if !has {
|
|
return nil, nil
|
|
}
|
|
if res, err := validation.IsValid(host); !res {
|
|
return nil, err
|
|
}
|
|
return host, nil
|
|
}
|
|
|
|
func FindFederationHostByKeyID(ctx context.Context, keyID string) (*FederationHost, error) {
|
|
return findFederationHostFromDB(ctx, "key_id=?", keyID)
|
|
}
|
|
|
|
func CreateFederationHost(ctx context.Context, host *FederationHost) error {
|
|
if res, err := validation.IsValid(host); !res {
|
|
return err
|
|
}
|
|
_, err := db.GetEngine(ctx).Insert(host)
|
|
return err
|
|
}
|
|
|
|
func UpdateFederationHost(ctx context.Context, host *FederationHost) error {
|
|
if res, err := validation.IsValid(host); !res {
|
|
return err
|
|
}
|
|
_, err := db.GetEngine(ctx).ID(host.ID).Update(host)
|
|
return err
|
|
}
|