@shareai-lab/kode 1.0.9

package/src/commands.ts

@@ -0,0 +1,139 @@
+import React from 'react'
+import bug from './commands/bug'
+import clear from './commands/clear'
+import compact from './commands/compact'
+import config from './commands/config'
+import cost from './commands/cost'
+import ctx_viz from './commands/ctx_viz'
+import doctor from './commands/doctor'
+import help from './commands/help'
+import init from './commands/init'
+import listen from './commands/listen'
+import login from './commands/login'
+import logout from './commands/logout'
+import mcp from './commands/mcp'
+import * as model from './commands/model'
+import modelstatus from './commands/modelstatus'
+import onboarding from './commands/onboarding'
+import pr_comments from './commands/pr_comments'
+import refreshCommands from './commands/refreshCommands'
+import releaseNotes from './commands/release-notes'
+import review from './commands/review'
+import terminalSetup from './commands/terminalSetup'
+import { Tool, ToolUseContext } from './Tool'
+import resume from './commands/resume'
+import agents from './commands/agents'
+import { getMCPCommands } from './services/mcpClient'
+import { loadCustomCommands } from './services/customCommands'
+import type { MessageParam } from '@anthropic-ai/sdk/resources/index.mjs'
+import { memoize } from 'lodash-es'
+import type { Message } from './query'
+import { isAnthropicAuthEnabled } from './utils/auth'
+
+type PromptCommand = {
+  type: 'prompt'
+  progressMessage: string
+  argNames?: string[]
+  getPromptForCommand(args: string): Promise<MessageParam[]>
+}
+
+type LocalCommand = {
+  type: 'local'
+  call(
+    args: string,
+    context: {
+      options: {
+        commands: Command[]
+        tools: Tool[]
+        slowAndCapableModel: string
+      }
+      abortController: AbortController
+      setForkConvoWithMessagesOnTheNextRender: (
+        forkConvoWithMessages: Message[],
+      ) => void
+    },
+  ): Promise<string>
+}
+
+type LocalJSXCommand = {
+  type: 'local-jsx'
+  call(
+    onDone: (result?: string) => void,
+    context: ToolUseContext & {
+      setForkConvoWithMessagesOnTheNextRender: (
+        forkConvoWithMessages: Message[],
+      ) => void
+    },
+  ): Promise<React.ReactNode>
+}
+
+export type Command = {
+  description: string
+  isEnabled: boolean
+  isHidden: boolean
+  name: string
+  aliases?: string[]
+  userFacingName(): string
+} & (PromptCommand | LocalCommand | LocalJSXCommand)
+
+const INTERNAL_ONLY_COMMANDS = [ctx_viz, resume, listen]
+
+// Declared as a function so that we don't run this until getCommands is called,
+// since underlying functions read from config, which can't be read at module initialization time
+const COMMANDS = memoize((): Command[] => [
+  agents,
+  clear,
+  compact,
+  config,
+  cost,
+  doctor,
+  help,
+  init,
+  mcp,
+  model,
+  modelstatus,
+  onboarding,
+  pr_comments,
+  refreshCommands,
+  releaseNotes,
+  bug,
+  review,
+  terminalSetup,
+  ...(isAnthropicAuthEnabled() ? [logout, login()] : []),
+  ...INTERNAL_ONLY_COMMANDS,
+])
+
+export const getCommands = memoize(async (): Promise<Command[]> => {
+  const [mcpCommands, customCommands] = await Promise.all([
+    getMCPCommands(),
+    loadCustomCommands(),
+  ])
+
+  return [...mcpCommands, ...customCommands, ...COMMANDS()].filter(
+    _ => _.isEnabled,
+  )
+})
+
+export function hasCommand(commandName: string, commands: Command[]): boolean {
+  return commands.some(
+    _ => _.userFacingName() === commandName || _.aliases?.includes(commandName),
+  )
+}
+
+export function getCommand(commandName: string, commands: Command[]): Command {
+  const command = commands.find(
+    _ => _.userFacingName() === commandName || _.aliases?.includes(commandName),
+  ) as Command | undefined
+  if (!command) {
+    throw ReferenceError(
+      `Command ${commandName} not found. Available commands: ${commands
+        .map(_ => {
+          const name = _.userFacingName()
+          return _.aliases ? `${name} (aliases: ${_.aliases.join(', ')})` : name
+        })
+        .join(', ')}`,
+    )
+  }
+
+  return command
+}

package/src/components/ApproveApiKey.tsx

@@ -0,0 +1,93 @@
+import React from 'react'
+import { Box, Text } from 'ink'
+import { getGlobalConfig, saveGlobalConfig } from '../utils/config'
+import { getTheme } from '../utils/theme'
+import { Select } from './CustomSelect/select'
+import { useExitOnCtrlCD } from '../hooks/useExitOnCtrlCD'
+import chalk from 'chalk'
+
+type Props = {
+  customApiKeyTruncated: string
+  onDone(): void
+}
+
+export function ApproveApiKey({
+  customApiKeyTruncated,
+  onDone,
+}: Props): React.ReactNode {
+  const theme = getTheme()
+
+  function onChange(value: 'yes' | 'no') {
+    const config = getGlobalConfig()
+    switch (value) {
+      case 'yes': {
+        saveGlobalConfig({
+          ...config,
+          customApiKeyResponses: {
+            ...config.customApiKeyResponses,
+            approved: [
+              ...(config.customApiKeyResponses?.approved ?? []),
+              customApiKeyTruncated,
+            ],
+          },
+        })
+        onDone()
+        break
+      }
+      case 'no': {
+        saveGlobalConfig({
+          ...config,
+          customApiKeyResponses: {
+            ...config.customApiKeyResponses,
+            rejected: [
+              ...(config.customApiKeyResponses?.rejected ?? []),
+              customApiKeyTruncated,
+            ],
+          },
+        })
+        onDone()
+        break
+      }
+    }
+  }
+
+  const exitState = useExitOnCtrlCD(() => process.exit(0))
+
+  return (
+    <>
+      <Box
+        flexDirection="column"
+        gap={1}
+        padding={1}
+        borderStyle="round"
+        borderColor={theme.warning}
+      >
+        <Text bold color={theme.warning}>
+          Detected a custom API key in your environment
+        </Text>
+        <Text>
+          Your environment sets{' '}
+          <Text color={theme.warning}>ANTHROPIC_API_KEY</Text>:{' '}
+          <Text bold>sk-ant-...{customApiKeyTruncated}</Text>
+        </Text>
+        <Text>Do you want to use this API key?</Text>
+        <Select
+          options={[
+            { label: `No (${chalk.bold('recommended')})`, value: 'no' },
+            { label: 'Yes', value: 'yes' },
+          ]}
+          onChange={value => onChange(value as 'yes' | 'no')}
+        />
+      </Box>
+      <Box marginLeft={3}>
+        <Text dimColor>
+          {exitState.pending ? (
+            <>Press {exitState.keyName} again to exit</>
+          ) : (
+            <>Enter to confirm</>
+          )}
+        </Text>
+      </Box>
+    </>
+  )
+}

package/src/components/AsciiLogo.tsx

@@ -0,0 +1,13 @@
+import { Box, Text } from 'ink'
+import React from 'react'
+import { getTheme } from '../utils/theme'
+import { ASCII_LOGO } from '../constants/product'
+
+export function AsciiLogo(): React.ReactNode {
+  const theme = getTheme()
+  return (
+    <Box flexDirection="column" alignItems="flex-start">
+      <Text color={theme.kode}>{ASCII_LOGO}</Text>
+    </Box>
+  )
+}

package/src/components/AutoUpdater.tsx

@@ -0,0 +1,148 @@
+import { Box, Text } from 'ink'
+import * as React from 'react'
+import { getTheme } from '../utils/theme'
+import { gte } from 'semver'
+import { useEffect, useState } from 'react'
+import { isAutoUpdaterDisabled } from '../utils/config'
+import {
+  AutoUpdaterResult,
+  getLatestVersion,
+  installGlobalPackage,
+} from '../utils/autoUpdater.js'
+import { useInterval } from '../hooks/useInterval'
+import { logEvent } from '../services/statsig'
+import { MACRO } from '../constants/macros'
+import { PRODUCT_COMMAND } from '../constants/product'
+type Props = {
+  debug: boolean
+  isUpdating: boolean
+  onChangeIsUpdating: (isUpdating: boolean) => void
+  onAutoUpdaterResult: (autoUpdaterResult: AutoUpdaterResult) => void
+  autoUpdaterResult: AutoUpdaterResult | null
+}
+
+export function AutoUpdater({
+  debug,
+  isUpdating,
+  onChangeIsUpdating,
+  onAutoUpdaterResult,
+  autoUpdaterResult,
+}: Props): React.ReactNode {
+  const theme = getTheme()
+  const [versions, setVersions] = useState<{
+    global?: string | null
+    latest?: string | null
+  }>({})
+  const checkForUpdates = React.useCallback(async () => {
+    if (process.env.NODE_ENV === 'test' || process.env.NODE_ENV === 'dev') {
+      return
+    }
+
+    if (isUpdating) {
+      return
+    }
+
+    // Get versions
+    const globalVersion = MACRO.VERSION
+    const latestVersion = await getLatestVersion()
+    const isDisabled = true //await isAutoUpdaterDisabled()
+
+    setVersions({ global: globalVersion, latest: latestVersion })
+
+    // Check if update needed and perform update
+    if (
+      !isDisabled &&
+      globalVersion &&
+      latestVersion &&
+      !gte(globalVersion, latestVersion)
+    ) {
+      const startTime = Date.now()
+      onChangeIsUpdating(true)
+      const installStatus = await installGlobalPackage()
+      onChangeIsUpdating(false)
+
+      if (installStatus === 'success') {
+        logEvent('tengu_auto_updater_success', {
+          fromVersion: globalVersion,
+          toVersion: latestVersion,
+          durationMs: String(Date.now() - startTime),
+        })
+      } else {
+        logEvent('tengu_auto_updater_fail', {
+          fromVersion: globalVersion,
+          attemptedVersion: latestVersion,
+          status: installStatus,
+          durationMs: String(Date.now() - startTime),
+        })
+      }
+
+      onAutoUpdaterResult({
+        version: latestVersion!,
+        status: installStatus,
+      })
+    }
+    // Don't re-render when isUpdating changes
+    // TODO: Find a cleaner way to do this
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, [onAutoUpdaterResult])
+
+  // Initial check
+  useEffect(() => {
+    // checkForUpdates()
+  }, [checkForUpdates])
+
+  // Check every 30 minutes
+  // useInterval(checkForUpdates, 30 * 60 * 1000)
+
+  if (debug) {
+    return (
+      <Box flexDirection="row">
+        <Text dimColor>
+          globalVersion: {versions.global} &middot; latestVersion:{' '}
+          {versions.latest}
+        </Text>
+      </Box>
+    )
+  }
+
+  if (!autoUpdaterResult?.version && (!versions.global || !versions.latest)) {
+    return null
+  }
+
+  if (!autoUpdaterResult?.version && !isUpdating) {
+    return null
+  }
+
+  return (
+    <Box flexDirection="row">
+      {debug && (
+        <Text dimColor>
+          globalVersion: {versions.global} &middot; latestVersion:{' '}
+          {versions.latest}
+        </Text>
+      )}
+      {isUpdating && (
+        <>
+          <Box>
+            <Text color={theme.secondaryText} dimColor wrap="end">
+              Auto-updating to v{versions.latest}…
+            </Text>
+          </Box>
+        </>
+      )}
+      {autoUpdaterResult?.status === 'success' && autoUpdaterResult?.version ? (
+        <Text color={theme.success}>
+          ✓ Update installed &middot; Restart to apply
+        </Text>
+      ) : null}
+      {(autoUpdaterResult?.status === 'install_failed' ||
+        autoUpdaterResult?.status === 'no_permissions') && (
+        <Text color={theme.error}>
+          ✗ Auto-update failed &middot; Try{' '}
+          <Text bold>{PRODUCT_COMMAND} doctor</Text> or{' '}
+          <Text bold>npm i -g {MACRO.PACKAGE_URL}</Text>
+        </Text>
+      )}
+    </Box>
+  )
+}