module-atproto-persistence / runtime / utils / utilAtprotoApplicationStates.ts

import { resolveActorServiceEndpoint, useAgent } from "#imports"

export function parseAtprotoStoreKey(
  key: string,
): { collection: string; rkey: string } | null {
  if (!key.startsWith('owd/')) return null

  const parts = key.split('/')

  if (parts[1] === 'application') {
    const last = parts[parts.length - 1]
    if (last === 'windows' || last === 'meta') {
      const nome = parts.slice(2, -1).join('/')
      return {
        collection: `org.owdproject.application.${last}`,
        rkey: nome,
      }
    }
  }

  // owd/desktop
  if (parts[1] === 'desktop') {
    const rkey = parts[2] ?? 'self'
    return {
      collection: 'org.owdproject.desktop',
      rkey,
    }
  }

  return null
}

export function listAtprotoApplicationStateRecords(
  agent: any,
  repo: string,
  collection: string,
) {
  return agent.com.atproto.repo
    .listRecords({
      repo,
      collection,
    })
    .then((response: any) => response.data)
    .then((data: any) => data.records)
    .catch(() => [])
}

export function getAtprotoApplicationState(
  agent: any,
  repo: string,
  collection: string,
  rkey: string,
) {
  return agent.com.atproto.repo.getRecord({
    repo,
    collection,
    rkey,
  })
}

export function putAtprotoApplicationState(
  agent: any,
  repo: string,
  collection: string,
  rkey: string,
  record: any,
) {
  return agent.com.atproto.repo.putRecord({
    repo,
    collection,
    rkey,
    record,
  })
}

/**
 * Load actor desktop
 *
 * @param actorDid
 */
export async function loadActorDesktopStateMap(actorDid: string) {
  const actorServiceEndpoint = await resolveActorServiceEndpoint(actorDid)
  const actorAgent = useAgent(actorServiceEndpoint)

  const [
    atprotoActorDesktopRecord,
    atprotoApplicationWindowsList,
    atprotoApplicationMetaList,
  ] = await Promise.allSettled([
    getAtprotoApplicationState(
      actorAgent,
      actorDid,
      'org.owdproject.desktop',
      'self',
    ),
    listAtprotoApplicationStateRecords(
      actorAgent,
      actorDid,
      'org.owdproject.application.windows',
    ),
    listAtprotoApplicationStateRecords(
      actorAgent,
      actorDid,
      'org.owdproject.application.meta',
    ),
  ])

  const stateMap: Record<string, any> = {}

  // desktop state
  if (
    atprotoActorDesktopRecord.status === 'fulfilled' &&
    atprotoActorDesktopRecord.value?.data
  ) {
    stateMap['owd/desktop'] = atprotoActorDesktopRecord.value.data.value
  }

  // windows state
  if (atprotoApplicationWindowsList.status === 'fulfilled') {
    for (const record of atprotoApplicationWindowsList.value) {
      const appId = record.uri.split('/').pop()
      stateMap[`owd/application/${appId}/windows`] = {
        windows: record.value.windows,
      }
    }
  }

  // meta state
  if (atprotoApplicationMetaList.status === 'fulfilled') {
    for (const record of atprotoApplicationMetaList.value) {
      const appId = record.uri.split('/').pop()
      stateMap[`owd/application/${appId}/meta`] = {
        ...record.value,
      }
    }
  }

  return stateMap
}