add update to the tests

.github/workflows/test.yml

@@ -77,7 +77,6 @@
 # "examples/stable_structures",
 # "examples/timers",
 # "examples/tuple_types",
-# "examples/update"
 
 name: Azle Tests
 on:
@@ -123,7 +122,8 @@ jobs:
               run: |
                   EXAMPLE_DIRECTORIES=$(cat << END
                   [
-                    "examples/query"
+                    "examples/query",
+                    "examples/update"
                   ]
                   END
                   )

examples/update/src/index.ts

@@ -1,15 +1,12 @@
-import { query, text, update, Void } from 'azle';
+import { query, Service, text, update, Void } from 'azle';
 
 let currentMessage: string = '';
 
-export default class {
-    @query([], text)
-    getCurrentMessage(): text {
+export default Service({
+    getCurrentMessage: query([], text, () => {
         return currentMessage;
-    }
-
-    @update([text], Void)
-    simpleUpdate(message: text): Void {
+    }),
+    simpleUpdate: update([text], Void, (message) => {
         currentMessage = message;
-    }
-}
+    })
+});

src/lib_functional/candid/reference/service.ts

@@ -1,10 +1,5 @@
 import { CanisterMethods } from '../../../compiler/utils/types';
-import { CanisterMethodInfo, query } from '../../canister_methods/query';
-
-// type CanisterMethodType = 'query';
-// type CanisterMethods = (typeof query & {
-//     canisterMethodType: CanisterMethodType;
-// })[];
+import { CanisterMethodInfo } from '../../canister_methods';
 
 type ServiceOptions = {
     [key: string]: CanisterMethodInfo;
@@ -37,6 +32,22 @@ export function Service(serviceOptions: ServiceOptions): CanisterMethods {
             };
         });
 
+    const updates = Object.entries(serviceOptions)
+        .filter((entry) => {
+            const key = entry[0];
+            const value = entry[1];
+
+            return value.type === 'update';
+        })
+        .map((entry) => {
+            const key = entry[0];
+            const value = entry[1];
+
+            return {
+                name: key
+            };
+        });
+
     // TODO loop through each key and simply grab the candid off
     // TODO grab the init/post_upgrade candid as well
     return {
@@ -45,11 +56,11 @@ export function Service(serviceOptions: ServiceOptions): CanisterMethods {
         .map((entry) => {
             return `${entry[0]}: ${entry[1].candid}`;
         })
-        .join('')}
+        .join('\n    ')}
 }
 `,
         queries,
-        updates: [],
+        updates,
         callbacks
     };
 }

src/lib_functional/canister_methods/index.ts

@@ -2,6 +2,13 @@ import { IDL } from '../../lib_new/index';
 import { ic } from '../../lib_new/ic';
 
 export * from './query';
+export * from './update';
+
+export type CanisterMethodInfo = {
+    type: 'query' | 'update';
+    callback: (...args: any) => any;
+    candid: string;
+};
 
 export function executeMethod(
     paramCandid: any,
@@ -11,7 +18,7 @@ export function executeMethod(
 ) {
     const decoded = IDL.decode(paramCandid[0] as any, args[0]);
 
-    const result = callback(decoded);
+    const result = callback(...decoded);
 
     if (
         result !== undefined &&

src/lib_functional/canister_methods/query.ts

@@ -2,15 +2,7 @@ import {
     handleRecursiveParams,
     handleRecursiveReturn
 } from '../../lib_new/method_decorators';
-import { IDL } from '../../lib_new/index';
-import { ic } from '../../lib_new/ic';
-import { executeMethod } from '.';
-
-export type CanisterMethodInfo = {
-    type: 'query';
-    callback: (...args: any) => any;
-    candid: string;
-};
+import { CanisterMethodInfo, executeMethod } from '.';
 
 // TODO execute the candid stuff and just store it on the function itself?
 export function query(

src/lib_functional/canister_methods/update.ts

@@ -1 +1,22 @@
-export function update() {}
+import {
+    handleRecursiveParams,
+    handleRecursiveReturn
+} from '../../lib_new/method_decorators';
+import { CanisterMethodInfo, executeMethod } from '.';
+
+export function update(
+    paramsIdls: any,
+    returnIdl: any,
+    callback: any
+): CanisterMethodInfo {
+    const paramCandid = handleRecursiveParams(paramsIdls);
+    const returnCandid = handleRecursiveReturn(returnIdl, paramCandid[2]);
+
+    return {
+        type: 'update',
+        callback: (...args) => {
+            executeMethod(paramCandid, returnCandid, args, callback);
+        },
+        candid: `(${paramCandid[1].join(', ')}) -> (${returnCandid[1]});`
+    };
+}