From 8b07f856a0ad37742500a0f0445297c04e3e0df8 Mon Sep 17 00:00:00 2001
From: Adam Sokolnicki <adam.sokolnicki@gmail.com>
Date: Fri, 15 Apr 2022 20:10:49 +0200
Subject: [PATCH] Add commands for disabling/enabling cops

Related to #38
---
 rubocop.el | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 50 insertions(+)

diff --git a/rubocop.el b/rubocop.el
index 6353abd..a370149 100644
--- a/rubocop.el
+++ b/rubocop.el
@@ -257,6 +257,56 @@ See also `rubocop-format-on-save' and `rubocop-autocorrect-on-save'."
   (unless (or (executable-find "rubocop") (rubocop-bundled-p))
     (error "RuboCop is not installed")))
 
+(defun rubocop--currently-disabled-cops ()
+  (save-excursion
+    (end-of-line)
+    (when (re-search-backward "# ?rubocop:disable \\([A-Z][a-z]+/[A-Za-z, /]+\\)")
+        (--map (s-trim it)
+               (split-string (match-string 1) ",")))))
+
+(defun rubocop--disable-cops (file-name)
+  (let* ((default-directory (or (rubocop-project-root 'no-error) default-directory))
+         (output (shell-command-to-string
+                  (rubocop-build-command rubocop-check-command (rubocop-local-file-name file-name))))
+         (line (line-number-at-pos))
+         (cops (--keep it
+                       (--map
+                        (if (string-match
+                             (format "^.+:%d:[0-9]+: [A-Z]: \\(?:\\[Correctable\\] \\)?\\([A-Z][a-z]+/[A-Za-z]+\\):" line)
+                             it)
+                            (match-string 1 it))
+                        (split-string output "[\n]")))))
+    (if cops
+        (save-excursion
+          (end-of-line)
+          (insert
+           (concat
+            " # rubocop:disable "
+            (string-join (-uniq cops) ", "))))
+      (error "No offences at current line"))))
+
+;;;###autoload
+(defun rubocop-disable-cops ()
+  "Disable cops on the current line."
+  (interactive)
+  (rubocop-ensure-installed)
+  (let ((file-name (buffer-file-name (current-buffer))))
+    (if file-name
+        (asok/rubocop--disable-cops file-name)
+      (error "Buffer is not visiting a file"))))
+
+;;;###autoload
+(defun asok/rubocop-enable-cops ()
+  "Remove the comments which disable the cops on the current line."
+  (interactive)
+  (save-excursion
+    (end-of-line)
+    (if (re-search-backward "[ ]*# ?rubocop:disable +\\(?:[A-Z][a-z]+/[A-Za-z, /]+\\)")
+        (progn
+          (backward-char)
+          (kill-line))
+      (error "No disabled cops at current line"))))
+
 ;;; Minor mode
 (defvar rubocop-mode-map
   (let ((map (make-sparse-keymap)))