Initial QSfera import

2026-06-07 10:20:04 +03:00
commit 2315f25754
16485 changed files with 4826827 additions and 0 deletions
@@ -0,0 +1,114 @@
+package assets
+
+import (
+	"bytes"
+	"io"
+	"io/fs"
+	"mime"
+	"net/http"
+	"path"
+	"path/filepath"
+
+	"golang.org/x/net/html"
+)
+
+type fileServer struct {
+	fsys http.FileSystem
+}
+
+const (
+	webTitle        = "КуСфера"
+	legacyUpdateURL = "https://update.opencloud.eu/"
+	qsferaUpdateURL = "https://update.qsfera.eu/"
+)
+
+// FileServer defines the http handler for the embedded files
+func FileServer(fsys fs.FS) http.Handler {
+	return &fileServer{http.FS(fsys)}
+}
+
+func (f *fileServer) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	uPath := path.Clean(path.Join("/", r.URL.Path))
+	r.URL.Path = uPath
+
+	tryIndex := func() {
+		r.URL.Path = "/index.html"
+
+		// not every fs contains a file named index.html,
+		// therefore, we need to check if the file exists and stop the recursion if it doesn't
+		file, err := f.fsys.Open(r.URL.Path)
+		if err != nil {
+			http.NotFound(w, r)
+			return
+		}
+		defer file.Close()
+
+		f.ServeHTTP(w, r)
+	}
+
+	asset, err := f.fsys.Open(uPath)
+	if err != nil {
+		tryIndex()
+		return
+	}
+	defer asset.Close()
+
+	s, _ := asset.Stat()
+	if s.IsDir() {
+		tryIndex()
+		return
+	}
+
+	w.Header().Set("Content-Type", mime.TypeByExtension(filepath.Ext(s.Name())))
+
+	buf := new(bytes.Buffer)
+
+	switch s.Name() {
+	case "index.html", "oidc-callback.html", "oidc-silent-redirect.html":
+		w.Header().Del("Expires")
+		w.Header().Set("Cache-Control", "no-cache")
+		_ = withBase(buf, asset, "/")
+	default:
+		_, _ = buf.ReadFrom(asset)
+	}
+
+	body := bytes.ReplaceAll(buf.Bytes(), []byte(legacyUpdateURL), []byte(qsferaUpdateURL))
+	_, _ = w.Write(body)
+}
+
+func withBase(w io.Writer, r io.Reader, base string) error {
+	doc, _ := html.Parse(r)
+	var parse func(*html.Node)
+	parse = func(n *html.Node) {
+		if n.Type == html.ElementNode && n.Data == "head" {
+			n.InsertBefore(&html.Node{
+				Type: html.ElementNode,
+				Data: "base",
+				Attr: []html.Attribute{
+					{
+						Key: "href",
+						Val: base,
+					},
+				},
+			}, n.FirstChild)
+		}
+
+		if n.Type == html.ElementNode && n.Data == "title" {
+			if n.FirstChild == nil {
+				n.AppendChild(&html.Node{
+					Type: html.TextNode,
+					Data: webTitle,
+				})
+			} else {
+				n.FirstChild.Data = webTitle
+			}
+		}
+
+		for c := n.FirstChild; c != nil; c = c.NextSibling {
+			parse(c)
+		}
+	}
+	parse(doc)
+
+	return html.Render(w, doc)
+}
@@ -0,0 +1,121 @@
+package assets
+
+import (
+	"fmt"
+	"io"
+	"io/fs"
+	"net/http"
+	"net/http/httptest"
+	"testing"
+	"testing/fstest"
+
+	"github.com/onsi/gomega"
+)
+
+func TestFileServer(t *testing.T) {
+	g := gomega.NewWithT(t)
+	recorderStatus := func(s int) string {
+		return fmt.Sprintf("%03d %s", s, http.StatusText(s))
+	}
+
+	{
+		s := FileServer(fstest.MapFS{})
+		w := httptest.NewRecorder()
+		req := httptest.NewRequest("GET", "/foo", nil)
+		//defer req.Body.Close()
+		s.ServeHTTP(w, req)
+		res := w.Result()
+		defer res.Body.Close()
+
+		g.Expect(res.Status).To(gomega.Equal(recorderStatus(http.StatusNotFound)))
+	}
+
+	for _, tt := range []struct {
+		name     string
+		url      string
+		fs       fstest.MapFS
+		expected string
+	}{
+		{
+			name: "not found fallback",
+			url:  "/index.txt",
+			fs: fstest.MapFS{
+				"index.html": &fstest.MapFile{
+					Data: []byte("index file content"),
+				},
+			},
+			expected: `<html><head><base href="/"/></head><body>index file content</body></html>`,
+		},
+		{
+			name: "directory fallback",
+			url:  "/some-folder",
+			fs: fstest.MapFS{
+				"some-folder": &fstest.MapFile{
+					Mode: fs.ModeDir,
+				},
+				"index.html": &fstest.MapFile{
+					Data: []byte("index file content"),
+				},
+			},
+			expected: `<html><head><base href="/"/></head><body>index file content</body></html>`,
+		},
+		{
+			name: "index.html",
+			url:  "/index.html",
+			fs: fstest.MapFS{
+				"index.html": &fstest.MapFile{
+					Data: []byte("index file content"),
+				},
+			},
+			expected: `<html><head><base href="/"/></head><body>index file content</body></html>`,
+		},
+		{
+			name: "oidc-callback.html",
+			url:  "/oidc-callback.html",
+			fs: fstest.MapFS{
+				"index.html": &fstest.MapFile{
+					Data: []byte("oidc-callback file content"),
+				},
+			},
+			expected: `<html><head><base href="/"/></head><body>oidc-callback file content</body></html>`,
+		},
+		{
+			name: "oidc-silent-redirect.html",
+			url:  "/oidc-silent-redirect.html",
+			fs: fstest.MapFS{
+				"index.html": &fstest.MapFile{
+					Data: []byte("oidc-silent-redirect file content"),
+				},
+			},
+			expected: `<html><head><base href="/"/></head><body>oidc-silent-redirect file content</body></html>`,
+		},
+		{
+			name: "some-file.txt",
+			url:  "/some-file.txt",
+			fs: fstest.MapFS{
+				"some-file.txt": &fstest.MapFile{
+					Data: []byte("some file content"),
+				},
+			},
+			expected: "some file content",
+		},
+	} {
+		t.Run(tt.name, func(t *testing.T) {
+			t.Parallel()
+
+			w := httptest.NewRecorder()
+			req := httptest.NewRequest("GET", tt.url, nil)
+			FileServer(tt.fs).ServeHTTP(w, req)
+			res := w.Result()
+			defer res.Body.Close()
+
+			g.Expect(res.Status).To(gomega.Equal(recorderStatus(http.StatusOK)))
+
+			data, err := io.ReadAll(res.Body)
+			g.Expect(err).ToNot(gomega.HaveOccurred())
+			g.Expect(string(data)).To(gomega.Equal(tt.expected))
+
+		})
+	}
+
+}