aboutsummaryrefslogtreecommitdiffstats
path: root/elisp/cmk.el
blob: 6e2b7e08ee413deec99e031e052930ab75cb22b3 (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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
;;; cmk.el --- CMK livestatus -*- lexical-binding: t; -*-
;;
;; Copyright (C) 2022 Óscar Nájera
;;
;; Author: Óscar Nájera <hi@oscarnajera.com>
;; Maintainer: Óscar Nájera <hi@oscarnajera.com>
;; Created: July 08, 2022
;; Modified: July 08, 2022
;; Version: 0.0.1
;; Keywords: abbrev bib c calendar comm convenience data docs emulations extensions faces files frames games hardware help hypermedia i18n internal languages lisp local maint mail matching mouse multimedia news outlines processes terminals tex tools unix vc wp
;; Homepage: https://github.com/titan/cmk
;; Package-Requires: ((emacs "27.1"))
;;
;; This file is not part of GNU Emacs.
;;
;;; Commentary:
;;
;;  CMK livestatus
;;
;;; Code:
(require 'subr-x)
(require 'cl-lib)
(require 'dash)

(defconst cmk-lq-problems-filter
  "Filter: service_state = 0
Filter: service_has_been_checked = 1
And: 2
Negate:
Filter: service_has_been_checked = 1
Filter: service_scheduled_downtime_depth = 0
Filter: host_scheduled_downtime_depth = 0
And: 2
Filter: service_acknowledged = 0
Filter: host_state = 1
Filter: host_has_been_checked = 1
And: 2
Negate:
Filter: host_state = 2
Filter: host_has_been_checked = 1
And: 2
Negate:\n\n")

(defun cmk-state-coloring (state)
  "Font color for numeric STATE input as string."
  (cl-check-type state string)
  (pcase (string-to-number state)
    (0 (propertize "OK" 'face 'font-lock-string-face))
    (1 (propertize "WARN" 'face 'font-lock-warning-face))
    (2 (propertize "CRIT" 'face 'font-lock-keyword-face))
    (3 (propertize "UNKN" 'face 'font-lock-builtin-face))
    (_ state)))

(defun cmk-timestamp-to-date (timestamp &optional fmt-str)
  "TIMESTAMP to human readable date follownig FMT-STR.
Default is  \"%Y-%m-%d %H:%M\"."
  (thread-last
    (if (stringp timestamp)
        (string-to-number timestamp)
      timestamp)
    (seconds-to-time)
    (format-time-string (or fmt-str "%Y-%m-%d %H:%M"))))

(defun cmk-purge-bangs (str)
  "Remove cmk status output bangs from STR."
  (replace-regexp-in-string (rx "(" (+ "!") ")") "" str))

(defun cmk-oneshot (&optional keep-buffer)
  "One shot socket connection. non-nil KEEP-BUFFER after socket closes."
  (make-network-process
   :name "Checkmk"
   :remote "/tmp/ingridcmk.socket"
   :buffer "CMK"
   :sentinel (lambda (process event)
               (message "Process: %s had the event '%s'" process (string-trim event))
               (unless keep-buffer
                 (kill-buffer (process-buffer process))))))

(defun cmk-colums-from-spec (spec)
  "Extract the column names from SPEC and return a livestatus column command."
  (concat "Columns: "
          (mapconcat
           (lambda (column)
             (or (plist-get (cdddr column) :column) (car column)))
           spec
           " ")))

(defconst cmk-problem-col-spec
  '(("STATE" 5 t         :column "service_state" :parser cmk-state-coloring )
    ("Host" 9 t          :column "host_name")
    ("Service" 20 t      :column "service_description")
    ("Since" 16 nil      :column "service_last_state_change" :parser cmk-timestamp-to-date)
    ("Check output" 18 t :column "service_plugin_output" :parser cmk-purge-bangs)
    ))

(defun cmk-parse-rows (buffer column-spec)
  "Return tabulated entries from BUFFER according to COLUMN-SPEC."
  (with-current-buffer buffer
    (goto-char (point-min))
    (cl-loop until (eobp)
             collect
             (list (count-lines 1 (point))
                   (cl-map 'vector
                           (lambda (entry spec)
                             (let ((parser (or (plist-get (cdddr spec) :parser) #'identity)))
                               (funcall parser entry)))
                           (split-string (buffer-substring-no-properties (point) (line-end-position)) ";")
                           column-spec))
             do (forward-line))))

;; (process-send-string cmks "GET hosts\nColumns: address name\n\n")
;;   (process-send-string cmks
;;                        "GET log
;; Columns: host_name service_description log_type log_plugin_output log_state log_state_type log_time
;; Filter: log_time >= 1657270676
;; Filter: class = 1
;; Filter: class = 3
;; Filter: class = 8
;; Or: 3\n\n")

(defun cmk-livestatus-query (query)
  "One shot livestatus QUERY, return network process."
  (let ((cmks (cmk-oneshot 'keep)))
    (with-current-buffer (process-buffer cmks) (erase-buffer))
    (process-send-string cmks query)
    cmks))

(defvar-local cmk-livestatus-query ""
  "Livestatus query in buffer.")

(defun cmk-table--refresh ()
  "Query and reprint the Livestatus query result table."
  (let ((cmks (cmk-livestatus-query cmk-livestatus-query)))
      (accept-process-output cmks 0.1)
      (setq tabulated-list-entries (cmk-parse-rows (process-buffer cmks) cmk-problem-col-spec))
      (tabulated-list-print 'remember)))

(define-derived-mode cmk-livestatus-mode prog-mode "livestatus"
  "Major mode for viewing journalctl output."
  ;; code for syntax highlighting
  (setq font-lock-defaults `(((,(rx bol "GET" eow) . font-lock-keyword-face)
                              (,(rx word-start (or "hosts" "services")  word-end) . font-lock-constant-face)
                              (,(rx bol (or "Filter" "Columns" "Stats" "Limit") ":" whitespace) . font-lock-function-name-face)
                              (,(rx bol (or "And" "Or" "Negate") ":" whitespace) . font-lock-keyword-face)
                              (,(rx whitespace (or "~" "=" "=~" "~~" "<" ">" "<=" ">=") whitespace) . font-lock-keyword-face)))))

(defun cmk-edit-lq (buffer)
  "Open window to edit the livestatus query of current BUFFER."
  (interactive (list (current-buffer)))
  (let ((query cmk-livestatus-query)
        (edit-buffer (get-buffer-create "*LQ*")))
    (with-current-buffer edit-buffer
      (cmk-livestatus-mode)
      (insert query)
      (local-set-key "\C-c\C-c"
                     (lambda ()
                       (interactive)
                       (let ((new-query (-> (buffer-string) (string-trim) (concat "\n\n"))))
                         (kill-buffer)
                         (switch-to-buffer buffer)
                         (setq-local cmk-livestatus-query new-query)
                         (cmk-table--refresh)))))
    (switch-to-buffer edit-buffer)))

(defun cmk-problems ()
  (interactive)
  (with-current-buffer (get-buffer-create "*CMK Problems*")
    (tabulated-list-mode)
    (setq-local cmk-livestatus-query
                (concat (string-join  `("GET services"
                                        ,(cmk-colums-from-spec cmk-problem-col-spec)
                                        ,cmk-lq-problems-filter) "\n")
                        "\n\n"))
    (setq tabulated-list-format (vconcat cmk-problem-col-spec))
    (setq tabulated-list-sort-key '("STATE" . nil))

    (local-set-key "Q" #'cmk-edit-lq)
    (add-hook 'tabulated-list-revert-hook #'cmk-table--refresh nil t)
    (tabulated-list-init-header)
    (cmk-table--refresh)
    (switch-to-buffer (current-buffer))))

;;; Get a different csv
;; (let* ((cmks (cmk-oneshot t)))
;;   (with-current-buffer (process-buffer cmks) (erase-buffer))
;;   (process-send-string cmks "GET services
;; Columns: service_description host_name service_perf_data
;; Filter: service_description ~ CPU utilization
;; OutputFormat: csv
;; Separators: 10 9 44 124\n\n")
;;   (accept-process-output cmks 0.1)
;;   (display-buffer (process-buffer cmks)))

;;; Get some metric data
;; (let* ((now (time-convert nil 'integer))
;;        (start (- now 7200))
;;        (cmks (cmk-oneshot t)))
;;   (with-current-buffer (process-buffer cmks) (erase-buffer))
;;   (process-send-string cmks (format "GET services
;; Columns: host_name rrddata:1:user:%d:%d:1
;; Filter: service_description ~ CPU utilization
;; OutputFormat: json\n\n" start now))
;;   (accept-process-output cmks 0.1)
;;   (with-current-buffer (process-buffer cmks)
;;     (goto-char (point-min))
;;     (display-buffer (current-buffer))
;;     (json-parse-buffer)))


(provide 'cmk)
;;; cmk.el ends here