package main

import (
	"context"

	"mbase/pkg/client"
	"mbase/pkg/mbctl"
)

func (util *Util) SetGrant(ctx context.Context) (*mbctl.SetGrantResult, error) {
	var err error
	res := &mbctl.SetGrantResult{}
	grpcConn, cli, err := client.NewClient(&util.access)
	if err != nil {
		return res, err
	}
	defer grpcConn.Close()

	params := &mbctl.SetGrantParams{
		Username:  util.username,
		AccountID: util.accountID,
		Operation: util.operation,
	}
	res, err = cli.SetGrant(ctx, params)
	if err != nil {
		return res, err
	}
	if err != nil {
		return res, err
	}
	return res, err
}

func (util *Util) DeleteGrant(ctx context.Context) (*mbctl.DeleteGrantResult, error) {
	var err error
	res := &mbctl.DeleteGrantResult{}
	grpcConn, cli, err := client.NewClient(&util.access)
	if err != nil {
		return res, err
	}
	defer grpcConn.Close()

	params := &mbctl.DeleteGrantParams{
		Username:  util.username,
		AccountID: util.accountID,
		Operation: util.operation,
	}
	res, err = cli.DeleteGrant(ctx, params)
	if err != nil {
		return res, err
	}
	return res, err
}