update token when http status_code ne 200

copilot.go

@@ -77,6 +77,12 @@ func Init() {
 // token return the access token for github copilot
 type copiloter interface {
 	token() (string, error)
+	refresh() (string, error)
+}
+
+func (c *copilot) refresh() (string, error) {
+	caches.Delete(c.GithubCom.OauthToken)
+	return c.token()
 }
 
 func (c *copilot) token() (string, error) {

proxy.go

@@ -7,6 +7,7 @@ import (
 	"fmt"
 	"io"
 	"net/http"
+	"os"
 	"strings"
 	"time"
 
@@ -34,23 +35,27 @@ type Proxy struct {
 }
 
 func (p *Proxy) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	write := io.MultiWriter(w, os.Stdout)
 	req, err := http.NewRequest(r.Method, r.URL.String(), r.Body)
 	if err != nil {
 		w.WriteHeader(http.StatusInternalServerError)
-		fmt.Fprintf(w, "http request error: %v", err)
+		fmt.Fprintf(write, "http request error: %v\n", err)
 	}
 	if p.Director != nil {
 		p.Director(req)
 	}
 	resp, err := http.DefaultClient.Do(req)
 	if err != nil {
 		w.WriteHeader(http.StatusInternalServerError)
-		fmt.Fprintf(w, "http client error: %v", err)
+		fmt.Fprintf(write, "http client error: %v\n", err)
 		return
 	}
+	defer resp.Body.Close()
 	if resp.StatusCode != http.StatusOK {
 		w.WriteHeader(resp.StatusCode)
-		fmt.Fprintf(w, "http status error: %d", resp.StatusCode)
+		body, _ := io.ReadAll(resp.Body)
+		fmt.Fprintf(write, "http status error: %d\nbody: %s\n", resp.StatusCode, string(body))
+		go getCopilot().refresh()
 		return
 	}
 	if p.stream {
@@ -133,6 +138,9 @@ func handleProxy(w http.ResponseWriter, r *http.Request) {
 		req.URL.Path = strings.ReplaceAll(req.URL.Path, "/v1/", "/")
 
 		accToken, err := getCopilot().token()
+		if err != nil {
+			accToken, _ = getCopilot().refresh()
+		}
 		if accToken == "" {
 			w.WriteHeader(http.StatusInternalServerError)
 			fmt.Fprintf(w, "get acc token error: %v", err)