Redirect / from browser to /ui

Author: vincent99

pkg/server/server.go

@@ -234,6 +234,21 @@ func (s *Server) Connect(session *melody.Session) {
 }
 
 func (s *Server) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
+	isUpgrade := strings.Contains(strings.ToLower(req.Header.Get("Connection")), "upgrade")
+	isBrowser := strings.Contains(strings.ToLower(req.Header.Get("User-Agent")), "mozilla")
+	isAjax := req.Header.Get("X-Requested-With") != ""
+
+	if req.URL.Path == "/" && isBrowser && !isAjax && !isUpgrade {
+		rw.Header().Set("Location", "/ui/")
+		rw.WriteHeader(302)
+		return
+	}
+
+	if req.URL.Path == "/favicon.ico" {
+		http.ServeFileFS(rw, req, static.UI, "/ui/favicon.ico")
+		return
+	}
+
 	if strings.HasPrefix(req.URL.Path, "/ui") {
 		path := req.URL.Path
 		if path == "/ui" || path == "/ui/" {
@@ -250,7 +265,7 @@ func (s *Server) ServeHTTP(rw http.ResponseWriter, req *http.Request) {
 	case http.MethodPost:
 		s.run(rw, req)
 	case http.MethodGet:
-		if req.URL.Path == "/" && strings.Contains(strings.ToLower(req.Header.Get("Connection")), "upgrade") {
+		if req.URL.Path == "/" && isUpgrade {
 			err := s.melody.HandleRequest(rw, req)
 			if err != nil {
 				http.Error(rw, err.Error(), http.StatusInternalServerError)

ui/Makefile

@@ -3,6 +3,7 @@ build: clean
 	yarn generate
 	rm -rf ../static/ui/_nuxt
 	cp -rp .output/public/* ../static/ui/
+	touch ../static/ui/_nuxt/_placeholder
 
 clean:
 	yarn clean

ui/components/left-nav.vue

@@ -4,7 +4,7 @@ const gptList = await useGpts().listAll()
 const runList = await useRuns().findAll()
 
 const gptLinks = computed(() => {
-  return (gptList.value || []).map(x => { return {
+  return (gptList || []).map(x => { return {
     label: x,
     icon: 'i-heroicons-wrench-screwdriver',
     to: `/gpt/` + x.split('/').map(y => encodeURIComponent(y)).join('/'),

ui/stores/gpts.ts

@@ -27,8 +27,8 @@ export const useGpts = defineStore('gpts', {
 
     async listAll() {
       const url = useRuntimeConfig().public.api
-      const { data } = (await useFetch(url) as any as {data: Ref<string[]>})
-      return data
+      const data = await $fetch(url, {headers: {'X-Requested-With': 'fetch'} }) as string[]
+      return reactive(data)
     }
   }
 })