mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2024-09-08 12:49:13 +02:00
112 lines
2.9 KiB
Go
112 lines
2.9 KiB
Go
|
// Copyright Earl Warren <contact@earl-warren.org>
|
||
|
// Copyright Loïc Dachary <loic@dachary.org>
|
||
|
// SPDX-License-Identifier: MIT
|
||
|
|
||
|
package driver
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"fmt"
|
||
|
|
||
|
"code.gitea.io/gitea/models/db"
|
||
|
org_model "code.gitea.io/gitea/models/organization"
|
||
|
user_model "code.gitea.io/gitea/models/user"
|
||
|
|
||
|
"code.forgejo.org/f3/gof3/v3/f3"
|
||
|
f3_tree "code.forgejo.org/f3/gof3/v3/tree/f3"
|
||
|
"code.forgejo.org/f3/gof3/v3/tree/generic"
|
||
|
f3_util "code.forgejo.org/f3/gof3/v3/util"
|
||
|
)
|
||
|
|
||
|
var _ f3_tree.ForgeDriverInterface = &organization{}
|
||
|
|
||
|
type organization struct {
|
||
|
common
|
||
|
|
||
|
forgejoOrganization *org_model.Organization
|
||
|
}
|
||
|
|
||
|
func (o *organization) SetNative(organization any) {
|
||
|
o.forgejoOrganization = organization.(*org_model.Organization)
|
||
|
}
|
||
|
|
||
|
func (o *organization) GetNativeID() string {
|
||
|
return fmt.Sprintf("%d", o.forgejoOrganization.ID)
|
||
|
}
|
||
|
|
||
|
func (o *organization) NewFormat() f3.Interface {
|
||
|
node := o.GetNode()
|
||
|
return node.GetTree().(f3_tree.TreeInterface).NewFormat(node.GetKind())
|
||
|
}
|
||
|
|
||
|
func (o *organization) ToFormat() f3.Interface {
|
||
|
if o.forgejoOrganization == nil {
|
||
|
return o.NewFormat()
|
||
|
}
|
||
|
return &f3.Organization{
|
||
|
Common: f3.NewCommon(fmt.Sprintf("%d", o.forgejoOrganization.ID)),
|
||
|
Name: o.forgejoOrganization.Name,
|
||
|
FullName: o.forgejoOrganization.FullName,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (o *organization) FromFormat(content f3.Interface) {
|
||
|
organization := content.(*f3.Organization)
|
||
|
o.forgejoOrganization = &org_model.Organization{
|
||
|
ID: f3_util.ParseInt(organization.GetID()),
|
||
|
Name: organization.Name,
|
||
|
FullName: organization.FullName,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (o *organization) Get(ctx context.Context) bool {
|
||
|
node := o.GetNode()
|
||
|
o.Trace("%s", node.GetID())
|
||
|
id := f3_util.ParseInt(string(node.GetID()))
|
||
|
organization, err := org_model.GetOrgByID(ctx, id)
|
||
|
if user_model.IsErrUserNotExist(err) {
|
||
|
return false
|
||
|
}
|
||
|
if err != nil {
|
||
|
panic(fmt.Errorf("organization %v %w", id, err))
|
||
|
}
|
||
|
o.forgejoOrganization = organization
|
||
|
return true
|
||
|
}
|
||
|
|
||
|
func (o *organization) Patch(ctx context.Context) {
|
||
|
o.Trace("%d", o.forgejoOrganization.ID)
|
||
|
if _, err := db.GetEngine(ctx).ID(o.forgejoOrganization.ID).Cols("full_name").Update(o.forgejoOrganization); err != nil {
|
||
|
panic(fmt.Errorf("UpdateOrganizationCols: %v %v", o.forgejoOrganization, err))
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (o *organization) Put(ctx context.Context) generic.NodeID {
|
||
|
node := o.GetNode()
|
||
|
o.Trace("%s", node.GetID())
|
||
|
|
||
|
doer, err := user_model.GetAdminUser(ctx)
|
||
|
if err != nil {
|
||
|
panic(fmt.Errorf("GetAdminUser %w", err))
|
||
|
}
|
||
|
err = org_model.CreateOrganization(ctx, o.forgejoOrganization, doer)
|
||
|
if err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
|
||
|
return generic.NodeID(fmt.Sprintf("%d", o.forgejoOrganization.ID))
|
||
|
}
|
||
|
|
||
|
func (o *organization) Delete(ctx context.Context) {
|
||
|
node := o.GetNode()
|
||
|
o.Trace("%s", node.GetID())
|
||
|
|
||
|
if err := org_model.DeleteOrganization(ctx, o.forgejoOrganization); err != nil {
|
||
|
panic(err)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func newOrganization() generic.NodeDriverInterface {
|
||
|
return &organization{}
|
||
|
}
|