Review: add todo to avoid custom query in enclave layer

go/enclave/enclave.go

@@ -855,6 +855,8 @@ func (e *enclaveImpl) GetTotalContractCount(ctx context.Context) (*big.Int, comm
 	return e.storage.GetContractCount(ctx)
 }
 
+// GetCustomQuery is a generic query method for queries that don't match the eth API.
+// todo: get rid of this method and use specific methods for each query (e.g. GetPersonalTransactions)
 func (e *enclaveImpl) GetCustomQuery(ctx context.Context, encryptedParams common.EncryptedParamsGetStorageAt) (*responses.PrivateQueryResponse, common.SystemError) {
 	// ensure the enclave is running
 	if e.stopControl.IsStopping() {

tools/walletextension/rpcapi/blockchain_api.go

@@ -165,6 +165,7 @@ func (api *BlockChainAPI) GetStorageAt(ctx context.Context, address gethcommon.A
 		if err != nil {
 			return nil, fmt.Errorf("unable to extract address from custom query params: %w", err)
 		}
+		// todo: we should be calling something like `ten_getPrivateTransactions` here, this custom query stuff only needs to be in the gateway layer
 		resp, err := ExecAuthRPC[any](ctx, api.we, &ExecCfg{account: userAddr}, "eth_getStorageAt", address.Hex(), params, nil)
 		if err != nil {
 			return nil, fmt.Errorf("unable to execute custom query: %w", err)