-- SPDX-FileCopyrightText: 2026 Yuki Otsuka
--
-- SPDX-License-Identifier: BSD-3

import ViE
import ViE.Color

open ViE.Color

def customConfig : ViE.EditorConfig := {
  ViE.defaultConfig with
  -- View
  showLineNumbers := true
  emptyLineMarker := "・"

  -- Splits
  vSplitStr := "│"
  hSplitStr := "─"

  -- Cursor / selection / search
  searchHighlightStyle := (toBg Color.white) ++ (toFg Color.black)
  searchHighlightCursorStyle := (toBg Color.yellow) ++ (toFg Color.black)
  visualSelectionStyle := "\x1b[7m"
  cursorCharStyle := (toBg Color.white) ++ (toFg Color.black)
  cursorSpaceStyle := (toBg Color.white) ++ (toFg Color.black)

  -- Explorer / preview
  explorerPreviewSplitRatio := 0.3

  -- Editing
  tabStop := 2

  -- Status line / terminal reset
  statusBarStyle := (toBg Color.brightBlack) ++ (toFg Color.white)
  resetStyle := ViE.Color.reset

  -- Explorer icons
  fileIcon := "📄 "
  dirIcon := "📁 "

  -- Search bloom cache
  searchBloomCacheMax := 1024
  searchBloomBuildLeafBits := true
  searchBloomBuildOnEdit := false

  -- History
  historyLimit := 200

  -- Examples:
  -- fileIcon := "File: "
  -- dirIcon := "Dir : "
  -- explorerPreviewSplitRatio := 0.62
  -- tabStop := 4
  -- showLineNumbers := false
}

def customKeyMap (commands : ViE.CommandMap) : ViE.KeyMap := {
  ViE.Key.makeKeyMap commands with
  normal := fun s k  => match k with
    -- Example custom binding: 'X' to quit
    | ViE.Key.char 'X' => pure { s with shouldQuit := true }
    | _ => (ViE.Key.makeKeyMap commands).normal s k
  visual := fun s k => match k with
    | ViE.Key.char 'o' => pure (ViE.Window.cycleWindow s)
    | _ => (ViE.Key.makeKeyMap commands).visual s k
}

def myCustomCommand (_ : List String) (state : ViE.EditorState) : IO ViE.EditorState := do
  return { state with message := "Executed custom command!" }

def main (args : List String) : IO Unit := do
  let myCommands : ViE.CommandMap := [
    ("mycmd", myCustomCommand),
    ("o", ViE.Command.cmdWinCycle)
  ]
  let commands := ViE.Command.defaultCommandMap ++ myCommands
  let languageConfigs : Lean.RBMap String ViE.LanguageRuntimeConfig compare :=
    let m0 : Lean.RBMap String ViE.LanguageRuntimeConfig compare := Lean.RBMap.empty
    let m1 := m0.insert "lean" ({ autoStartLsp := true, autoOpenInfoView := true } : ViE.LanguageRuntimeConfig)
    m1.insert "markdown" ({ autoStartLsp := false, autoOpenInfoView := false } : ViE.LanguageRuntimeConfig)

  let config : ViE.Config := {
    settings := customConfig
    bindings := customKeyMap commands
    commands := commands
    languageConfigs := languageConfigs
  }

  ViE.start config args