package mutations

import (
	"context"

	"git.aiterp.net/rpdata/api/graphql/resolver/types"
	"git.aiterp.net/rpdata/api/internal/session"
	"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) {
	user := session.FromContext(ctx).User()
	if user == nil || !user.Permitted("member") {
		return nil, ErrUnauthorized
	}

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

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

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