;;; EWW

;; TODO: Fix `eww-forward-url' infinite forwarding.

;; TODO: Extend the history / bookmarks view to display tags, mark and search engine.
;; With colors: [mark] title url (tags)
;; Mark is red if no search engine, green otherwise.
;; Tags have their own face.

;; TODO: Make something useful with the tags.  Helm function?  Could chain two
;; functions: tag selection then filtered bookmark selection, then tag selection
;; again, etc.
;; Alternative: use special syntax in prompt, like find-files does.  This does
;; not allow for listing tags though.

(require 'helm-eww nil t)

(setq eww-bookmarks-directory "~/personal/bookmarks")
;; (add-to-list 'auto-mode-alist '("eww-bookmarks$" . emacs-lisp-mode))

(defvar ambrevar/eww-bookmarks-file (expand-file-name "eww-bookmarks.gpg"
                                                      eww-bookmarks-directory))

(defun ambrevar/eww-copy-page-title ()
  "Copy the URL of the current page into the kill ring."
  (interactive)
  (message "%s" (plist-get eww-data :title))
  (kill-new (plist-get eww-data :title)))

(defun ambrevar/eww-next-url (&optional backward)
  "Like `eww-next-url' but if no next URL is found, go to next URL numerically.
The URL index is the last number after the last '/'."
  (interactive)
  (condition-case nil
      (if backward
          (eww-previous-url)
        (eww-next-url))
    (user-error
     (when (eq major-mode 'eww-mode)
       (require 'rx)
       (let* ((url (plist-get eww-data :url))
              (re (rx (group (one-or-more digit))
                      (zero-or-more (not (any "/")))
                      line-end)))
         (if (and (string-match re url)
                  (or (not backward)
                      (> (string-to-number (match-string 1 url)) 0)))
             (eww
              (replace-regexp-in-string
               re
               (format (format "%%0.%dd" (length (match-string 1 url))) ; In case matched number is zero-padded.
                       (funcall (if backward '1- '1+) (string-to-number (match-string 1 url))))
               url nil nil 1))
           (message "No index in URL.")))))))

(defun ambrevar/eww-previous-url ()
  "Like `eww-previous-url' but if no next URL is found, go to next URL numerically.
The URL index is the last number after the last '/'."
  (interactive)
  (ambrevar/eww-next-url 'backward))

(defun ambrevar/eww-reload-all (&optional buffers)
  "Like `eww-reload' but for multiple EWW BUFFERS.
If BUFFERS is not specified, then reload all buffers."
  (interactive)
  (dolist (b (or buffers (buffer-list)))
    (with-current-buffer b
      (when (derived-mode-p 'eww-mode)
        (eww-reload)))))

(when (require 'patch-helm nil 'noerror)
  (helm-defswitcher
   "eww"
   (lambda (b)
     (with-current-buffer b
       (derived-mode-p 'eww-mode)))
   (lambda (&optional _)
     (interactive)
     (if (thing-at-point-url-at-point)
         ;; This only works fine if no EWW buffer is up.
         (call-interactively 'eww)
       (helm-eww)))
   helm-eww))

(defun ambrevar/eww (url)
  "Fetch URL and render the page.
If the input doesn't look like an URL or a domain name, the
word(s) will be searched for via `eww-search-prefix'."
  (interactive
   (let* ((uris (eww-suggested-uris))
          (prompt (concat "Enter URL or keywords: "))) ; PATCH
     (list (read-string prompt (car uris) nil uris)))) ; PATCH
  (setq url (eww--dwim-expand-url url))
  (pop-to-buffer-same-window
   (if (eq major-mode 'eww-mode)
       (current-buffer)
     (get-buffer-create "*eww*")))
  (eww-setup-buffer)
  ;; Check whether the domain only uses "Highly Restricted" Unicode
  ;; IDNA characters.  If not, transform to punycode to indicate that
  ;; there may be funny business going on.
  (let ((parsed (url-generic-parse-url url)))
    (when (url-host parsed)
      (unless (puny-highly-restrictive-domain-p (url-host parsed))
        (setf (url-host parsed) (puny-encode-domain (url-host parsed)))
        (setq url (url-recreate-url parsed)))))
  (plist-put eww-data :url url)
  (plist-put eww-data :title "")
  (eww-update-header-line-format)
  (let ((inhibit-read-only t))
    (insert (format "Loading %s..." url))
    (goto-char (point-min)))
  (url-retrieve url 'eww-render
                (list url nil (current-buffer))))
(advice-add 'eww :override 'ambrevar/eww)

(defun ambrevar/eww-browse-with-external-browser (&optional url)
  "Browse the current URL with an external browser.
The browser to used is specified by the `shr-external-browser' variable."
  (interactive)
  (funcall shr-external-browser (or url
                                    (car (eww-suggested-uris))
                                    (plist-get eww-data :url))))
(advice-add 'eww-browse-with-external-browser
            :override
            'ambrevar/eww-browse-with-external-browser)

(defun ambrevar/eww-open-in-new-buffer (url)
  "Fetch URL and render the page.
If the input doesn't look like an URL or a domain name, the
word(s) will be searched for via `eww-search-prefix'."
  (interactive
   (let* ((uris (eww-suggested-uris))
          (prompt (concat "Open URL or keywords in new buffer: ")))
     (list (read-string prompt (car uris) nil uris))))
  (setq url (eww--dwim-expand-url url))
  (pop-to-buffer-same-window
   (if (eq major-mode 'eww-mode)
       (clone-buffer)
     (generate-new-buffer "*eww*")))
  (unless (equal url (eww-current-url))
    (eww-mode)
    (eww (if (consp url) (car url) url))))
(advice-add 'eww-open-in-new-buffer :override 'ambrevar/eww-open-in-new-buffer)

(defun ambrevar/eww-name-buffer-with-title ()
  "Include the page title in current EWW buffer name."
  (interactive)
  (when (derived-mode-p 'eww-mode)
    (rename-buffer (format "*eww: %s*" (plist-get eww-data :title)) t)))

(defun ambrevar/eww-update-header-line-format ()
  (setq header-line-format
        (and eww-header-line-format
             (let ((title (plist-get eww-data :title))
                   (peer (plist-get eww-data :peer)))
               (when (zerop (length title))
                 (setq title "[untitled]"))
               ;; This connection has is https.
               (when peer
                 (setq title
                       (propertize title 'face
                                   (if (plist-get peer :warnings)
                                       'eww-invalid-certificate
                                     'eww-valid-certificate))))
               (replace-regexp-in-string
                "%" "%%"
                (format-spec
                 eww-header-line-format
                 `((?u . ,(or (plist-get eww-data :url) ""))
                   (?t . ,title)))))))
  ;; PATCH
  (ambrevar/eww-name-buffer-with-title))
(advice-add 'eww-update-header-line-format :override 'ambrevar/eww-update-header-line-format)

;; TODO: Fix quickmarks bindings.  Or maybe just display the quickmarks buffer
;; and start `eww', which follows the quickmarks when first word is the mark.
;; TODO: Merge qutebrowser quickmarks.
;; TODO: Add bookmark editing functions such as edit title, tags, quickmark,
;; search-engine.  Use eww-buffers and Helm.
(defun ambrevar/eww-add-bookmark (&optional url title)
  "Bookmark the current page.
With prefix argument, prompt for bookmark title."
  (setq url (or url (plist-get eww-data :url)))
  (setq title (or title (plist-get eww-data :title)))
  (interactive)
  (eww-read-bookmarks)
  (let (tag-list existing-bookmark)
    (dolist (bookmark eww-bookmarks)
      (when (equal
             ;; PATCH: Ignore protocol when sorting.
             ;; TODO: Include "sort-bookmarks": Warn for unique tags, warn for same URL up to paragraph.  Make this customizable.
             ;; TODO: Smarter: Store bookmarks as an alist/hash-table indexed by domain.tld.
             (replace-regexp-in-string "^https?" "" url)
             (replace-regexp-in-string "^https?" "" (plist-get bookmark :url)))
        (if existing-bookmark
            (user-error "Duplicate bookmarks: %s, %s" existing-bookmark bookmark)
          (setq existing-bookmark bookmark)))
      (setq tag-list (append tag-list (plist-get bookmark :tags))))
    (cl-delete-duplicates tag-list)
    (let ((tags (completing-read-multiple (format "%s for bookmark (comma separated): "
                                                  (if existing-bookmark "Update tags" "Tags"))
                                          tag-list
                                          nil nil nil nil
                                          (and existing-bookmark
                                               (mapconcat 'identity
                                                          (plist-get existing-bookmark :tags) ","))))
          (existing-title (or (and existing-bookmark (plist-get existing-bookmark :title))
                              ""))
          (sanitize-title (lambda (title)
                            (setq title (string-trim title))
                            (setq title (replace-regexp-in-string "[\n\t\r]" " " title)))))
      (setq title (funcall sanitize-title title))
      (setq existing-title (funcall sanitize-title existing-title))
      (setq title (or (and (not current-prefix-arg)
                           (not (string-empty-p (if existing-bookmark existing-title title)))
                           (if existing-bookmark existing-title title))
                      (completing-read "Title: "
                                       (if existing-bookmark
                                           (list title existing-title)
                                         (list title))
                                       nil nil nil nil
                                       (if existing-bookmark existing-title title))))
      (setq title (funcall sanitize-title title))
      (let ((new-bookmark `(:url ,url
                                 :title ,title
                                 :time ,(or (and existing-bookmark (plist-get existing-bookmark :time))
                                            (current-time-string))
                                 ,@(if tags (list :tags tags)))))
        (if existing-bookmark
            (cl-nsubstitute new-bookmark existing-bookmark eww-bookmarks :count 1)
          (push new-bookmark eww-bookmarks)))
      (eww-write-bookmarks)
      (message "Bookmarked %s (%s)" url title))))
(advice-add 'eww-add-bookmark :override 'ambrevar/eww-add-bookmark)

(defvar ambrevar/eww-bookmark-timestamp nil)
(defun ambrevar/eww-read-bookmarks ()
  (let* ((file ambrevar/eww-bookmarks-file)
         (mtime (file-attribute-modification-time (file-attributes file))))
    (when (or (null ambrevar/eww-bookmark-timestamp)
              (null eww-bookmarks)
              (time-less-p ambrevar/eww-bookmark-timestamp
                           mtime))
      (setq ambrevar/eww-bookmark-timestamp mtime)
      (setq eww-bookmarks
            (unless (zerop (or (file-attribute-size (file-attributes file)) 0))
              (with-temp-buffer
                (insert-file-contents file)
                (read (current-buffer))))))))
(advice-add 'eww-read-bookmarks :override 'ambrevar/eww-read-bookmarks)

(defun ambrevar/eww-write-bookmarks ()
  (require 'rx)
  ;; PATCH
  (setq eww-bookmarks
        (sort eww-bookmarks
              (lambda (a b) (string<
                             ;; Ignore protocol when sorting.
                             (replace-regexp-in-string "^[a-zA-Z]+://" "" (plist-get a :url))
                             (replace-regexp-in-string "^[a-zA-Z]+://" "" (plist-get b :url))))))
  ;; PATCH: Remove newline for tags.
  (let* ((visiting (find-buffer-visiting ambrevar/eww-bookmarks-file))
         (buffer (or visiting (find-file-noselect ambrevar/eww-bookmarks-file))))
    (with-current-buffer buffer
      (erase-buffer)
      (insert
       (replace-regexp-in-string
        (rx ":tags" ?\n (1+ space)) ":tags "
        (with-temp-buffer
          (insert ";; Auto-generated file; don't edit -*- mode: emacs-lisp; -*-\n")
          (pp eww-bookmarks (current-buffer))
          (buffer-string))))
      (save-buffer))
    (unless visiting (kill-buffer buffer))
    (setq ambrevar/eww-bookmark-timestamp
          (file-attribute-modification-time ambrevar/eww-bookmark-timestamp))))
(advice-add 'eww-write-bookmarks :override 'ambrevar/eww-write-bookmarks)

(defvar ambrevar/eww-quickmark-prefix ""
  "Prefix to load a quickmark.")

(defun ambrevar/eww-bookmark-prepare ()
  ;; PATCH: Don't load if already loaded.  This allows for overrides (e.g. quickmarks).
  (unless eww-bookmarks
    (eww-read-bookmarks))
  (unless eww-bookmarks
    (user-error "No bookmarks are defined"))
  (set-buffer (get-buffer-create "*eww bookmarks*"))
  (eww-bookmark-mode)
  (let* ((width (/ (window-width) 2))
         (format (format "%%-%ds %%s" width))
         (inhibit-read-only t)
         start title)
    (erase-buffer)
    (setq header-line-format (concat " " (format format "Title" "URL")))
    (dolist (bookmark eww-bookmarks)
      (setq start (point)
            title (plist-get bookmark :title))
      ;; PATCH: Show quickmark.
      (when (plist-get bookmark :mark)
        (setq title (format "[%s] %s" (plist-get bookmark :mark) title)))
      (when (> (length title) width)
        (setq title (truncate-string-to-width title width)))
      (insert (format format title
                      (concat (plist-get bookmark :url)
                              (when (plist-get bookmark :tags)
                                (concat " ("
                                        (mapconcat 'identity (plist-get bookmark :tags) ",")
                                        ")"))))
              "\n")
      (put-text-property start (1+ start) 'eww-bookmark bookmark)
      ;; PATCH: Bind keys
      (when (plist-get bookmark :mark)
        (define-key eww-bookmark-mode-map
          (kbd (concat ambrevar/eww-quickmark-prefix (plist-get bookmark :mark)))
          (lambda (&optional new-window)
            (interactive "P")
            (if new-window
                (ambrevar/eww-open-in-new-buffer (plist-get bookmark :url))
              (eww (plist-get bookmark :url)))))))
    (goto-char (point-min))))
(advice-add 'eww-bookmark-prepare :override 'ambrevar/eww-bookmark-prepare)

(defun ambrevar/eww-quickmarks (&optional new-window)
  "Display quickmarks."
  (interactive "P")
  (let ((eww-bookmarks (seq-filter (lambda (b) (plist-get b :mark)) eww-bookmarks)))
    (eww-list-bookmarks)))

(defun ambrevar/eww-bookmarks-list-by-tags (&optional arg)
  "Return bookmarks matching one of the specified tags.
With prefix argument or ARG, bookmarks much match all tags."
  (let ((tag-list (delq nil (mapcar (lambda (b) (plist-get b :tags)) eww-bookmarks))))
    (seq-uniq (mapcar 'append tag-list))
    (let ((tags (completing-read-multiple "Tags for bookmark (comma separated): " tag-list)))
      (seq-filter (lambda (b)
                    (if arg
                        (null (seq-difference tags (plist-get b :tags)))
                      (seq-intersection tags (plist-get b :tags))))
                  eww-bookmarks))))

(defun ambrevar/eww-bookmarks-by-tags (&optional arg)
  "Display bookmarks matching one of the specified tags.
With prefix argument or ARG, bookmarks much match all tags."
  (interactive "P")
  (let ((eww-bookmarks (ambrevar/eww-bookmarks-list-by-tags arg)))
    (eww-list-bookmarks)))

(defun ambrevar/eww--dwim-expand-url (url)
  (setq url (string-trim url))
  (cond ((string-match-p "\\`file:/" url))
        ;; Don't mangle file: URLs at all.
        ((string-match-p "\\`ftp://" url)
         (user-error "FTP is not supported"))
        (t
         ;; Anything that starts with something that vaguely looks
         ;; like a protocol designator is interpreted as a full URL.
         (if (or (string-match "\\`[A-Za-z]+:" url)
                 ;; Also try to match "naked" URLs like
                 ;; en.wikipedia.org/wiki/Free software
                 (string-match "\\`[A-Za-z_]+\\.[A-Za-z._]+/" url)
                 (and (= (length (split-string url)) 1)
                      (or (and (not (string-match-p "\\`[\"'].*[\"']\\'" url))
                               (> (length (split-string url "[.:]")) 1))
                          (string-match eww-local-regex url))))
             (progn
               (unless (string-match-p "\\`[a-zA-Z][-a-zA-Z0-9+.]*://" url)
                 (setq url (concat "http://" url)))
               ;; Some sites do not redirect final /
               (when (string= (url-filename (url-generic-parse-url url)) "")
                 (setq url (concat url "/"))))
           ;; PATCH: Add support for search engines and quickmarks.
           (string-match (rx (group (1+ (not space)))
                             (0+ space)
                             (? (group (0+ any))))
                         url)
           (let* ((first-word (match-string 1 url))
                  (rest-url (match-string 2 url))
                  (marks (make-hash-table :test 'equal))
                  (engines (make-hash-table :test 'equal)))
             (dolist (b eww-bookmarks)
               (let ((mark (plist-get b :mark))
                     engine)
                 (when mark
                   (puthash mark (plist-get b :url) marks)
                   (setq engine (plist-get b :search))
                   (when engine
                     (puthash mark (concat (let ((case-fold-search t))
                                             (unless (string-match "^https?://" engine)
                                               (plist-get b :url)))
                                           engine)
                              engines)))))
             (cond
              ((and (gethash first-word engines)
                    (not (string= rest-url "")) )
               (setq url (format (gethash first-word engines) ; Engines must have exactly one "%s".
                                 (mapconcat
                                  #'url-hexify-string (split-string rest-url) "+"))))
              ((and (gethash first-word marks)
                    (string= rest-url ""))
               (setq url (gethash first-word marks)))
              (t (setq url (concat eww-search-prefix
                                   (mapconcat
                                    #'url-hexify-string (split-string url) "+")))))))))
  url)
(advice-add 'eww--dwim-expand-url :override 'ambrevar/eww--dwim-expand-url)

(provide 'init-eww)