mirror of
https://github.com/NousResearch/hermes-agent.git
synced 2026-07-31 19:16:29 +00:00
253 lines
8.5 KiB
TypeScript
253 lines
8.5 KiB
TypeScript
// @vitest-environment jsdom
|
|
// downloadGatewayMediaFile drives an <a download> click, so these need a DOM.
|
|
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
|
|
|
|
import { $connection } from '@/store/session'
|
|
|
|
import {
|
|
downloadGatewayMediaFile,
|
|
filePathFromMediaPath,
|
|
gatewayMediaDataUrl,
|
|
isInlineMediaSrc,
|
|
isRemoteGateway,
|
|
mediaExternalUrl,
|
|
resolveMediaDisplaySrc,
|
|
resolveMediaPlaybackSrc
|
|
} from './media'
|
|
|
|
describe('isRemoteGateway', () => {
|
|
afterEach(() => {
|
|
$connection.set(null)
|
|
})
|
|
|
|
it('is false with no connection', () => {
|
|
$connection.set(null)
|
|
expect(isRemoteGateway()).toBe(false)
|
|
})
|
|
|
|
it('is false in local mode', () => {
|
|
$connection.set({ mode: 'local' } as never)
|
|
expect(isRemoteGateway()).toBe(false)
|
|
})
|
|
|
|
it('is true in remote mode', () => {
|
|
$connection.set({ mode: 'remote' } as never)
|
|
expect(isRemoteGateway()).toBe(true)
|
|
})
|
|
})
|
|
|
|
describe('filePathFromMediaPath', () => {
|
|
it('passes through a plain path', () => {
|
|
expect(filePathFromMediaPath('/home/u/.hermes/images/a.png')).toBe('/home/u/.hermes/images/a.png')
|
|
})
|
|
|
|
it('decodes a file:// URL with encoded characters', () => {
|
|
expect(filePathFromMediaPath('file:///tmp/a%20b.png')).toBe('/tmp/a b.png')
|
|
})
|
|
})
|
|
|
|
describe('mediaExternalUrl', () => {
|
|
afterEach(() => {
|
|
$connection.set(null)
|
|
})
|
|
|
|
it('passes through http(s) URLs untouched', () => {
|
|
$connection.set({ mode: 'remote', baseUrl: 'https://gw', token: 't' } as never)
|
|
expect(mediaExternalUrl('https://example.com/a.png')).toBe('https://example.com/a.png')
|
|
})
|
|
|
|
it('keeps file:// form in local mode', () => {
|
|
$connection.set({ mode: 'local' } as never)
|
|
expect(mediaExternalUrl('/tmp/a.png')).toBe('file:///tmp/a.png')
|
|
expect(mediaExternalUrl('file:///tmp/a.png')).toBe('file:///tmp/a.png')
|
|
})
|
|
|
|
it('rewrites gateway-local paths to an authenticated download URL', () => {
|
|
$connection.set({ mode: 'remote', baseUrl: 'https://gw', token: 's e/cret' } as never)
|
|
expect(mediaExternalUrl('file:///tmp/a b.png')).toBe(
|
|
'https://gw/api/files/download?path=%2Ftmp%2Fa%20b.png&token=s%20e%2Fcret'
|
|
)
|
|
expect(mediaExternalUrl('/tmp/a b.png')).toBe(
|
|
'https://gw/api/files/download?path=%2Ftmp%2Fa%20b.png&token=s%20e%2Fcret'
|
|
)
|
|
})
|
|
|
|
it('falls back to file:// when remote connection lacks a token', () => {
|
|
$connection.set({ mode: 'remote', baseUrl: 'https://gw' } as never)
|
|
expect(mediaExternalUrl('/tmp/a.png')).toBe('file:///tmp/a.png')
|
|
})
|
|
})
|
|
|
|
describe('resolveMediaDisplaySrc', () => {
|
|
const api = vi.fn(async ({ path }: { path: string }) => {
|
|
if (path.startsWith('/api/fs/read-data-url?')) {
|
|
return { dataUrl: 'data:image/png;base64,ZHVtbXk=' }
|
|
}
|
|
|
|
throw new Error(`unexpected path ${path}`)
|
|
})
|
|
|
|
beforeEach(() => {
|
|
api.mockClear()
|
|
})
|
|
|
|
afterEach(() => {
|
|
vi.unstubAllGlobals()
|
|
$connection.set(null)
|
|
})
|
|
|
|
it('recognizes inline image URLs', () => {
|
|
expect(isInlineMediaSrc('https://example.com/a.png')).toBe(true)
|
|
expect(isInlineMediaSrc('data:image/png;base64,ZHVtbXk=')).toBe(true)
|
|
expect(isInlineMediaSrc('/Users/me/a.png')).toBe(false)
|
|
})
|
|
|
|
it('leaves web, data, and relative markdown image sources unchanged', async () => {
|
|
vi.stubGlobal('window', { hermesDesktop: { api } })
|
|
$connection.set({ mode: 'remote', profile: 'remote-work' } as never)
|
|
|
|
await expect(resolveMediaDisplaySrc('https://example.com/a.png')).resolves.toBe('https://example.com/a.png')
|
|
await expect(resolveMediaDisplaySrc('data:image/png;base64,ZHVtbXk=')).resolves.toBe(
|
|
'data:image/png;base64,ZHVtbXk='
|
|
)
|
|
await expect(resolveMediaDisplaySrc('images/a.png')).resolves.toBe('images/a.png')
|
|
await expect(resolveMediaDisplaySrc('./images/a.png')).resolves.toBe('./images/a.png')
|
|
await expect(resolveMediaDisplaySrc('../images/a.png')).resolves.toBe('../images/a.png')
|
|
expect(api).not.toHaveBeenCalled()
|
|
})
|
|
|
|
it('reads remote gateway-local file paths through the desktop fs bridge', async () => {
|
|
vi.stubGlobal('window', { hermesDesktop: { api } })
|
|
$connection.set({ mode: 'remote', profile: 'remote-work' } as never)
|
|
|
|
await expect(resolveMediaDisplaySrc('/Users/me/project/a b.png')).resolves.toBe('data:image/png;base64,ZHVtbXk=')
|
|
expect(api).toHaveBeenCalledWith({
|
|
path: '/api/fs/read-data-url?path=%2FUsers%2Fme%2Fproject%2Fa%20b.png',
|
|
profile: 'remote-work'
|
|
})
|
|
})
|
|
|
|
it('reads local desktop file paths from the local desktop shell', async () => {
|
|
const readFileDataUrl = vi.fn(async () => 'data:image/png;base64,bG9jYWw=')
|
|
|
|
vi.stubGlobal('window', { hermesDesktop: { readFileDataUrl } })
|
|
$connection.set({ mode: 'local' } as never)
|
|
|
|
await expect(resolveMediaDisplaySrc('file:///Users/me/project/a%20b.png')).resolves.toBe(
|
|
'data:image/png;base64,bG9jYWw='
|
|
)
|
|
expect(readFileDataUrl).toHaveBeenCalledWith('/Users/me/project/a b.png')
|
|
})
|
|
})
|
|
|
|
describe('resolveMediaPlaybackSrc', () => {
|
|
afterEach(() => {
|
|
vi.unstubAllGlobals()
|
|
$connection.set(null)
|
|
})
|
|
|
|
it('keeps a remote HTTPS video URL unchanged', async () => {
|
|
vi.stubGlobal('window', { hermesDesktop: { api: vi.fn() } })
|
|
$connection.set({ mode: 'remote', baseUrl: 'https://gateway.test', token: 'secret' } as never)
|
|
|
|
await expect(resolveMediaPlaybackSrc('https://cdn.example.com/render.mp4')).resolves.toBe(
|
|
'https://cdn.example.com/render.mp4'
|
|
)
|
|
})
|
|
|
|
it('routes gateway-local video through the authenticated download endpoint', async () => {
|
|
vi.stubGlobal('window', { hermesDesktop: { api: vi.fn() } })
|
|
$connection.set({ mode: 'remote', baseUrl: 'https://gateway.test', token: 's e/cret' } as never)
|
|
|
|
await expect(resolveMediaPlaybackSrc('/root/outputs/render.mp4')).resolves.toBe(
|
|
'https://gateway.test/api/files/download?path=%2Froot%2Foutputs%2Frender.mp4&token=s%20e%2Fcret'
|
|
)
|
|
})
|
|
|
|
it('uses the Electron streaming protocol for local desktop video', async () => {
|
|
vi.stubGlobal('window', { hermesDesktop: { api: vi.fn() } })
|
|
$connection.set({ mode: 'local' } as never)
|
|
|
|
await expect(resolveMediaPlaybackSrc('C:\\renders\\demo.mp4')).resolves.toBe(
|
|
'hermes-media://stream/C%3A%5Crenders%5Cdemo.mp4'
|
|
)
|
|
})
|
|
})
|
|
|
|
describe('gatewayMediaDataUrl', () => {
|
|
const api = vi.fn(async ({ path }: { path: string }) => {
|
|
if (path.startsWith('/api/fs/read-data-url?')) {
|
|
return { dataUrl: 'data:image/png;base64,ZHVtbXk=' }
|
|
}
|
|
|
|
throw new Error(`unexpected path ${path}`)
|
|
})
|
|
|
|
beforeEach(() => {
|
|
api.mockClear()
|
|
vi.stubGlobal('window', { hermesDesktop: { api } })
|
|
$connection.set({ mode: 'remote' } as never)
|
|
})
|
|
|
|
afterEach(() => {
|
|
vi.unstubAllGlobals()
|
|
$connection.set(null)
|
|
})
|
|
|
|
it('reads gateway media through the desktop fs bridge instead of /api/media roots', async () => {
|
|
const url = await gatewayMediaDataUrl('/home/u/.hermes/skills/demo/images/a b.png')
|
|
|
|
expect(url).toBe('data:image/png;base64,ZHVtbXk=')
|
|
expect(api).toHaveBeenCalledWith({
|
|
path: '/api/fs/read-data-url?path=%2Fhome%2Fu%2F.hermes%2Fskills%2Fdemo%2Fimages%2Fa%20b.png'
|
|
})
|
|
})
|
|
})
|
|
|
|
describe('downloadGatewayMediaFile', () => {
|
|
const api = vi.fn(async ({ path }: { path: string }) => {
|
|
if (path.startsWith('/api/fs/read-data-url?')) {
|
|
return { dataUrl: 'data:text/markdown;base64,IyByZXBvcnQ=' }
|
|
}
|
|
|
|
throw new Error(`unexpected path ${path}`)
|
|
})
|
|
|
|
let clickSpy: ReturnType<typeof vi.spyOn>
|
|
|
|
beforeEach(() => {
|
|
api.mockClear()
|
|
vi.stubGlobal('window', { hermesDesktop: { api }, setTimeout: vi.fn() })
|
|
vi.stubGlobal(
|
|
'fetch',
|
|
vi.fn(async () => ({ blob: async () => new Blob(['# report'], { type: 'text/markdown' }) }))
|
|
)
|
|
URL.createObjectURL = vi.fn(() => 'blob:remote-artifact')
|
|
URL.revokeObjectURL = vi.fn()
|
|
clickSpy = vi.spyOn(HTMLAnchorElement.prototype, 'click').mockImplementation(() => {})
|
|
$connection.set({ mode: 'remote' } as never)
|
|
})
|
|
|
|
afterEach(() => {
|
|
vi.unstubAllGlobals()
|
|
vi.clearAllMocks()
|
|
clickSpy.mockRestore()
|
|
$connection.set(null)
|
|
})
|
|
|
|
it('downloads gateway files through the desktop fs bridge', async () => {
|
|
await downloadGatewayMediaFile('file:///Users/me/project/report.md')
|
|
|
|
expect(api).toHaveBeenCalledWith({
|
|
path: '/api/fs/read-data-url?path=%2FUsers%2Fme%2Fproject%2Freport.md'
|
|
})
|
|
expect(clickSpy).toHaveBeenCalledOnce()
|
|
})
|
|
|
|
it('rejects when the gateway refuses the file read', async () => {
|
|
api.mockRejectedValueOnce(new Error('403 File is not readable'))
|
|
|
|
await expect(downloadGatewayMediaFile('/Users/me/project/report.md')).rejects.toThrow('403')
|
|
expect(clickSpy).not.toHaveBeenCalled()
|
|
})
|
|
})
|