266 lines
10 KiB
Go
266 lines
10 KiB
Go
package graph
|
|
|
|
// This file will be automatically regenerated based on the schema, any resolver implementations
|
|
// will be copied through when generating and any unknown code will be moved to the end.
|
|
// Code generated by github.com/99designs/gqlgen version v0.17.40
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"somepi.ddns.net/gitea/gilex-dev/YetAnotherToDoList/globals"
|
|
"somepi.ddns.net/gitea/gilex-dev/YetAnotherToDoList/graph/model"
|
|
"somepi.ddns.net/gitea/gilex-dev/YetAnotherToDoList/server/auth"
|
|
)
|
|
|
|
// CreateUser is the resolver for the createUser field.
|
|
func (r *mutationResolver) CreateUser(ctx context.Context, input model.NewUser) (*model.User, error) {
|
|
// Access managed by directive
|
|
todo, err := globals.DB.CreateUser(input)
|
|
if err != nil {
|
|
globals.Logger.Println("Failed to add new user:", err)
|
|
return nil, errors.New("failed to add new user")
|
|
}
|
|
return todo, nil
|
|
}
|
|
|
|
// CreateTodo is the resolver for the createTodo field.
|
|
func (r *mutationResolver) CreateTodo(ctx context.Context, input model.NewTodo) (*model.Todo, error) {
|
|
if auth.ForContext(ctx).UserId != input.UserID {
|
|
return nil, errors.New("only the owner can create a todo")
|
|
}
|
|
todo, err := globals.DB.CreateTodo(input)
|
|
if err != nil {
|
|
globals.Logger.Println("Failed to add new todo:", err)
|
|
return nil, errors.New("failed to add new todo")
|
|
}
|
|
return todo, nil
|
|
}
|
|
|
|
// CreateRole is the resolver for the createRole field.
|
|
func (r *mutationResolver) CreateRole(ctx context.Context, input model.NewRole) (*model.Role, error) {
|
|
if !auth.ForContext(ctx).IsAdmin {
|
|
return nil, errors.New("only an admin can create a role")
|
|
}
|
|
role, err := globals.DB.CreateRole(&input)
|
|
if err != nil {
|
|
globals.Logger.Println("Failed to add new role:", err)
|
|
return nil, errors.New("failed to add new role")
|
|
}
|
|
return role, nil
|
|
}
|
|
|
|
// CreateRefreshToken is the resolver for the createRefreshToken field.
|
|
func (r *mutationResolver) CreateRefreshToken(ctx context.Context, input model.NewRefreshToken) (*model.RefreshToken, error) {
|
|
// TODO: unify model.RefreshToken & auth.RefreshToken
|
|
userId := auth.ForContext(ctx).UserId
|
|
refreshToken, tokenId, err := globals.DB.IssueRefreshToken(userId, input.TokenName)
|
|
if err != nil {
|
|
globals.Logger.Println("Failed to create refresh token:", err)
|
|
return nil, errors.New("failed to create refresh token")
|
|
}
|
|
return &model.RefreshToken{ID: tokenId, ExpiryDate: refreshToken.ExpiryDate, TokenName: input.TokenName, Selector: &refreshToken.Selector, Token: &refreshToken.Token, UserID: userId}, nil
|
|
}
|
|
|
|
// UpdateTodo is the resolver for the updateTodo field.
|
|
func (r *mutationResolver) UpdateTodo(ctx context.Context, id string, changes model.UpdateTodo) (*model.Todo, error) {
|
|
owner, err := globals.DB.GetTodoOwner(&model.Todo{ID: id})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if owner.ID != auth.ForContext(ctx).UserId && !auth.ForContext(ctx).IsAdmin {
|
|
return nil, errors.New("only the owner can update a todo")
|
|
}
|
|
return globals.DB.UpdateTodo(id, &changes)
|
|
}
|
|
|
|
// UpdateUser is the resolver for the updateUser field.
|
|
func (r *mutationResolver) UpdateUser(ctx context.Context, id string, changes model.UpdateUser) (*model.User, error) {
|
|
if auth.ForContext(ctx).UserId != id {
|
|
return nil, errors.New("can only update yourself")
|
|
}
|
|
return globals.DB.UpdateUser(id, &changes)
|
|
}
|
|
|
|
// UpdateRole is the resolver for the updateRole field.
|
|
func (r *mutationResolver) UpdateRole(ctx context.Context, id string, changes model.UpdateRole) (*model.Role, error) {
|
|
// Access managed by directive
|
|
return globals.DB.UpdateRole(id, &changes)
|
|
}
|
|
|
|
// UpdateRefreshToken is the resolver for the updateRefreshToken field.
|
|
func (r *mutationResolver) UpdateRefreshToken(ctx context.Context, id string, changes model.UpdateRefreshToken) (*model.RefreshToken, error) {
|
|
ownerId, err := globals.DB.GetRefreshTokenOwner(id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if ownerId != auth.ForContext(ctx).UserId && !auth.ForContext(ctx).IsAdmin {
|
|
return nil, errors.New("only the owner can update a refresh token")
|
|
}
|
|
return globals.DB.UpdateRefreshToken(id, &changes)
|
|
}
|
|
|
|
// DeleteUser is the resolver for the deleteUser field.
|
|
func (r *mutationResolver) DeleteUser(ctx context.Context, id string) (*string, error) {
|
|
// Access managed by directive
|
|
return globals.DB.DeleteUser(id)
|
|
}
|
|
|
|
// DeleteTodo is the resolver for the deleteTodo field.
|
|
func (r *mutationResolver) DeleteTodo(ctx context.Context, id string) (*string, error) {
|
|
owner, err := globals.DB.GetTodoOwner(&model.Todo{ID: id})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if owner.ID != auth.ForContext(ctx).UserId && !auth.ForContext(ctx).IsAdmin {
|
|
return nil, errors.New("only the owner can delete a todo")
|
|
}
|
|
return globals.DB.DeleteTodo(id)
|
|
}
|
|
|
|
// DeleteRole is the resolver for the deleteRole field.
|
|
func (r *mutationResolver) DeleteRole(ctx context.Context, id string) (*string, error) {
|
|
// Access managed by directive
|
|
return globals.DB.DeleteRole(id)
|
|
}
|
|
|
|
// DeleteRefreshToken is the resolver for the deleteRefreshToken field.
|
|
func (r *mutationResolver) DeleteRefreshToken(ctx context.Context, id string) (*string, error) {
|
|
ownerId, err := globals.DB.GetRefreshTokenOwner(id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if ownerId != auth.ForContext(ctx).UserId && !auth.ForContext(ctx).IsAdmin {
|
|
return nil, errors.New("only the owner can delete a refresh token")
|
|
}
|
|
return globals.DB.RevokeRefreshToken(id)
|
|
}
|
|
|
|
// AddUserRole is the resolver for the addUserRole field.
|
|
func (r *mutationResolver) AddUserRole(ctx context.Context, userID string, roleID string, userIsRoleManager bool) ([]*model.RelationUserRole, error) {
|
|
// Access managed by directive
|
|
if _, err := globals.DB.AddUserRole(userID, roleID, userIsRoleManager); err != nil {
|
|
return nil, err
|
|
}
|
|
return globals.DB.GetRolesFrom(userID)
|
|
}
|
|
|
|
// UpdateUserRole is the resolver for the UpdateUserRole field.
|
|
func (r *mutationResolver) UpdateUserRole(ctx context.Context, userID string, roleID string, userIsRoleManager bool) ([]*model.RelationUserRole, error) {
|
|
// Access managed by directive
|
|
if _, err := globals.DB.UpdateUserRole(userID, roleID, userIsRoleManager); err != nil {
|
|
return nil, err
|
|
}
|
|
return globals.DB.GetRolesFrom(userID)
|
|
}
|
|
|
|
// RemoveUserRole is the resolver for the RemoveUserRole field.
|
|
func (r *mutationResolver) RemoveUserRole(ctx context.Context, userID string, roleID string) ([]*model.RelationUserRole, error) {
|
|
// Access managed by directive
|
|
if _, err := globals.DB.RemoveUserRole(userID, roleID); err != nil {
|
|
return nil, err
|
|
}
|
|
return globals.DB.GetRolesFrom(userID)
|
|
}
|
|
|
|
// Todos is the resolver for the todos field.
|
|
func (r *queryResolver) Todos(ctx context.Context) ([]*model.Todo, error) {
|
|
// Access managed by directive
|
|
return globals.DB.GetAllTodos()
|
|
}
|
|
|
|
// Users is the resolver for the users field.
|
|
func (r *queryResolver) Users(ctx context.Context) ([]*model.User, error) {
|
|
return globals.DB.GetAllUsers()
|
|
}
|
|
|
|
// Roles is the resolver for the roles field.
|
|
func (r *queryResolver) Roles(ctx context.Context) ([]*model.Role, error) {
|
|
return globals.DB.GetAllRoles()
|
|
}
|
|
|
|
// RefreshTokens is the resolver for the refreshTokens field.
|
|
func (r *queryResolver) RefreshTokens(ctx context.Context) ([]*model.RefreshToken, error) {
|
|
// Access managed by directive
|
|
return globals.DB.GetAllRefreshTokens()
|
|
}
|
|
|
|
// User is the resolver for the user field.
|
|
func (r *queryResolver) User(ctx context.Context, id string) (*model.User, error) {
|
|
return globals.DB.GetUser(&model.User{ID: id})
|
|
}
|
|
|
|
// Todo is the resolver for the todo field.
|
|
func (r *queryResolver) Todo(ctx context.Context, id string) (*model.Todo, error) {
|
|
owner, err := globals.DB.GetTodoOwner(&model.Todo{ID: id})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if owner.ID != auth.ForContext(ctx).UserId && !auth.ForContext(ctx).IsAdmin {
|
|
return nil, errors.New("only the owner can view a todo")
|
|
}
|
|
return globals.DB.GetTodo(&model.Todo{ID: id})
|
|
}
|
|
|
|
// Role is the resolver for the role field.
|
|
func (r *queryResolver) Role(ctx context.Context, id string) (*model.Role, error) {
|
|
return globals.DB.GetRole(&model.Role{ID: id})
|
|
}
|
|
|
|
// RefreshToken is the resolver for the refreshToken field.
|
|
func (r *queryResolver) RefreshToken(ctx context.Context, id string) (*model.RefreshToken, error) {
|
|
ownerId, err := globals.DB.GetRefreshTokenOwner(id)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if ownerId != auth.ForContext(ctx).UserId && !auth.ForContext(ctx).IsAdmin {
|
|
return nil, errors.New("only the owner can view a refresh token")
|
|
}
|
|
return globals.DB.GetRefreshToken(&model.RefreshToken{ID: id, UserID: ownerId})
|
|
}
|
|
|
|
// RoleMembers is the resolver for the roleMembers field.
|
|
func (r *roleResolver) RoleMembers(ctx context.Context, obj *model.Role) ([]*model.RelationRoleUser, error) {
|
|
return globals.DB.GetRoleMembers(obj.ID)
|
|
}
|
|
|
|
// User is the resolver for the user field.
|
|
func (r *todoResolver) User(ctx context.Context, obj *model.Todo) (*model.User, error) {
|
|
// TODO: implement dataloader
|
|
return globals.DB.GetUser(obj.User)
|
|
}
|
|
|
|
// Todos is the resolver for the todos field.
|
|
func (r *userResolver) Todos(ctx context.Context, obj *model.User) ([]*model.Todo, error) {
|
|
if auth.ForContext(ctx).UserId != obj.ID && !auth.ForContext(ctx).IsAdmin {
|
|
return nil, errors.New("only the owner can see this")
|
|
}
|
|
return globals.DB.GetTodosFrom(obj)
|
|
}
|
|
|
|
// Roles is the resolver for the roles field.
|
|
func (r *userResolver) Roles(ctx context.Context, obj *model.User) ([]*model.RelationUserRole, error) {
|
|
return globals.DB.GetRolesFrom(obj.ID)
|
|
}
|
|
|
|
// Mutation returns MutationResolver implementation.
|
|
func (r *Resolver) Mutation() MutationResolver { return &mutationResolver{r} }
|
|
|
|
// Query returns QueryResolver implementation.
|
|
func (r *Resolver) Query() QueryResolver { return &queryResolver{r} }
|
|
|
|
// Role returns RoleResolver implementation.
|
|
func (r *Resolver) Role() RoleResolver { return &roleResolver{r} }
|
|
|
|
// Todo returns TodoResolver implementation.
|
|
func (r *Resolver) Todo() TodoResolver { return &todoResolver{r} }
|
|
|
|
// User returns UserResolver implementation.
|
|
func (r *Resolver) User() UserResolver { return &userResolver{r} }
|
|
|
|
type mutationResolver struct{ *Resolver }
|
|
type queryResolver struct{ *Resolver }
|
|
type roleResolver struct{ *Resolver }
|
|
type todoResolver struct{ *Resolver }
|
|
type userResolver struct{ *Resolver }
|