alexcaza · alexcaza · Aug 31, 2024 · Jul 26, 2024 · Jul 26, 2024 · Jul 26, 2024
diff --git a/lib/__specs__/main.spec.ts b/lib/__specs__/main.spec.ts
@@ -1,7 +1,7 @@
 import { describe, it, expect } from "bun:test";
 import { mkConfig } from "../config.ts";
 import { asBlob, generateCsv } from "../generator.ts";
-import { ConfigOptions } from "../types.ts";
+import { ConfigOptions, MediaType } from "../types.ts";
 import { asString } from "../helpers.ts";
 
 const mockData = [
@@ -344,6 +344,27 @@ describe("ExportToCsv", () => {
     expect(firstLine).toBe("Test Csv 2\r");
   });
 
+  it("should allow for custom file extensions", () => {
+    const csvOpts: ConfigOptions = {
+      mediaType: MediaType.csv,
+    };
+    const csvConf = mkConfig(csvOpts);
+
+    const txtOpts: ConfigOptions = {
+      mediaType: MediaType.plain,
+    };
+    const txtConf = mkConfig(txtOpts);
+
+    const tsvOpts: ConfigOptions = {
+      mediaType: MediaType.tsv,
+    };
+    const tsvConf = mkConfig(tsvOpts);
+
+    expect(csvConf.mediaType).toBe(MediaType.csv);
+    expect(txtConf.mediaType).toBe(MediaType.plain);
+    expect(tsvConf.mediaType).toBe(MediaType.tsv);
+  });
+
   describe("asBlob", () => {
     it("should construct a valid blob based on options", async () => {
       const options: ConfigOptions = {

diff --git a/lib/config.ts b/lib/config.ts
@@ -1,4 +1,4 @@
-import { WithDefaults, ConfigOptions } from "./types.ts";
+import { WithDefaults, ConfigOptions, MediaType } from "./types.ts";
 
 export const defaults: WithDefaults<ConfigOptions> = {
   fieldSeparator: ",",
@@ -11,6 +11,7 @@ export const defaults: WithDefaults<ConfigOptions> = {
   showColumnHeaders: true,
   useTextFile: false,
   fileExtension: "csv",
+  mediaType: MediaType.csv,
   useBom: true,
   columnHeaders: [],
   useKeysAsHeaders: false,

diff --git a/lib/generator.ts b/lib/generator.ts
@@ -67,10 +67,12 @@ export const asBlob =
     const withDefaults = mkConfig(config);
     const data = unpack(csvOutput);
 
-    // Create blob from CsvOutput either as text or csv file.
-    const fileType = withDefaults.useTextFile ? "plain" : "csv";
+    // Create blob from CsvOutput using the supplied mime type.
+    const mimeType = withDefaults.useTextFile
+      ? "text/plain"
+      : withDefaults.mediaType;
     const blob = new Blob([data], {
-      type: `text/${fileType};charset=utf8;`,
+      type: `${mimeType};charset=utf8;`,
     });
 
     return blob;

diff --git a/lib/index.ts b/lib/index.ts
@@ -1,4 +1,4 @@
 export * from "./generator.ts";
 export { mkConfig } from "./config.ts";
-export { CsvOutput, ConfigOptions, ColumnHeader } from "./types.ts";
+export { CsvOutput, ConfigOptions, ColumnHeader, MediaType } from "./types.ts";
 export { asString } from "./helpers.ts";
diff --git a/lib/types.ts b/lib/types.ts
@@ -7,6 +7,12 @@ export type WithDefaults<T> = Required<T>;
 
 export type ColumnHeader = string | { key: string; displayLabel: string };
 
+export enum MediaType {
+  csv = "text/csv",
+  tsv = "text/tab-separated-values",
+  plain = "text/plain",
+}
+
 export type ConfigOptions = {
   filename?: string;
   fieldSeparator?: string;
@@ -16,8 +22,16 @@ export type ConfigOptions = {
   showColumnHeaders?: boolean;
   showTitle?: boolean;
   title?: string;
+  /**
+   * Use `fileExtension` instead.
+   *
+   * Will be removed in the next major version (`2.x.x`)
+   *
+   * @deprecated
+   */
   useTextFile?: boolean;
   fileExtension?: string;
+  mediaType?: MediaType;
   useBom?: boolean;
   columnHeaders?: Array<ColumnHeader>;
   useKeysAsHeaders?: boolean;