Add 'aiui/' from commit 'e30ac1d1069532fb6d652d87e2d4a2fe9d1b4773'
git-subtree-dir: aiui git-subtree-mainline:0c4826f8ccgit-subtree-split:e30ac1d106
This commit is contained in:
archipelago
@@ -0,0 +1,307 @@
|
||||
import { ref, readonly } from 'vue'
|
||||
import { archyBridge } from '@/services/archyBridge'
|
||||
import { useTheme } from '@/composables/useTheme'
|
||||
import {
|
||||
mockArchyApps, mockArchySystem, mockArchyNetwork,
|
||||
mockArchyWallet, mockArchyBitcoin, mockArchyFiles,
|
||||
} from '@/mocks/archy'
|
||||
|
||||
type AIContextCategory = 'apps' | 'system' | 'network' | 'wallet' | 'files' | 'bitcoin'
|
||||
|
||||
interface ArchyApp {
|
||||
id: string
|
||||
name: string
|
||||
state: string
|
||||
status: string
|
||||
}
|
||||
|
||||
interface ArchySystemInfo {
|
||||
version?: string
|
||||
name?: string
|
||||
}
|
||||
|
||||
interface ArchyNetworkInfo {
|
||||
connected?: boolean
|
||||
}
|
||||
|
||||
export interface ArchyWalletInfo {
|
||||
available?: boolean
|
||||
status?: string
|
||||
alias?: string
|
||||
num_active_channels?: number
|
||||
num_peers?: number
|
||||
synced_to_chain?: boolean
|
||||
block_height?: number
|
||||
balance_sats?: number
|
||||
channel_balance_sats?: number
|
||||
pending_open_balance?: number
|
||||
message?: string
|
||||
}
|
||||
|
||||
export interface ArchyFileEntry {
|
||||
name: string
|
||||
path: string
|
||||
size?: number
|
||||
modified?: string
|
||||
type: 'file' | 'folder'
|
||||
}
|
||||
|
||||
export interface ArchyBitcoinInfo {
|
||||
available: boolean
|
||||
block_height?: number
|
||||
sync_progress?: number
|
||||
chain?: string
|
||||
mempool_tx_count?: number
|
||||
mempool_size?: number
|
||||
}
|
||||
|
||||
// Singleton reactive state (shared across all components using this composable)
|
||||
const isEmbedded = ref(false)
|
||||
const isInitialized = ref(false)
|
||||
const permissions = ref<AIContextCategory[]>([])
|
||||
const accentColor = ref<string | null>(null)
|
||||
const installedApps = ref<ArchyApp[]>([])
|
||||
const systemInfo = ref<ArchySystemInfo>({})
|
||||
const networkInfo = ref<ArchyNetworkInfo>({})
|
||||
const walletInfo = ref<ArchyWalletInfo>({})
|
||||
const fileList = ref<ArchyFileEntry[]>([])
|
||||
const bitcoinInfo = ref<ArchyBitcoinInfo>({ available: false })
|
||||
let cleanups: (() => void)[] = []
|
||||
|
||||
/**
|
||||
* Reactive composable wrapping archyBridge for Archy ↔ AIUI integration.
|
||||
* Call `init()` once in App.vue when `?embedded=true` is detected.
|
||||
*/
|
||||
export function useArchy() {
|
||||
/** Initialize the bridge and start listening for Archy messages */
|
||||
function init() {
|
||||
if (isInitialized.value) return
|
||||
|
||||
const embedded = !!(window as unknown as Record<string, unknown>).__AIUI_EMBEDDED__
|
||||
isEmbedded.value = embedded
|
||||
|
||||
// Dev mock mode: load realistic Archy data for standalone testing
|
||||
const useMock = import.meta.env.VITE_MOCK_ARCHY === 'true' ||
|
||||
new URLSearchParams(window.location.search).has('mockArchy')
|
||||
if (useMock && !embedded) {
|
||||
isInitialized.value = true
|
||||
isEmbedded.value = true
|
||||
permissions.value = ['apps', 'system', 'network', 'wallet', 'bitcoin', 'files']
|
||||
installedApps.value = mockArchyApps as unknown as ArchyApp[]
|
||||
systemInfo.value = mockArchySystem
|
||||
networkInfo.value = mockArchyNetwork
|
||||
walletInfo.value = mockArchyWallet
|
||||
bitcoinInfo.value = mockArchyBitcoin
|
||||
fileList.value = mockArchyFiles
|
||||
console.log('[AIUI] Mock Archy data loaded for dev testing')
|
||||
return
|
||||
}
|
||||
|
||||
if (!embedded || !archyBridge.isInArchy()) return
|
||||
|
||||
archyBridge.init()
|
||||
isInitialized.value = true
|
||||
|
||||
// Listen for permission updates
|
||||
const unsubPerms = archyBridge.onPermissionsUpdate((cats) => {
|
||||
permissions.value = cats
|
||||
// Auto-fetch context for newly permitted categories
|
||||
fetchPermittedContext(cats)
|
||||
})
|
||||
cleanups.push(unsubPerms)
|
||||
|
||||
// Listen for theme updates — Archy always reports mode:'dark' today, but
|
||||
// honor whatever it sends rather than hardcoding that assumption here;
|
||||
// App.vue's mount-time isEmbeddedFlag check already forces dark
|
||||
// immediately without waiting for this round trip, so this is a
|
||||
// corroborating update for whenever it does arrive.
|
||||
const unsubTheme = archyBridge.onThemeUpdate((theme) => {
|
||||
accentColor.value = theme.accent
|
||||
applyAccentColor(theme.accent)
|
||||
useTheme().setTheme(theme.mode)
|
||||
})
|
||||
cleanups.push(unsubTheme)
|
||||
|
||||
// Request theme on init
|
||||
archyBridge.requestTheme()
|
||||
}
|
||||
|
||||
/** Fetch context for all permitted categories */
|
||||
async function fetchPermittedContext(cats: AIContextCategory[]) {
|
||||
const fetches: Promise<void>[] = []
|
||||
|
||||
function fetchCategory<T>(cat: AIContextCategory, setter: (data: T) => void, validator: (data: unknown) => boolean = () => true) {
|
||||
return archyBridge.requestContext(cat).then((res) => {
|
||||
if (!res.permitted) {
|
||||
console.warn(`[AIUI Archy] ${cat}: not permitted — user should enable in Archy Settings`)
|
||||
return
|
||||
}
|
||||
if (res.data && validator(res.data)) {
|
||||
setter(res.data as T)
|
||||
}
|
||||
}).catch((err) => {
|
||||
console.warn(`[AIUI Archy] ${cat} fetch failed:`, err?.message ?? err)
|
||||
})
|
||||
}
|
||||
|
||||
if (cats.includes('apps')) {
|
||||
fetches.push(fetchCategory('apps', (data) => { installedApps.value = data as ArchyApp[] }, Array.isArray))
|
||||
}
|
||||
|
||||
if (cats.includes('system')) {
|
||||
fetches.push(fetchCategory('system', (data) => { systemInfo.value = data as ArchySystemInfo }))
|
||||
}
|
||||
|
||||
if (cats.includes('network')) {
|
||||
fetches.push(fetchCategory('network', (data) => { networkInfo.value = data as ArchyNetworkInfo }))
|
||||
}
|
||||
|
||||
if (cats.includes('wallet')) {
|
||||
fetches.push(fetchCategory('wallet', (data) => { walletInfo.value = data as ArchyWalletInfo }))
|
||||
}
|
||||
|
||||
if (cats.includes('bitcoin')) {
|
||||
fetches.push(fetchCategory('bitcoin', (data) => { bitcoinInfo.value = data as ArchyBitcoinInfo }))
|
||||
}
|
||||
|
||||
if (cats.includes('files')) {
|
||||
fetches.push(fetchCategory('files', (data) => { fileList.value = data as ArchyFileEntry[] }, Array.isArray))
|
||||
}
|
||||
|
||||
await Promise.all(fetches)
|
||||
}
|
||||
|
||||
/** Refresh context data (call when user returns to chat) */
|
||||
async function refreshContext() {
|
||||
if (!isInitialized.value) return
|
||||
await fetchPermittedContext(permissions.value)
|
||||
}
|
||||
|
||||
/** Request Archy to perform an action */
|
||||
async function requestAction(action: string, params: Record<string, string> = {}) {
|
||||
if (!isInitialized.value) return { success: false, error: 'Not initialized' }
|
||||
return archyBridge.requestAction(action, params)
|
||||
}
|
||||
|
||||
/** Read a file's text content via FileBrowser */
|
||||
async function readFile(path: string): Promise<{ content: string; truncated: boolean; size: number } | null> {
|
||||
const res = await requestAction('read-file', { path })
|
||||
const data = (res as unknown as Record<string, unknown>).data
|
||||
if (res.success && data) {
|
||||
return data as { content: string; truncated: boolean; size: number }
|
||||
}
|
||||
return null
|
||||
}
|
||||
|
||||
/** Tail recent logs for an app */
|
||||
async function tailLogs(appId: string, lines = 50): Promise<string[] | null> {
|
||||
const res = await requestAction('tail-logs', { appId, lines: String(lines) })
|
||||
const data = (res as unknown as Record<string, unknown>).data
|
||||
if (res.success && data) {
|
||||
return (data as { lines: string[] }).lines
|
||||
}
|
||||
return null
|
||||
}
|
||||
|
||||
/** Apply accent color as CSS custom property */
|
||||
function applyAccentColor(color: string) {
|
||||
document.documentElement.style.setProperty('--color-accent', color)
|
||||
}
|
||||
|
||||
/** Build context string for AI system prompt */
|
||||
function buildArchyContext(): string {
|
||||
if (!isInitialized.value) return ''
|
||||
|
||||
const sections: string[] = []
|
||||
|
||||
if (permissions.value.includes('apps') && installedApps.value.length > 0) {
|
||||
const appList = installedApps.value
|
||||
.map((a) => `- ${a.name} (${a.state}${a.status ? ', ' + a.status : ''})`)
|
||||
.join('\n')
|
||||
sections.push(`**Installed apps on this node:**\n${appList}\nYou can view recent app logs by requesting the tail-logs action with an appId.`)
|
||||
}
|
||||
|
||||
if (permissions.value.includes('system') && systemInfo.value.name) {
|
||||
const sys = systemInfo.value
|
||||
sections.push(`**System:** ${sys.name}${sys.version ? ' v' + sys.version : ''}`)
|
||||
}
|
||||
|
||||
if (permissions.value.includes('network')) {
|
||||
const net = networkInfo.value
|
||||
sections.push(`**Network:** ${net.connected ? 'Connected' : 'Disconnected'}`)
|
||||
}
|
||||
|
||||
if (permissions.value.includes('wallet') && walletInfo.value.available) {
|
||||
const w = walletInfo.value
|
||||
const parts: string[] = []
|
||||
if (w.alias) parts.push(w.alias)
|
||||
if (w.num_active_channels !== undefined) parts.push(`${w.num_active_channels} channels`)
|
||||
if (w.num_peers !== undefined) parts.push(`${w.num_peers} peers`)
|
||||
if (w.balance_sats !== undefined) parts.push(`On-chain: ${w.balance_sats.toLocaleString()} sats`)
|
||||
if (w.channel_balance_sats !== undefined) parts.push(`In channels: ${w.channel_balance_sats.toLocaleString()} sats`)
|
||||
if (w.synced_to_chain !== undefined) parts.push(w.synced_to_chain ? 'synced' : 'syncing')
|
||||
sections.push(`**Lightning (LND):** ${parts.join(' | ')}`)
|
||||
}
|
||||
|
||||
if (permissions.value.includes('bitcoin') && bitcoinInfo.value.available) {
|
||||
const btc = bitcoinInfo.value
|
||||
const syncPct = btc.sync_progress ? (btc.sync_progress * 100).toFixed(2) + '%' : 'unknown'
|
||||
const parts = [`Block ${btc.block_height?.toLocaleString() ?? '?'}`, `${syncPct} synced`]
|
||||
if (btc.chain) parts.push(btc.chain)
|
||||
if (btc.mempool_tx_count) parts.push(`mempool: ${btc.mempool_tx_count.toLocaleString()} txs`)
|
||||
sections.push(`**Bitcoin:** ${parts.join(', ')}`)
|
||||
}
|
||||
|
||||
if (permissions.value.includes('files') && fileList.value.length > 0) {
|
||||
const files = fileList.value
|
||||
const folders = files.filter(f => f.type === 'folder')
|
||||
const fileItems = files.filter(f => f.type === 'file')
|
||||
const images = fileItems.filter(f => /\.(jpg|jpeg|png|gif|webp|svg|heic|heif)$/i.test(f.name))
|
||||
const videos = fileItems.filter(f => /\.(mp4|mkv|avi|mov|webm)$/i.test(f.name))
|
||||
const music = fileItems.filter(f => /\.(mp3|flac|wav|ogg|m4a|aac|opus)$/i.test(f.name))
|
||||
const docs = fileItems.filter(f => /\.(pdf|doc|docx|txt|md|ods|xlsx|csv)$/i.test(f.name))
|
||||
|
||||
const parts: string[] = [`${files.length} items`]
|
||||
if (folders.length > 0) parts.push(`${folders.length} folders (${folders.map(f => f.name).join(', ')})`)
|
||||
if (images.length > 0) parts.push(`${images.length} images`)
|
||||
if (videos.length > 0) parts.push(`${videos.length} videos`)
|
||||
if (music.length > 0) parts.push(`${music.length} audio files`)
|
||||
if (docs.length > 0) parts.push(`${docs.length} documents`)
|
||||
|
||||
const recent = fileItems.slice(0, 15).map(f => f.name).join(', ')
|
||||
sections.push(`**Files:** ${parts.join(' | ')}\nRecent: ${recent}\nYou can read file contents by requesting the read-file action with a file path.`)
|
||||
}
|
||||
|
||||
if (sections.length === 0) return ''
|
||||
|
||||
return `\n\n**Archy Node Context** (this user is running AIUI on their Archipelago node):\n${sections.join('\n')}\n\nYou can help the user manage their node, check service status, browse files, and recommend apps. Available actions: open an app (open-app), install an app (install-app), tail app logs (tail-logs), read a file (read-file), navigate in Archy (navigate). When recommending apps, use [[app_ext:...]] tags and check if they're already installed. When discussing the user's files, mention specific files you can see. If the user asks about their photos, videos, or music, reference the file counts above.`
|
||||
}
|
||||
|
||||
/** Clean up on component unmount */
|
||||
function destroy() {
|
||||
for (const cleanup of cleanups) cleanup()
|
||||
cleanups = []
|
||||
archyBridge.destroy()
|
||||
isInitialized.value = false
|
||||
}
|
||||
|
||||
return {
|
||||
isEmbedded: readonly(isEmbedded),
|
||||
isInitialized: readonly(isInitialized),
|
||||
permissions: readonly(permissions),
|
||||
accentColor: readonly(accentColor),
|
||||
installedApps: readonly(installedApps),
|
||||
systemInfo: readonly(systemInfo),
|
||||
networkInfo: readonly(networkInfo),
|
||||
walletInfo: readonly(walletInfo),
|
||||
fileList: readonly(fileList),
|
||||
bitcoinInfo: readonly(bitcoinInfo),
|
||||
init,
|
||||
destroy,
|
||||
refreshContext,
|
||||
requestAction,
|
||||
readFile,
|
||||
tailLogs,
|
||||
buildArchyContext,
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user