Add hooks to TEE and missing items to index

Author: nolag

packages/cre-sdk/src/sdk/cre/index.ts

@@ -2,12 +2,18 @@
  * Public API for the CRE SDK.
  */
 
-import { ClientCapability as AptosClient } from '@cre/generated-sdk/capabilities/blockchain/aptos/v1alpha/client_sdk_gen'
+import {
+	ClientCapability as AptosClient,
+	ClientRestrictor as AptosRestrictor,
+} from '@cre/generated-sdk/capabilities/blockchain/aptos/v1alpha/client_sdk_gen'
 import {
 	ClientCapability as EVMClient,
 	ClientRestrictor as EVMRestrictor,
 } from '@cre/generated-sdk/capabilities/blockchain/evm/v1alpha/client_sdk_gen'
-import { ClientCapability as SolanaClient } from '@cre/generated-sdk/capabilities/blockchain/solana/v1alpha/client_sdk_gen'
+import {
+	ClientCapability as SolanaClient,
+	ClientRestrictor as SolanaRestrictor,
+} from '@cre/generated-sdk/capabilities/blockchain/solana/v1alpha/client_sdk_gen'
 import {
 	ClientCapability as ConfidentialHTTPClient,
 	ClientRestrictor as ConfidentialHTTPRestrictor,
@@ -45,6 +51,7 @@ export { TeeType } from '@cre/generated/sdk/v1alpha/sdk_pb'
 // Aptos Capability
 export {
 	ClientCapability as AptosClient,
+	ClientRestrictor as AptosRestrictor,
 	type WriteCreReportRequest as AptosWriteCreReportRequest,
 	type WriteCreReportRequestJson as AptosWriteCreReportRequestJson,
 } from '@cre/generated-sdk/capabilities/blockchain/aptos/v1alpha/client_sdk_gen'
@@ -59,6 +66,7 @@ export {
 // Solana Capability
 export {
 	ClientCapability as SolanaClient,
+	ClientRestrictor as SolanaRestrictor,
 	type WriteCreReportRequest as SolanaWriteCreReportRequest,
 	type WriteCreReportRequestJson as SolanaWriteCreReportRequestJson,
 } from '@cre/generated-sdk/capabilities/blockchain/solana/v1alpha/client_sdk_gen'
@@ -116,6 +124,8 @@ export const cre = {
 		ConfidentialHTTPRestrictor,
 		HTTPClientRestrictor,
 		EVMRestrictor,
+		SolanaRestrictor,
+		AptosRestrictor,
 	},
 	handler,
 	handlerInTee,

packages/cre-sdk/src/sdk/workflow.ts

@@ -78,10 +78,12 @@ export const handlerInTee = <
 	trigger: Trigger<TRawTriggerOutput, TTriggerOutput>,
 	fn: HandlerFn<TConfig, TTriggerOutput, TResult, TeeRuntime<TConfig>>,
 	tees: TeeConstraint,
+	hooks?: Hooks<TConfig, TTriggerOutput>,
 ): HandlerEntry<TConfig, TRawTriggerOutput, TTriggerOutput, TResult, TeeRuntime<TConfig>> => ({
 	trigger,
 	fn,
 	requirements: buildTeeRequirements(tees),
+	hooks,
 })
 
 export type SecretsProvider = {