aboutsummaryrefslogtreecommitdiff
path: root/lib/stitched_article.ml
blob: c71ca14e0cdbdcd1678a9c477c0d3fa264079bd1 (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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
module Grep = Grep
module Common = Common
open Notty
module Input_screen = Input_screen
module Help_screen = Help_screen

type state =
  { pos : int * int
  ; scroll : int
  ; content : (string * int * string * int) array
  ; go_back : unit -> unit
  ; content_pretty : Grep.display_type array
  ; todo_state : Todos.state option
  }

(* TODO: Use grep -l to filter notes by regexp and rerender those files*)
let rec render t ({ pos; scroll; content_pretty; go_back; content; todo_state } as state) =
  let size_x, size_y = Common.Term.size t in
  let x, y = pos in
  let img =
    let elements =
      Array.mapi
        (fun i el -> Compontent.current_line size_x y scroll el i)
        (* TODO: Fix this ugly slow conversion *)
        (Array.to_seq content_pretty |> Seq.drop scroll |> Array.of_seq)
    in
    let open I in
    Array.fold_left (fun sum el -> el </> sum) I.empty elements
  in
  Common.Term.image t img;
  let content_length = Array.length content_pretty in
  let scroll_up () =
    let scroll = if y - scroll = 0 then max (scroll - 1) 0 else scroll in
    render t @@ { state with pos = x, max (y - 1) 0; scroll }
  in
  let scroll_down () =
    let scroll = if y - scroll >= size_y - 1 then scroll + 1 else scroll in
    render t @@ { state with pos = x, min (y + 1) (content_length - 1); scroll }
  in
  match Common.Term.event t with
  | `End | `Key (`Escape, []) | `Key (`ASCII 'q', []) | `Key (`ASCII 'C', [ `Ctrl ]) -> ()
  | `Mouse (`Press (`Scroll s), _, _) ->
    (match s with
     | `Down -> scroll_down ()
     | `Up -> scroll_up ())
  | `Resize _ -> render t state
  | `Key (`ASCII 's', []) -> go_back ()
  | `Mouse ((`Press _ | `Drag), (_, y), _) ->
    render t { state with pos = 0, min y content_length }
  | `Key (`ASCII 'j', []) | `Key (`ASCII 'N', [ `Ctrl ]) -> scroll_down ()
  | `Key (`ASCII 'k', []) | `Key (`ASCII 'P', [ `Ctrl ]) -> scroll_up ()
  | `Key (`ASCII '?', []) -> Help_screen.render t { go_back = (fun () -> render t state) }
  | `Key (`ASCII '2', []) ->
    (match todo_state with
     | Some todo_state ->
       Todos.render
         t
         { todo_state with
           goto_headlines =
             (fun new_state -> render t { state with todo_state = Some new_state })
         }
     | None ->
       let todo_content = Grep.get_todos () |> Grep.parse_todo_string in
       let todo_pretty = Grep.pretty_format_todo todo_content in
       let todo_state =
         Todos.init
           ~goto_headlines:(fun new_state ->
             render t { state with todo_state = Some new_state })
           ~content:(todo_content |> Array.of_list)
           ~content_pretty:(todo_pretty |> Array.of_list)
       in
       Todos.render t todo_state)
  | `Key (`ASCII 'e', []) | `Key (`Enter, []) ->
    (* Editor might be set with extra args, in that case we need to separate these *)
    let[@warning "-8"] (editor :: args) =
      String.split_on_char ' ' (Sys.getenv "EDITOR")
    in
    let selected_file, line_number, _, file_number_offset = Array.get content y in
    let full_path_file = Grep.execution_directory ^ "/" ^ selected_file in
    (* Because each file title consists of two lines, we need to account for the offset
       it adds by removing the file_number *)
    let line_number_arg = "+" ^ Int.to_string (line_number - file_number_offset) in
    let full_args =
      Array.append (Array.of_list args) [| line_number_arg; full_path_file |]
    in
    Common.Term.cursor t (Some (0, 0));
    let _ =
      Unix.create_process_env
        editor
        full_args
        (Unix.environment ())
        Unix.stdin
        Unix.stdout
        Unix.stderr
    in
    let rec run_editor () =
      match Unix.wait () with
      | _, _ ->
        Common.Term.cursor t None;
        render t state
      (* Capture resizing events *)
      | exception Unix.Unix_error (Unix.EINTR, _, _) -> run_editor ()
      | exception Unix.Unix_error (_, _, _) -> failwith "ERROR"
    in
    run_editor ()
  | `Key (`Arrow d, _) ->
    (match d with
     | `Up -> scroll_up ()
     | `Down -> scroll_down ()
     | _ -> render t state)
  | _ -> render t state