aboutsummaryrefslogtreecommitdiff
path: root/lib/input_prompt.ml
blob: aec60212dae75083d2ea374105fb179d72a9004d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
module Grep = Grep
module Common = Common
open Notty

type state =
  { user_input : string
  ; on_enter : string -> unit
  ; on_cancel : unit -> unit
  ; prompt : string
  ; screen : I.t
  }

let rec render t ({ user_input; on_enter; on_cancel; prompt; screen } as state) =
  let size_x, size_y = Common.Term.size t in
  Common.Term.cursor
    t
    (Some (String.length user_input + String.length prompt + 3, size_y));
  let img =
    let open I in
    I.strf " %s " ~attr:(A.st A.reverse) prompt
    <|> I.strf " %s%s" user_input (String.make size_x ' ')
    |> I.pad ~l:0 ~t:(size_y - 1)
    </> screen
  in
  Common.Term.image t img;
  match Common.Term.event t with
  | `End | `Key (`Escape, []) | `Key (`ASCII 'G', [ `Ctrl ]) | `Key (`ASCII 'C', [ `Ctrl ])
    -> on_cancel ()
  | `Key (`Enter, []) -> on_enter user_input
  | `Key (`Backspace, []) ->
    if String.equal "" user_input
    then on_cancel ()
    else (
      let state =
        { state with
          user_input = String.sub user_input 0 (max (String.length user_input - 1) 0)
        }
      in
      render t state)
  | `Resize _ -> render t state
  | `Key (`ASCII c, []) ->
    let state = { state with user_input = user_input ^ String.make 1 c } in
    render t state
  | _ -> render t state