mirror of
https://github.com/Boof2015/prism.git
synced 2026-08-16 16:21:11 +02:00
astra integration module
This commit is contained in:
Boof2015
@@ -0,0 +1,380 @@
|
||||
import assert from 'node:assert/strict'
|
||||
import { mkdir, mkdtemp, rm, writeFile } from 'node:fs/promises'
|
||||
import { dirname } from 'node:path'
|
||||
import { tmpdir } from 'node:os'
|
||||
import { join } from 'node:path'
|
||||
import test from 'node:test'
|
||||
import { AstraIntegrationService, normalizeAstraIntegrationConfig } from '../src/main/services/astraIntegration'
|
||||
import { DEFAULT_ASTRA_BASE_URL, type AstraIntegrationConfig } from '../src/types/astra'
|
||||
|
||||
function createJsonResponse(body: unknown, status = 200): Response {
|
||||
return new Response(JSON.stringify(body), {
|
||||
status,
|
||||
headers: {
|
||||
'content-type': 'application/json; charset=utf-8',
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
function createPngResponse(payload = 'artwork'): Response {
|
||||
return new Response(Buffer.from(payload), {
|
||||
status: 200,
|
||||
headers: {
|
||||
'content-type': 'image/png',
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
function createHeaders(init?: RequestInit): Headers {
|
||||
return new Headers(init?.headers)
|
||||
}
|
||||
|
||||
function createFakeTimers(): {
|
||||
clearTimeoutImpl: (handle: ReturnType<typeof setTimeout>) => void
|
||||
nextDelay: () => number | null
|
||||
pendingCount: () => number
|
||||
runNext: () => void
|
||||
setTimeoutImpl: typeof setTimeout
|
||||
} {
|
||||
let nextHandle = 1
|
||||
const timers = new Map<number, { callback: () => void; delay: number }>()
|
||||
|
||||
return {
|
||||
setTimeoutImpl(callback: TimerHandler, delay?: number): ReturnType<typeof setTimeout> {
|
||||
const handle = nextHandle
|
||||
nextHandle += 1
|
||||
timers.set(handle, {
|
||||
callback: typeof callback === 'function' ? callback as () => void : () => {},
|
||||
delay: typeof delay === 'number' ? delay : 0,
|
||||
})
|
||||
return handle as ReturnType<typeof setTimeout>
|
||||
},
|
||||
clearTimeoutImpl(handle: ReturnType<typeof setTimeout>): void {
|
||||
timers.delete(Number(handle))
|
||||
},
|
||||
nextDelay: () => {
|
||||
const next = timers.values().next().value as { delay: number } | undefined
|
||||
return next?.delay ?? null
|
||||
},
|
||||
pendingCount: () => timers.size,
|
||||
runNext: () => {
|
||||
const next = timers.entries().next().value as [number, { callback: () => void }] | undefined
|
||||
if (!next) return
|
||||
const [handle, timer] = next
|
||||
timers.delete(handle)
|
||||
timer.callback()
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
function createSseStream(): {
|
||||
close: () => void
|
||||
pushEvent: (event: string, payload: unknown) => void
|
||||
response: Response
|
||||
} {
|
||||
const encoder = new TextEncoder()
|
||||
let controllerRef: ReadableStreamDefaultController<Uint8Array> | null = null
|
||||
|
||||
const response = new Response(new ReadableStream<Uint8Array>({
|
||||
start(controller) {
|
||||
controllerRef = controller
|
||||
},
|
||||
}), {
|
||||
status: 200,
|
||||
headers: {
|
||||
'content-type': 'text/event-stream; charset=utf-8',
|
||||
},
|
||||
})
|
||||
|
||||
return {
|
||||
response,
|
||||
pushEvent(event, payload) {
|
||||
controllerRef?.enqueue(encoder.encode(`event: ${event}\ndata: ${JSON.stringify(payload)}\n\n`))
|
||||
},
|
||||
close() {
|
||||
controllerRef?.close()
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
async function waitFor(predicate: () => boolean, message: string): Promise<void> {
|
||||
const deadline = Date.now() + 2000
|
||||
while (Date.now() < deadline) {
|
||||
if (predicate()) {
|
||||
return
|
||||
}
|
||||
await new Promise((resolve) => setTimeout(resolve, 0))
|
||||
}
|
||||
|
||||
assert.fail(message)
|
||||
}
|
||||
|
||||
async function createConfigFile(config: AstraIntegrationConfig): Promise<{
|
||||
cleanup: () => Promise<void>
|
||||
configPath: string
|
||||
}> {
|
||||
const rootDir = await mkdtemp(join(tmpdir(), 'prism-astra-tests-'))
|
||||
const configPath = join(rootDir, 'userData', 'astra-integration.json')
|
||||
await mkdir(dirname(configPath), { recursive: true })
|
||||
await writeFile(configPath, JSON.stringify(config, null, 2), 'utf8')
|
||||
return {
|
||||
configPath,
|
||||
cleanup: () => rm(rootDir, { recursive: true, force: true }),
|
||||
}
|
||||
}
|
||||
|
||||
test('normalizeAstraIntegrationConfig applies defaults and trims input', () => {
|
||||
const config = normalizeAstraIntegrationConfig({
|
||||
baseUrl: ' http://127.0.0.1:38401/ ',
|
||||
token: ' secret ',
|
||||
})
|
||||
|
||||
assert.deepEqual(config, {
|
||||
baseUrl: DEFAULT_ASTRA_BASE_URL,
|
||||
token: 'secret',
|
||||
})
|
||||
})
|
||||
|
||||
test('service initializes from config, hydrates artwork, and applies SSE updates', async () => {
|
||||
const harness = await createConfigFile({
|
||||
baseUrl: DEFAULT_ASTRA_BASE_URL,
|
||||
token: 'secret-token',
|
||||
})
|
||||
const sse = createSseStream()
|
||||
const artworkUrl = `${DEFAULT_ASTRA_BASE_URL}/v1/artwork/current?trackId=track-1`
|
||||
const calls: Array<{ init?: RequestInit; url: string }> = []
|
||||
|
||||
const fetchImpl: typeof fetch = async (input, init) => {
|
||||
const url = String(input)
|
||||
calls.push({ url, init })
|
||||
const pathname = new URL(url).pathname
|
||||
const headers = createHeaders(init)
|
||||
assert.equal(headers.get('authorization'), 'Bearer secret-token')
|
||||
|
||||
if (pathname === '/v1/now-playing') {
|
||||
return createJsonResponse({
|
||||
playbackState: 'playing',
|
||||
currentTime: 15,
|
||||
duration: 180,
|
||||
queueLength: 2,
|
||||
outputDeviceLabel: 'Built-in Output',
|
||||
visualizerLineColor: '#4ade80',
|
||||
updatedAt: 1000,
|
||||
currentTrack: {
|
||||
id: 'track-1',
|
||||
title: 'Song One',
|
||||
artist: 'Artist One',
|
||||
album: 'Album One',
|
||||
isFavorite: false,
|
||||
artworkUrl,
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
if (pathname === '/v1/artwork/current') {
|
||||
return createPngResponse('cover-one')
|
||||
}
|
||||
|
||||
if (pathname === '/v1/events') {
|
||||
assert.equal(headers.get('accept'), 'text/event-stream')
|
||||
return sse.response
|
||||
}
|
||||
|
||||
throw new Error(`Unexpected request: ${url}`)
|
||||
}
|
||||
|
||||
const service = new AstraIntegrationService({
|
||||
configPath: harness.configPath,
|
||||
fetchImpl,
|
||||
now: () => 1000,
|
||||
})
|
||||
|
||||
try {
|
||||
await service.initialize()
|
||||
assert.equal(service.getState().connectionState, 'disabled')
|
||||
await service.setConsumerActive(1, true)
|
||||
await waitFor(() => service.getState().connectionState === 'connected', 'expected connected Astra state')
|
||||
|
||||
const initialState = service.getState()
|
||||
assert.equal(initialState.snapshot?.currentTrack?.title, 'Song One')
|
||||
assert.match(initialState.snapshot?.currentTrack?.artworkDataUrl ?? '', /^data:image\/png;base64,/)
|
||||
assert.equal(calls.some((call) => call.url.endsWith('/v1/artwork/current?trackId=track-1')), true)
|
||||
|
||||
sse.pushEvent('now-playing', {
|
||||
playbackState: 'paused',
|
||||
currentTime: 40,
|
||||
duration: 180,
|
||||
queueLength: 2,
|
||||
outputDeviceLabel: 'Built-in Output',
|
||||
visualizerLineColor: '#f97316',
|
||||
updatedAt: 2500,
|
||||
currentTrack: {
|
||||
id: 'track-2',
|
||||
title: 'Song Two',
|
||||
artist: 'Artist Two',
|
||||
album: 'Album Two',
|
||||
isFavorite: true,
|
||||
artworkUrl: null,
|
||||
},
|
||||
})
|
||||
|
||||
await waitFor(() => service.getState().snapshot?.currentTrack?.id === 'track-2', 'expected SSE track update')
|
||||
const nextState = service.getState()
|
||||
assert.equal(nextState.snapshot?.playbackState, 'paused')
|
||||
assert.equal(nextState.snapshot?.currentTrack?.artworkDataUrl, null)
|
||||
|
||||
await service.setConsumerActive(1, false)
|
||||
assert.equal(service.getState().connectionState, 'disabled')
|
||||
assert.equal(service.getState().snapshot, null)
|
||||
} finally {
|
||||
await service.dispose()
|
||||
await harness.cleanup()
|
||||
}
|
||||
})
|
||||
|
||||
test('service schedules reconnect when the SSE stream closes', async () => {
|
||||
const harness = await createConfigFile({
|
||||
baseUrl: DEFAULT_ASTRA_BASE_URL,
|
||||
token: 'secret-token',
|
||||
})
|
||||
const timers = createFakeTimers()
|
||||
const steadyStream = createSseStream()
|
||||
let eventStreamRequests = 0
|
||||
|
||||
const fetchImpl: typeof fetch = async (input) => {
|
||||
const url = String(input)
|
||||
const pathname = new URL(url).pathname
|
||||
|
||||
if (pathname === '/v1/now-playing') {
|
||||
return createJsonResponse({
|
||||
playbackState: 'stopped',
|
||||
currentTime: 0,
|
||||
duration: 0,
|
||||
queueLength: 0,
|
||||
outputDeviceLabel: null,
|
||||
visualizerLineColor: '#38bdf8',
|
||||
updatedAt: 0,
|
||||
currentTrack: null,
|
||||
})
|
||||
}
|
||||
|
||||
if (pathname === '/v1/events') {
|
||||
eventStreamRequests += 1
|
||||
if (eventStreamRequests === 1) {
|
||||
return new Response(new ReadableStream<Uint8Array>({
|
||||
start(controller) {
|
||||
controller.close()
|
||||
},
|
||||
}), {
|
||||
status: 200,
|
||||
headers: {
|
||||
'content-type': 'text/event-stream; charset=utf-8',
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
return steadyStream.response
|
||||
}
|
||||
|
||||
throw new Error(`Unexpected request: ${url}`)
|
||||
}
|
||||
|
||||
const service = new AstraIntegrationService({
|
||||
configPath: harness.configPath,
|
||||
fetchImpl,
|
||||
setTimeoutImpl: timers.setTimeoutImpl,
|
||||
clearTimeoutImpl: timers.clearTimeoutImpl,
|
||||
})
|
||||
|
||||
try {
|
||||
await service.initialize()
|
||||
await service.setConsumerActive(1, true)
|
||||
await waitFor(() => service.getState().connectionState === 'error', 'expected reconnect error state')
|
||||
assert.equal(timers.nextDelay(), 1000)
|
||||
assert.equal(timers.pendingCount(), 1)
|
||||
|
||||
timers.runNext()
|
||||
|
||||
await waitFor(() => eventStreamRequests === 2, 'expected second SSE connection attempt')
|
||||
await waitFor(() => service.getState().connectionState === 'connected', 'expected connected state after reconnect')
|
||||
} finally {
|
||||
await service.dispose()
|
||||
await harness.cleanup()
|
||||
}
|
||||
})
|
||||
|
||||
test('service surfaces 401 and 403 control errors and clears them after success', async () => {
|
||||
const harness = await createConfigFile({
|
||||
baseUrl: DEFAULT_ASTRA_BASE_URL,
|
||||
token: 'secret-token',
|
||||
})
|
||||
const sse = createSseStream()
|
||||
let controlRequests = 0
|
||||
|
||||
const fetchImpl: typeof fetch = async (input, init) => {
|
||||
const url = String(input)
|
||||
const pathname = new URL(url).pathname
|
||||
|
||||
if (pathname === '/v1/now-playing') {
|
||||
return createJsonResponse({
|
||||
playbackState: 'paused',
|
||||
currentTime: 20,
|
||||
duration: 180,
|
||||
queueLength: 1,
|
||||
outputDeviceLabel: 'Built-in Output',
|
||||
visualizerLineColor: '#38bdf8',
|
||||
updatedAt: 0,
|
||||
currentTrack: {
|
||||
id: 'track-1',
|
||||
title: 'Track',
|
||||
artist: 'Artist',
|
||||
album: 'Album',
|
||||
isFavorite: false,
|
||||
artworkUrl: null,
|
||||
},
|
||||
})
|
||||
}
|
||||
|
||||
if (pathname === '/v1/events') {
|
||||
return sse.response
|
||||
}
|
||||
|
||||
if (pathname === '/v1/control') {
|
||||
const headers = createHeaders(init)
|
||||
assert.equal(headers.get('authorization'), 'Bearer secret-token')
|
||||
controlRequests += 1
|
||||
if (controlRequests === 1) {
|
||||
return createJsonResponse({ error: 'Unauthorized' }, 401)
|
||||
}
|
||||
if (controlRequests === 2) {
|
||||
return createJsonResponse({ error: 'External playback controls are disabled.' }, 403)
|
||||
}
|
||||
return createJsonResponse({ ok: true }, 200)
|
||||
}
|
||||
|
||||
throw new Error(`Unexpected request: ${url}`)
|
||||
}
|
||||
|
||||
const service = new AstraIntegrationService({
|
||||
configPath: harness.configPath,
|
||||
fetchImpl,
|
||||
})
|
||||
|
||||
try {
|
||||
await service.initialize()
|
||||
await service.setConsumerActive(1, true)
|
||||
await waitFor(() => service.getState().connectionState === 'connected', 'expected connected state before controls')
|
||||
|
||||
await assert.rejects(() => service.sendControl('next'), /Unauthorized/)
|
||||
assert.equal(service.getState().lastControlError, 'Unauthorized')
|
||||
|
||||
await assert.rejects(() => service.sendControl('previous'), /External playback controls are disabled\./)
|
||||
assert.equal(service.getState().lastControlError, 'External playback controls are disabled.')
|
||||
|
||||
await service.sendControl('play')
|
||||
assert.equal(service.getState().lastControlError, null)
|
||||
} finally {
|
||||
await service.dispose()
|
||||
await harness.cleanup()
|
||||
}
|
||||
})
|
||||
Reference in New Issue
Block a user