gptel-gh: Default to saving github-token in a file

Author: marcus-lundgren

gptel-gh.el

@@ -182,14 +182,46 @@
                          (:constructor gptel--make-gh))
   sessionid machineid)
 
+(defcustom gptel-gh-github-token-file (expand-file-name ".cache/copilot-chat/github-token"
+                                                        user-emacs-directory)
+  "File where the GitHub token is stored."
+  :type 'string
+  :group 'gptel)
+
+(defun gptel--gh-restore-from-file ()
+  "Restore saved object from gptel-gh-github-token-file."
+  (when (file-exists-p gptel-gh-github-token-file)
+    ;; We set the coding system to `utf-8-auto-dos' when reading so that
+    ;; files with CR EOL can still be read properly
+    (let ((coding-system-for-read 'utf-8-auto-dos))
+      (with-temp-buffer
+        (set-buffer-multibyte nil)
+        (insert-file-contents-literally gptel-gh-github-token-file)
+        (goto-char (point-min))
+        (read (current-buffer))))))
+
+(defun gptel--gh-save-to-file (obj)
+   "Save OBJ to gptel--gh-save-to-file."
+  (let ((print-length nil)
+        (print-level nil)
+        (coding-system-for-write 'utf-8-unix))
+    (make-directory (file-name-directory gptel-gh-github-token-file) t)
+    (write-region (prin1-to-string obj) nil gptel-gh-github-token-file nil :silent)
+    obj))
+
+(defvar gptel-gh-github-token-load 'gptel--gh-restore-from-file
+  "Function to load the current github token.")
+
+(defvar gptel-gh-github-token-save 'gptel--gh-save-to-file
+  "Function to save the new github token.")
+
 (defconst gptel--gh-auth-common-headers
   `(("editor-plugin-version" . "gptel/*")
     ("editor-version" . ,(concat "emacs/" emacs-version))))
 
 (defconst gptel--gh-client-id "Iv1.b507a08c87ecfe98")
 
 (defvar gptel--gh-token nil)
-(defvar gptel--gh-github-token nil)
 
 ;; https://en.wikipedia.org/wiki/Universally_unique_identifier#Version_4_(random)
 (defun gptel--gh-uuid ()
@@ -210,10 +242,6 @@
       (setq hex (nconc hex (list (aref hex-chars (random 16))))))
     (apply #'string hex)))
 
-(defun gptel--gh-github-token-save (token)
-  "Saves the new github token for use."
-  (setq gptel--gh-github-token token))
-
 (defun gptel-gh-login ()
   "Login to GitHub Copilot API.
 
@@ -244,13 +272,11 @@ If your browser does not open automatically, browse to %s."
                   :device_code ,device_code
                   :grant_type "urn:ietf:params:oauth:grant-type:device_code"))
        :access_token)
-      (gptel--gh-github-token-save))
-    (if (and gptel--gh-github-token
-             (not (string-empty-p gptel--gh-github-token)))
-        (progn
-          (message "New GitHub token: %s" (prin1-to-string gptel--gh-github-token))
-          (message "Successfully logged in to GitHub Copilot"))
-      (user-error "Error: You might not have access to GitHub Copilot Chat!"))))
+      (funcall gptel-gh-github-token-save))
+    (let ((github-token (funcall gptel-gh-github-token-load)))
+      (if (and github-token (not (string-empty-p github-token)))
+          (message "Successfully logged in to GitHub Copilot")
+        (user-error "Error: You might not have access to GitHub Copilot Chat!")))))
 
 (defun gptel--gh-renew-token ()
   "Renew session token."
@@ -259,7 +285,7 @@ If your browser does not open automatically, browse to %s."
              "https://api.github.com/copilot_internal/v2/token"
            :method 'get
            :headers `(("authorization"
-                       . ,(format "token %s" gptel--gh-github-token))
+                       . ,(format "token %s" (funcall gptel-gh-github-token-load)))
                       ,@gptel--gh-auth-common-headers))))
     (if (not (plist-get token :token))
         (user-error "Error: You might not have access to GitHub Copilot Chat!")
@@ -270,7 +296,7 @@ If your browser does not open automatically, browse to %s."
 
 We first need github authorization (github token).
 Then we need a session token."
-  (unless gptel--gh-github-token
+  (unless (funcall gptel-gh-github-token-load)
     (gptel-gh-login))
 
   (pcase-let (((map :token :expires_at)
@@ -298,8 +324,7 @@ Then we need a session token."
           (protocol "https")
           (endpoint "/chat/completions")
           (stream t)
-          (models gptel--gh-models)
-          (github-token-init nil))
+          (models gptel--gh-models))
   "Register a Github Copilot chat backend for gptel with NAME.
 
 Keyword arguments:
@@ -351,9 +376,6 @@ parameters (as plist keys) and values supported by the API.  Use
 these to set parameters that gptel does not provide user options
 for."
   (declare (indent 1))
-  (if (and (not gptel--gh-github-token)
-           github-token-init)
-      (gptel--gh-github-token-save (funcall github-token-init)))
   (let ((backend (gptel--make-gh
                   :name name
                   :host host