v4: adding events (#429)

JackalLabs · Apr 4, 2024 · bb0971d · bb0971d
2 parents 5e4a32c + 6e76ab4
commit bb0971d
Show file tree

Hide file tree

Showing 28 changed files with 549 additions and 9 deletions.
diff --git a/x/filetree/keeper/files.go b/x/filetree/keeper/files.go
@@ -16,6 +16,14 @@ func (k Keeper) SetFiles(ctx sdk.Context, files types.Files) {
 		files.Address,
 		files.Owner,
 	), b)
+
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			types.EventTypeUpdateFile,
+			sdk.NewAttribute(types.AttributeKeyFileAddress, files.Address),
+			sdk.NewAttribute(types.AttributeKeyOwner, files.Owner),
+		),
+	)
 }
 
 // GetFiles returns a files from its index
@@ -49,6 +57,13 @@ func (k Keeper) RemoveFiles(
 		address,
 		ownerAddress,
 	))
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			types.EventTypeRemoveFile,
+			sdk.NewAttribute(types.AttributeKeyFileAddress, address),
+			sdk.NewAttribute(types.AttributeKeyOwner, ownerAddress),
+		),
+	)
 }
 
 // GetAllFiles returns all files

diff --git a/x/filetree/keeper/msg_server_add_editors.go b/x/filetree/keeper/msg_server_add_editors.go
@@ -45,6 +45,19 @@ func (k msgServer) AddEditors(goCtx context.Context, msg *types.MsgAddEditors) (
 	file.EditAccess = newEditors
 
 	k.SetFiles(ctx, file)
-
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+		),
+	)
+
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			types.EventTypeAddEditors,
+			sdk.NewAttribute(types.AttributeKeySigner, msg.Creator),
+			sdk.NewAttribute(types.AttributeKeyFileAddress, msg.Address),
+		),
+	)
 	return &types.MsgAddEditorsResponse{}, nil
 }
diff --git a/x/filetree/keeper/msg_server_add_viewers.go b/x/filetree/keeper/msg_server_add_viewers.go
@@ -45,6 +45,19 @@ func (k msgServer) AddViewers(goCtx context.Context, msg *types.MsgAddViewers) (
 	file.ViewingAccess = newviewers
 
 	k.SetFiles(ctx, file)
-
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+		),
+	)
+
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			types.EventTypeAddViewers,
+			sdk.NewAttribute(types.AttributeKeySigner, msg.Creator),
+			sdk.NewAttribute(types.AttributeKeyFileAddress, msg.Address),
+		),
+	)
 	return &types.MsgAddViewersResponse{}, nil
 }
diff --git a/x/filetree/keeper/msg_server_change_owner.go b/x/filetree/keeper/msg_server_change_owner.go
@@ -37,6 +37,20 @@ func (k msgServer) ChangeOwner(goCtx context.Context, msg *types.MsgChangeOwner)
 	k.SetFiles(ctx, file)
 	// Delete old file
 	k.RemoveFiles(ctx, msg.Address, currentOwner)
-
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+		),
+	)
+
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			types.EventTypeChangeOwner,
+			sdk.NewAttribute(types.AttributeKeySigner, msg.Creator),
+			sdk.NewAttribute(types.AttributeKeyNewOwner, msg.NewOwner),
+			sdk.NewAttribute(types.AttributeKeyFileAddress, msg.Address),
+		),
+	)
 	return &types.MsgChangeOwnerResponse{}, nil
 }
diff --git a/x/filetree/keeper/msg_server_delete_file.go b/x/filetree/keeper/msg_server_delete_file.go
@@ -23,6 +23,11 @@ func (k msgServer) DeleteFile(goCtx context.Context, msg *types.MsgDeleteFile) (
 	}
 
 	k.RemoveFiles(ctx, msg.HashPath, ownerAddress)
-
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+		),
+	)
 	return &types.MsgDeleteFileResponse{}, nil
 }
diff --git a/x/filetree/keeper/msg_server_make_root.go b/x/filetree/keeper/msg_server_make_root.go
@@ -36,6 +36,18 @@ func (k msgServer) ProvisionFileTree(goCtx context.Context, msg *types.MsgProvis
 	ctx := sdk.UnwrapSDKContext(goCtx)
 
 	k.MakeRootFolder(ctx, msg.Creator, msg.Viewers, msg.Editors, msg.TrackingNumber)
-
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+		),
+	)
+
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			types.EventTypeMakeRoot,
+			sdk.NewAttribute(types.AttributeKeySigner, msg.Creator),
+		),
+	)
 	return &types.MsgProvisionFileTreeResponse{}, nil
 }
diff --git a/x/filetree/keeper/msg_server_post_file.go b/x/filetree/keeper/msg_server_post_file.go
@@ -45,5 +45,21 @@ func (k msgServer) PostFile(goCtx context.Context, msg *types.MsgPostFile) (*typ
 
 	k.SetFiles(ctx, file)
 
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+		),
+	)
+
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			types.EventTypePostFile,
+			sdk.NewAttribute(types.AttributeKeySigner, msg.Creator),
+			sdk.NewAttribute(types.AttributeKeyFileAddress, fullMerklePath),
+			sdk.NewAttribute(types.AttributeKeyOwner, owner),
+		),
+	)
+
 	return &types.MsgPostFileResponse{Path: fullMerklePath}, nil
 }
diff --git a/x/filetree/keeper/msg_server_postkey.go b/x/filetree/keeper/msg_server_postkey.go
@@ -15,6 +15,12 @@ func (k msgServer) PostKey(goCtx context.Context, msg *types.MsgPostKey) (*types
 		Key:     msg.Key,
 	}
 	k.SetPubkey(ctx, pubKey)
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+		),
+	)
 
 	return &types.MsgPostKeyResponse{}, nil
 }
diff --git a/x/filetree/keeper/msg_server_remove_editors.go b/x/filetree/keeper/msg_server_remove_editors.go
@@ -45,6 +45,19 @@ func (k msgServer) RemoveEditors(goCtx context.Context, msg *types.MsgRemoveEdit
 	file.EditAccess = newEditors
 
 	k.SetFiles(ctx, file)
-
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+		),
+	)
+
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			types.EventTypeRemoveEditors,
+			sdk.NewAttribute(types.AttributeKeySigner, msg.Creator),
+			sdk.NewAttribute(types.AttributeKeyFileAddress, msg.Address),
+		),
+	)
 	return &types.MsgRemoveEditorsResponse{}, nil
 }
diff --git a/x/filetree/keeper/msg_server_remove_viewers.go b/x/filetree/keeper/msg_server_remove_viewers.go
@@ -45,6 +45,19 @@ func (k msgServer) RemoveViewers(goCtx context.Context, msg *types.MsgRemoveView
 	file.ViewingAccess = newviewers
 
 	k.SetFiles(ctx, file)
-
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+		),
+	)
+
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			types.EventTypeRemoveViewers,
+			sdk.NewAttribute(types.AttributeKeySigner, msg.Creator),
+			sdk.NewAttribute(types.AttributeKeyFileAddress, msg.Address),
+		),
+	)
 	return &types.MsgRemoveViewersResponse{}, nil
 }
diff --git a/x/filetree/keeper/msg_server_reset_editors.go b/x/filetree/keeper/msg_server_reset_editors.go
@@ -44,6 +44,19 @@ func (k msgServer) ResetEditors(goCtx context.Context, msg *types.MsgResetEditor
 	file.EditAccess = newEditors
 
 	k.SetFiles(ctx, file)
-
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+		),
+	)
+
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			types.EventTypeResetEditors,
+			sdk.NewAttribute(types.AttributeKeySigner, msg.Creator),
+			sdk.NewAttribute(types.AttributeKeyFileAddress, msg.Address),
+		),
+	)
 	return &types.MsgResetEditorsResponse{}, nil
 }
diff --git a/x/filetree/keeper/msg_server_reset_viewers.go b/x/filetree/keeper/msg_server_reset_viewers.go
@@ -45,5 +45,20 @@ func (k msgServer) ResetViewers(goCtx context.Context, msg *types.MsgResetViewer
 
 	k.SetFiles(ctx, file)
 
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+		),
+	)
+
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			types.EventTypeResetViewers,
+			sdk.NewAttribute(types.AttributeKeySigner, msg.Creator),
+			sdk.NewAttribute(types.AttributeKeyFileAddress, msg.Address),
+		),
+	)
+
 	return &types.MsgResetViewersResponse{}, nil
 }
diff --git a/x/filetree/keeper/pubkey.go b/x/filetree/keeper/pubkey.go
@@ -13,6 +13,14 @@ func (k Keeper) SetPubkey(ctx sdk.Context, pubkey types.Pubkey) {
 	store.Set(types.PubkeyKey(
 		pubkey.Address,
 	), b)
+
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			types.EventTypePostKey,
+			sdk.NewAttribute(types.AttributeKeySigner, pubkey.Address),
+			sdk.NewAttribute(types.AttributeKeyKey, pubkey.Key),
+		),
+	)
 }
 
 // GetPubkey returns a pubkey from its index

diff --git a/x/filetree/types/events.go b/x/filetree/types/events.go
@@ -0,0 +1,27 @@
+package types
+
+// filetree module event types
+const (
+	EventTypeUpdateFile = "file_updated"
+	EventTypeRemoveFile = "file_removed"
+	EventTypePostKey    = "key_posted"
+
+	AttributeKeyOwner       = "file_owner" // update file
+	AttributeKeyFileAddress = "file_address"
+
+	EventTypeAddViewers    = "viewers_added"
+	EventTypeAddEditors    = "editors_added"
+	EventTypeRemoveEditors = "editors_removed"
+	EventTypeRemoveViewers = "viewers_removed"
+	EventTypeResetViewers  = "viewers_reset"
+	EventTypeResetEditors  = "editors_reset"
+	EventTypeChangeOwner   = "owner_changed"
+	EventTypeMakeRoot      = "root_made"
+	EventTypePostFile      = "file_posted"
+
+	AttributeValueCategory = ModuleName
+
+	AttributeKeySigner   = "signer"
+	AttributeKeyNewOwner = "new_owner"
+	AttributeKeyKey      = "key"
+)
diff --git a/x/storage/keeper/msg_server_attest.go b/x/storage/keeper/msg_server_attest.go
@@ -59,6 +59,13 @@ func (k Keeper) Attest(ctx sdk.Context, prover string, merkle []byte, owner stri
 
 	k.RemoveAttestation(ctx, form.Prover, form.Merkle, form.Owner, form.Start)
 
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+		),
+	)
+
 	return nil
 }
 
@@ -128,6 +135,13 @@ func (k Keeper) RequestAttestation(ctx sdk.Context, merkle []byte, owner string,
 
 	k.SetAttestationForm(ctx, form)
 
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+		),
+	)
+
 	return providerAddresses, nil
 }
 
@@ -150,6 +164,13 @@ func (k msgServer) RequestAttestationForm(goCtx context.Context, msg *types.MsgR
 		errorString = err.Error()
 	}
 
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+		),
+	)
+
 	return &types.MsgRequestAttestationFormResponse{
 		Providers: providerAddresses,
 		Success:   success,

diff --git a/x/storage/keeper/msg_server_buy_storage.go b/x/storage/keeper/msg_server_buy_storage.go
@@ -178,6 +178,23 @@ func (k msgServer) BuyStorage(goCtx context.Context, msg *types.MsgBuyStorage) (
 		}
 	}
 
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+		),
+	)
+
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			types.EventTypeBuyStorage,
+			sdk.NewAttribute(types.AttributeKeyBuyer, msg.Creator),
+			sdk.NewAttribute(types.AttributeKeyReceiver, msg.ForAddress),
+			sdk.NewAttribute(types.AttributeKeyBytesBought, fmt.Sprintf("%d", msg.Bytes)),
+			sdk.NewAttribute(types.AttributeKeyTimeBought, hours.String()),
+		),
+	)
+
 	return &types.MsgBuyStorageResponse{}, nil
 }
 

diff --git a/x/storage/keeper/msg_server_file_delete.go b/x/storage/keeper/msg_server_file_delete.go
@@ -2,6 +2,7 @@ package keeper
 
 import (
 	"context"
+	"encoding/hex"
 
 	sdk "github.com/cosmos/cosmos-sdk/types"
 	"github.com/jackalLabs/canine-chain/v3/x/storage/types"
@@ -12,5 +13,20 @@ func (k msgServer) DeleteFile(goCtx context.Context, msg *types.MsgDeleteFile) (
 
 	k.Keeper.RemoveFile(ctx, msg.Merkle, msg.Creator, msg.Start)
 
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+		),
+	)
+
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			types.EventTypeCancelContract,
+			sdk.NewAttribute(types.AttributeKeySigner, msg.Creator),
+			sdk.NewAttribute(types.AttributeKeyContract, hex.EncodeToString(msg.Merkle)),
+		),
+	)
+
 	return &types.MsgDeleteFileResponse{}, nil
 }
diff --git a/x/storage/keeper/msg_server_init_provider.go b/x/storage/keeper/msg_server_init_provider.go
@@ -81,5 +81,12 @@ func (k msgServer) ShutdownProvider(goCtx context.Context, msg *types.MsgShutdow
 
 	k.RemoveProviders(ctx, msg.Creator)
 
+	ctx.EventManager().EmitEvent(
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+		),
+	)
+
 	return &types.MsgShutdownProviderResponse{}, nil
 }