feat(asset): add headers to StoreArgs (#928)

* add headers to StoreArgs * fix
dfinity · Sep 13, 2024 · 6310461 · 6310461
1 parent cad8bde
commit 6310461
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/packages/assets/src/canisters/assets_idl.js b/packages/assets/src/canisters/assets_idl.js
@@ -2,9 +2,11 @@ export const idlFactory = ({ IDL }) => {
   const ClearArguments = IDL.Record({});
   const BatchId = IDL.Nat;
   const Key = IDL.Text;
+  const HeaderField = IDL.Tuple(IDL.Text, IDL.Text);  
   const CreateAssetArguments = IDL.Record({
     key: Key,
     content_type: IDL.Text,
+    headers: IDL.Opt(IDL.Vec(HeaderField)),
   });
   const UnsetAssetContentArguments = IDL.Record({
     key: Key,
@@ -25,7 +27,6 @@ export const idlFactory = ({ IDL }) => {
     SetAssetContent: SetAssetContentArguments,
     Clear: ClearArguments,
   });
-  const HeaderField = IDL.Tuple(IDL.Text, IDL.Text);
   const HttpRequest = IDL.Record({
     url: IDL.Text,
     method: IDL.Text,

diff --git a/packages/assets/src/canisters/assets_service.ts b/packages/assets/src/canisters/assets_service.ts
@@ -13,6 +13,7 @@ export type ClearArguments = Record<string, never>;
 export interface CreateAssetArguments {
   key: Key;
   content_type: string;
+  headers: [] | [Array<HeaderField>];
 }
 
 export interface DeleteAssetArguments {

diff --git a/packages/assets/src/index.ts b/packages/assets/src/index.ts
@@ -60,6 +60,11 @@ export interface StoreConfig {
    * @default File/Blob object type or type from file name extension
    */
   contentType?: string;
+  /**
+   * Custom headers to be sent with the asset
+   * @default []
+   */
+  headers?: Array<[string, string]>;
   /**
    * Content encoding
    * @default 'identity'
@@ -330,9 +335,10 @@ class AssetManagerBatch {
         }),
       );
       await readable.close();
+      const headers: [] | [[string,string][]] = config?.headers ? [config.headers] : [];
       return [
         {
-          CreateAsset: { key, content_type: config?.contentType ?? readable.contentType },
+          CreateAsset: { key, content_type: config?.contentType ?? readable.contentType, headers },
         },
         {
           SetAssetContent: {