diff --git a/packages/assets/src/canisters/assets_idl.js b/packages/assets/src/canisters/assets_idl.js index 4bf539156..e27f3a7d9 100644 --- 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 index c3cbc9bbd..ee5aee405 100644 --- a/packages/assets/src/canisters/assets_service.ts +++ b/packages/assets/src/canisters/assets_service.ts @@ -13,6 +13,7 @@ export type ClearArguments = Record; export interface CreateAssetArguments { key: Key; content_type: string; + headers: [] | [Array]; } export interface DeleteAssetArguments { diff --git a/packages/assets/src/index.ts b/packages/assets/src/index.ts index 73e6fd273..a2a2579de 100644 --- 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: {