// plugins/auto-refresh/src/index.ts
import type { PiniaColadaPlugin, UseQueryEntry } from '@pinia/colada'

export interface PiniaColadaAutoRefetchOptions {
  /**
   * Whether to enable auto refresh by default.
   * Can be overridden per query.
   * @default true
   */
  enabled?: boolean
}

/**
 * Plugin that automatically refreshes queries when they become stale
 */
export function PiniaColadaAutoRefetch(
  options: PiniaColadaAutoRefetchOptions = {},
): PiniaColadaPlugin {
  // Default to enabled
  const { enabled = true } = options

  return ({ queryCache, scope }) => {
    // Keep track of active entries and their timeouts
    const refreshTimeouts = new Map<UseQueryEntry, NodeJS.Timeout>()

    queryCache.$onAction(({ name, after, args }) => {
      if (name === 'create') {
        after((entry) => {
          // Skip if auto-refresh is disabled globally or for this query
          const queryEnabled = entry.options?.autoRefetch ?? enabled
          if (!queryEnabled) return

          scope.run(() => {
            console.log('>> Running', entry.key)

            watch(
              () => entry.stale,
              (stale) => {
                console.log('>> Stale', stale)
                if (stale) {
                  // Clear any existing timeout
                  const existingTimeout = refreshTimeouts.get(entry)
                  if (existingTimeout) {
                    clearTimeout(existingTimeout)
                  }

                  // Schedule a refresh
                  const timeout = setTimeout(() => {
                    if (entry.active) {
                      console.log('>> Refreshing stale query:', entry.key)
                      queryCache.fetch(entry)
                    }
                    refreshTimeouts.delete(entry)
                  }, 0)

                  refreshTimeouts.set(entry, timeout)
                }
              },
              { immediate: true },
            )
          })
        })
      }

      // Clean up timeouts when entry is removed
      if (name === 'remove') {
        console.log('>> Remove', args)
        const [entry] = args
        const timeout = refreshTimeouts.get(entry)
        if (timeout) {
          clearTimeout(timeout)
          refreshTimeouts.delete(entry)
        }
      }
    })
  }
}

// Add types for the new option
declare module '@pinia/colada' {
  interface UseQueryOptions {
    /**
     * Whether to automatically refresh this query when it becomes stale.
     * Requires the PiniaColadaAutoRefetch plugin.
     */
    autoRefetch?: boolean
  }
}