reset.ts

import {Args, Command, Flags} from '@oclif/core'

import {
  SettingsEvents,
  type SettingsGetRequest,
  type SettingsGetResponse,
  type SettingsItemDTO,
  type SettingsResetRequest,
  type SettingsResetResponse,
} from '../../../shared/transport/events/settings-events.js'
import {formatCount, formatDuration} from '../../../shared/utils/format-duration.js'
import {type DaemonClientOptions, formatConnectionError, withDaemonRetry} from '../../lib/daemon-client.js'
import {writeJsonResponse} from '../../lib/json-response.js'

export default class SettingsReset extends Command {
  public static args = {
    key: Args.string({description: 'Settings key to reset', required: true}),
  }
  public static description =
    'Restore one settings value to its default. Changes apply after `brv restart`.'
  public static examples = [
    '<%= config.bin %> settings reset agentPool.maxSize',
    '<%= config.bin %> settings reset agentPool.maxSize --format json',
  ]
  public static flags = {
    format: Flags.string({
      default: 'text',
      description: 'Output format (text or json)',
      options: ['text', 'json'],
    }),
  }

  protected async fetchDescriptor(key: string, options?: DaemonClientOptions): Promise<SettingsGetResponse> {
    return withDaemonRetry<SettingsGetResponse>(
      async (client) =>
        client.requestWithAck<SettingsGetResponse>(SettingsEvents.GET, {key} satisfies SettingsGetRequest),
      options,
    )
  }

  protected async resetSetting(key: string, options?: DaemonClientOptions): Promise<SettingsResetResponse> {
    return withDaemonRetry<SettingsResetResponse>(
      async (client) =>
        client.requestWithAck<SettingsResetResponse>(SettingsEvents.RESET, {key} satisfies SettingsResetRequest),
      options,
    )
  }

  public async run(): Promise<void> {
    const {args, flags} = await this.parse(SettingsReset)
    const format = flags.format as 'json' | 'text'

    try {
      const descriptor = await this.fetchDescriptor(args.key)
      if (!descriptor.ok) {
        process.exitCode = 1
        if (format === 'json') {
          writeJsonResponse({command: 'settings reset', data: {error: descriptor.error}, success: false})
        } else {
          this.log(descriptor.error.message)
        }

        return
      }

      const response = await this.resetSetting(args.key)

      if (response.ok) {
        if (format === 'json') {
          writeJsonResponse({
            command: 'settings reset',
            data: {restartRequired: response.restartRequired},
            success: true,
          })
        } else {
          const base = `Setting reset: ${args.key} back to default (${renderValue(descriptor, descriptor.default)}).`
          this.log(descriptor.restartRequired ? `${base} Run \`brv restart\` to apply.` : base)
        }

        return
      }

      process.exitCode = 1
      if (format === 'json') {
        writeJsonResponse({command: 'settings reset', data: {error: response.error}, success: false})
      } else {
        this.log(response.error.message)
      }
    } catch (error) {
      process.exitCode = 1
      if (format === 'json') {
        writeJsonResponse({command: 'settings reset', data: {error: formatConnectionError(error)}, success: false})
      } else {
        this.log(formatConnectionError(error))
      }
    }
  }
}

function renderValue(item: SettingsItemDTO, value: boolean | number | string): string {
  if (typeof value === 'boolean') return value ? 'true' : 'false'
  if (typeof value === 'string') return value
  if (item.unit === 'ms') return formatDuration(value)
  return formatCount(value)
}