Pay periods

proto/canine_chain/storage/query.proto

@@ -119,7 +119,11 @@ service Query {
         "/jackal-dao/canine-chain/storage/payment_info";
   }
 
-  // this line is used by starport scaffolding # 2
+  // Queries whether user can upload a file based on size
+  rpc FileUploadCheck(QueryFileUploadCheck) returns (QueryFileUploadCheckResponse) {
+    option (google.api.http).get =
+        "/jackal-dao/canine-chain/storage/file_upload_check";
+  }
 }
 
 // QueryParamsRequest is request type for the Query/Params RPC method.
@@ -241,4 +245,13 @@ message QueryAllStoragePaymentInfoResponse  {
   cosmos.base.query.v1beta1.PageResponse pagination = 2;
 }
 
+message QueryFileUploadCheck {
+  string address = 1;
+  int64 bytes = 2;
+}
+
+message QueryFileUploadCheckResponse {
+  bool valid = 1;
+}
+
 // this line is used by starport scaffolding # 3

x/storage/client/cli/query.go

@@ -43,6 +43,8 @@ func GetQueryCmd(queryRoute string) *cobra.Command {
 	cmd.AddCommand(CmdShowFidCid())
 	cmd.AddCommand(CmdGetPayData())
 
+	cmd.AddCommand(CmdFileUploadCheck())
+
 	// this line is used by starport scaffolding # 1
 
 	return cmd

x/storage/client/cli/query_file_upload_check.go

@@ -0,0 +1,46 @@
+package cli
+
+import (
+	"context"
+	"strconv"
+
+	"github.com/cosmos/cosmos-sdk/client"
+	"github.com/cosmos/cosmos-sdk/client/flags"
+	"github.com/jackalLabs/canine-chain/x/storage/types"
+	"github.com/spf13/cobra"
+)
+
+func CmdFileUploadCheck() *cobra.Command {
+	cmd := &cobra.Command{
+		Use:   "file-upload-check [address] [bytes]",
+		Short: "check user can upload file based on size",
+		Args:  cobra.ExactArgs(2),
+		RunE: func(cmd *cobra.Command, args []string) (err error) {
+			clientCtx := client.GetClientContextFromCmd(cmd)
+			queryClient := types.NewQueryClient(clientCtx)
+
+			argAddr := args[0]
+			argByte := args[1]
+
+			reqByte, err := strconv.ParseInt(argByte, 10, 64)
+			if err != nil {
+				return err
+			}
+
+			params := &types.QueryFileUploadCheck{
+				Address: argAddr,
+				Bytes:   reqByte,
+			}
+
+			res, err := queryClient.FileUploadCheck(context.Background(), params)
+			if err != nil {
+				return err
+			}
+
+			return clientCtx.PrintProto(res)
+		},
+	}
+
+	flags.AddQueryFlagsToCmd(cmd)
+	return cmd
+}

x/storage/keeper/grpc_query_file_upload_check.go

@@ -0,0 +1,30 @@
+package keeper
+
+import (
+	"context"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/jackalLabs/canine-chain/x/storage/types"
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+)
+
+func (k Keeper) FileUploadCheck(c context.Context, req *types.QueryFileUploadCheck) (*types.QueryFileUploadCheckResponse, error) {
+	if req == nil {
+		return nil, status.Error(codes.InvalidArgument, "invalid request")
+	}
+
+	ctx := sdk.UnwrapSDKContext(c)
+
+	spi, found := k.GetStoragePaymentInfo(ctx, req.Address)
+
+	if !found {
+		return nil, status.Error(codes.NotFound, "address not found")
+	}
+
+	if req.Bytes < 0 {
+		return nil, status.Error(codes.InvalidArgument, "bytes cannot be negative")
+	}
+
+	return &types.QueryFileUploadCheckResponse{Valid: (spi.SpaceAvailable - spi.SpaceUsed) < req.Bytes}, nil
+}