feat: update steno to support multiple content types (#32)

* feat: update steno to support multiple file types

* chore: revert other changes

* chore: remove lint changes to reduce PR complexity

* test: re-add tests for new supported types

src/index.ts

@@ -12,6 +12,7 @@ function getTempFilename(file: PathLike): string {
 
 type Resolve = () => void
 type Reject = (error: Error) => void
+export type writeableData = Parameters<typeof writeFile>[1]
 
 export class Writer {
   #filename: PathLike
@@ -20,10 +21,10 @@ export class Writer {
   #prev: [Resolve, Reject] | null = null
   #next: [Resolve, Reject] | null = null
   #nextPromise: Promise<void> | null = null
-  #nextData: string | null = null
+  #nextData: writeableData | null = null
 
   // File is locked, add data for later
-  #add(data: string): Promise<void> {
+  #add(data: writeableData): Promise<void> {
     // Only keep most recent data
     this.#nextData = data
 
@@ -39,7 +40,7 @@ export class Writer {
   }
 
   // File isn't locked, write data
-  async #write(data: string): Promise<void> {
+  async #write(data: writeableData): Promise<void> {
     // Lock file
     this.#locked = true
     try {
@@ -75,7 +76,7 @@ export class Writer {
     this.#tempFilename = getTempFilename(filename)
   }
 
-  async write(data: string): Promise<void> {
+  async write(data: writeableData): Promise<void> {
     return this.#locked ? this.#add(data) : this.#write(data)
   }
 }

src/test.ts

@@ -1,11 +1,12 @@
 import { strictEqual as equal } from 'node:assert'
-import fs from 'node:fs'
-import os from 'node:os'
-import path from 'node:path'
-import url from 'node:url'
+import * as fs from 'node:fs'
+import * as os from 'node:os'
+import * as path from 'node:path'
+import * as url from 'node:url'
+
 import { Writer } from './index.js'
 
-export async function testSteno(): Promise<void> {
+export async function testStenoStrings(): Promise<void> {
   const max = 1000
 
   const dir = fs.mkdtempSync(path.join(os.tmpdir(), 'steno-test-'))
@@ -27,3 +28,47 @@ export async function testSteno(): Promise<void> {
     equal(parseInt(fs.readFileSync(file, 'utf-8')), max)
   }
 }
+
+export async function testStenoUintArrays(): Promise<void> {
+  const max = 1000
+  const dir = fs.mkdtempSync(path.join(os.tmpdir(), 'steno-test-2'))
+
+  const file = path.join(dir, 'file.txt')
+  const fileURL = url.pathToFileURL(path.join(dir, 'fileURL.txt'))
+  for (const f of [file, fileURL]) {
+    const textEncoder = new TextEncoder()
+    const textDecoder = new TextDecoder()
+    const writer = new Writer(f)
+    const promises = []
+
+    // Test race condition
+    for (let i = 1; i <= max; ++i) {
+      promises.push(writer.write(textEncoder.encode(String(i))))
+    }
+
+    // All promises should resolve
+    await Promise.all(promises)
+    equal(parseInt(textDecoder.decode(fs.readFileSync(file))), max)
+  }
+}
+
+export async function testStenoBuffers(): Promise<void> {
+  const max = 1000
+  const dir = fs.mkdtempSync(path.join(os.tmpdir(), 'steno-test-2'))
+
+  const file = path.join(dir, 'file.txt')
+  const fileURL = url.pathToFileURL(path.join(dir, 'fileURL.txt'))
+  for (const f of [file, fileURL]) {
+    const writer = new Writer(f)
+    const promises = []
+
+    // Test race condition
+    for (let i = 1; i <= max; ++i) {
+      promises.push(writer.write(Buffer.from(String(i), 'utf-8')))
+    }
+
+    // All promises should resolve
+    await Promise.all(promises)
+    equal(parseInt(fs.readFileSync(file, 'utf-8')), max)
+  }
+}