api/graphql/resolver/mutations/removeFile.go

package mutations

import (
	"context"

	"git.aiterp.net/rpdata/api/graphql/resolver/types"
	"git.aiterp.net/rpdata/api/internal/auth"
	"git.aiterp.net/rpdata/api/model/file"
)

// RemoveFileArgs is an arg
type RemoveFileArgs struct {
	ID string
}

// RemoveFile resolves the removeFIle mutation
func (r *MutationResolver) RemoveFile(ctx context.Context, args *RemoveFileArgs) (*types.FileResolver, error) {
	token := auth.TokenFromContext(ctx)
	if !token.Permitted("member") {
		return nil, ErrUnauthorized
	}

	file, err := file.FindID(args.ID)
	if err != nil {
		return nil, err
	}
	if file.Author != token.UserID && !token.Permitted("file.remove") {
		return nil, ErrUnauthorized
	}

	err = file.Delete()
	if err != nil {
		return nil, err
	}

	return &types.FileResolver{F: file}, nil
}