import {autorun} from 'mobx'
import {Platform} from 'react-native'
import {sessionClient as AtpApi, SessionServiceClient} from '@atproto/api'
import {RootStoreModel} from './models/root-store'
import * as apiPolyfill from './lib/api-polyfill'
import * as storage from './lib/storage'

export const LOCAL_DEV_SERVICE =
  Platform.OS === 'ios' ? 'http://localhost:2583' : 'http://10.0.2.2:2583'
export const STAGING_SERVICE = 'https://pds.staging.bsky.dev'
export const PROD_SERVICE = 'https://bsky.social'
export const DEFAULT_SERVICE = PROD_SERVICE
const ROOT_STATE_STORAGE_KEY = 'root'
const STATE_FETCH_INTERVAL = 15e3

export async function setupState(serviceUri = DEFAULT_SERVICE) {
  let rootStore: RootStoreModel
  let data: any

  apiPolyfill.doPolyfill()

  const api = AtpApi.service(serviceUri) as SessionServiceClient
  rootStore = new RootStoreModel(api)
  try {
    data = (await storage.load(ROOT_STATE_STORAGE_KEY)) || {}
    rootStore.log.debug('Initial hydrate', {hasSession: !!data.session})
    rootStore.hydrate(data)
  } catch (e: any) {
    rootStore.log.error('Failed to load state from storage', e)
  }

  rootStore.session
    .connect()
    .then(() => {
      rootStore.log.debug('Session connected')
      return rootStore.fetchStateUpdate()
    })
    .catch((e: any) => {
      rootStore.log.warn('Failed initial connect', e)
    })
  // @ts-ignore .on() is correct -prf
  api.sessionManager.on('session', () => {
    if (!api.sessionManager.session && rootStore.session.hasSession) {
      // reset session
      rootStore.session.clear()
    } else if (api.sessionManager.session) {
      rootStore.session.updateAuthTokens(api.sessionManager.session)
    }
  })

  // track changes & save to storage
  autorun(() => {
    const snapshot = rootStore.serialize()
    storage.save(ROOT_STATE_STORAGE_KEY, snapshot)
  })

  // periodic state fetch
  setInterval(() => {
    rootStore.fetchStateUpdate()
  }, STATE_FETCH_INTERVAL)

  return rootStore
}

export {useStores, RootStoreModel, RootStoreProvider} from './models/root-store'