web_router_test.go

package pgs

import (
	"fmt"
	"io"
	"log/slog"
	"net/http"
	"net/http/httptest"
	"strings"
	"testing"
	"time"

	"github.com/picosh/pgs/db/memory"
	"github.com/picosh/pgs/storage"
	sst "github.com/picosh/pobj/storage"
)

type ApiExample struct {
	name        string
	path        string
	reqHeaders  map[string]string
	want        string
	wantUrl     string
	status      int
	contentType string

	storage map[string]map[string]string
}

type PgsDb struct {
	*memory.MemoryDB
}

func NewPgsDb(logger *slog.Logger) *PgsDb {
	sb := memory.NewDBMemory(logger)
	sb.SetupTestData()
	_, err := sb.InsertProject(sb.Users[0].GetID(), "test", "test")
	if err != nil {
		panic(err)
	}
	return &PgsDb{
		MemoryDB: sb,
	}
}

func (p *PgsDb) mkpath(path string) string {
	return fmt.Sprintf("https://%s-test.pgs.test%s", p.Users[0].GetName(), path)
}

func TestApiBasic(t *testing.T) {
	logger := slog.Default()
	dbpool := NewPgsDb(logger)
	bucketName := GetAssetBucketName(dbpool.Users[0].GetID())

	tt := []*ApiExample{
		{
			name:        "basic",
			path:        "/",
			want:        "hello world!",
			status:      http.StatusOK,
			contentType: "text/html",

			storage: map[string]map[string]string{
				bucketName: {
					"test/index.html": "hello world!",
				},
			},
		},
		{
			name:        "direct-file",
			path:        "/test.html",
			want:        "hello world!",
			status:      http.StatusOK,
			contentType: "text/html",

			storage: map[string]map[string]string{
				bucketName: {
					"test/test.html": "hello world!",
				},
			},
		},
		{
			name:        "subdir-301-redirect",
			path:        "/subdir",
			want:        `<a href="/subdir/">Moved Permanently</a>.`,
			status:      http.StatusMovedPermanently,
			contentType: "text/html; charset=utf-8",

			storage: map[string]map[string]string{
				bucketName: {
					"test/subdir/index.html": "hello world!",
				},
			},
		},
		{
			name:        "redirects-file-301",
			path:        "/anything",
			want:        `<a href="/about.html">Moved Permanently</a>.`,
			status:      http.StatusMovedPermanently,
			contentType: "text/html; charset=utf-8",

			storage: map[string]map[string]string{
				bucketName: {
					"test/_redirects": "/anything /about.html 301",
					"test/about.html": "hello world!",
				},
			},
		},
		{
			name:        "subdir-direct",
			path:        "/subdir/index.html",
			want:        "hello world!",
			status:      http.StatusOK,
			contentType: "text/html",

			storage: map[string]map[string]string{
				bucketName: {
					"test/subdir/index.html": "hello world!",
				},
			},
		},
		{
			name:        "spa",
			path:        "/anything",
			want:        "hello world!",
			status:      http.StatusOK,
			contentType: "text/html",

			storage: map[string]map[string]string{
				bucketName: {
					"test/_redirects": "/* /index.html 200",
					"test/index.html": "hello world!",
				},
			},
		},
		{
			name:        "not-found",
			path:        "/anything",
			want:        "404 not found",
			status:      http.StatusNotFound,
			contentType: "text/plain; charset=utf-8",

			storage: map[string]map[string]string{
				bucketName: {},
			},
		},
		{
			name:        "not-found-custom",
			path:        "/anything",
			want:        "boom!",
			status:      http.StatusNotFound,
			contentType: "text/html",

			storage: map[string]map[string]string{
				bucketName: {
					"test/404.html": "boom!",
				},
			},
		},
		{
			name:        "images",
			path:        "/profile.jpg",
			want:        "image",
			status:      http.StatusOK,
			contentType: "image/jpeg",

			storage: map[string]map[string]string{
				bucketName: {
					"test/profile.jpg": "image",
				},
			},
		},
		{
			name:        "redirects-query-param",
			path:        "/anything?query=param",
			want:        `<a href="/about.html?query=param">Moved Permanently</a>.`,
			wantUrl:     "/about.html?query=param",
			status:      http.StatusMovedPermanently,
			contentType: "text/html; charset=utf-8",

			storage: map[string]map[string]string{
				bucketName: {
					"test/_redirects": "/anything /about.html 301",
					"test/about.html": "hello world!",
				},
			},
		},
		{
			name: "conditional-if-modified-since-future",
			path: "/test.html",
			reqHeaders: map[string]string{
				"If-Modified-Since": time.Now().UTC().Add(time.Hour).Format(http.TimeFormat),
			},
			want:        "",
			status:      http.StatusNotModified,
			contentType: "",

			storage: map[string]map[string]string{
				bucketName: {
					"test/test.html": "hello world!",
				},
			},
		},
		{
			name: "conditional-if-modified-since-past",
			path: "/test.html",
			reqHeaders: map[string]string{
				"If-Modified-Since": time.Now().UTC().Add(-time.Hour).Format(http.TimeFormat),
			},
			want:        "hello world!",
			status:      http.StatusOK,
			contentType: "text/html",

			storage: map[string]map[string]string{
				bucketName: {
					"test/test.html": "hello world!",
				},
			},
		},
		{
			name: "conditional-if-none-match-pass",
			path: "/test.html",
			reqHeaders: map[string]string{
				"If-None-Match": "\"static-etag-for-testing-purposes\"",
			},
			want:        "",
			status:      http.StatusNotModified,
			contentType: "",

			storage: map[string]map[string]string{
				bucketName: {
					"test/test.html": "hello world!",
				},
			},
		},
		{
			name: "conditional-if-none-match-fail",
			path: "/test.html",
			reqHeaders: map[string]string{
				"If-None-Match": "\"non-matching-etag\"",
			},
			want:        "hello world!",
			status:      http.StatusOK,
			contentType: "text/html",

			storage: map[string]map[string]string{
				bucketName: {
					"test/test.html": "hello world!",
				},
			},
		},
		{
			name: "conditional-if-none-match-and-if-modified-since",
			path: "/test.html",
			reqHeaders: map[string]string{
				// The matching etag should take precedence over the past mod time
				"If-None-Match":     "\"static-etag-for-testing-purposes\"",
				"If-Modified-Since": time.Now().UTC().Add(-time.Hour).Format(http.TimeFormat),
			},
			want:        "",
			status:      http.StatusNotModified,
			contentType: "",

			storage: map[string]map[string]string{
				bucketName: {
					"test/test.html": "hello world!",
				},
			},
		},
	}

	for _, tc := range tt {
		t.Run(tc.name, func(t *testing.T) {
			request := httptest.NewRequest("GET", dbpool.mkpath(tc.path), strings.NewReader(""))
			for key, val := range tc.reqHeaders {
				request.Header.Set(key, val)
			}
			responseRecorder := httptest.NewRecorder()

			st, _ := storage.NewStorageMemory(tc.storage)
			cfg := NewConfigSite(logger, dbpool, st)
			cfg.Domain = "pgs.test"
			router := NewWebRouter(logger, dbpool, st, cfg.Domain, cfg.TxtPrefix)
			router.ServeHTTP(responseRecorder, request)

			if responseRecorder.Code != tc.status {
				t.Errorf("Want status '%d', got '%d'", tc.status, responseRecorder.Code)
			}

			ct := responseRecorder.Header().Get("content-type")
			if ct != tc.contentType {
				t.Errorf("Want status '%s', got '%s'", tc.contentType, ct)
			}

			body := strings.TrimSpace(responseRecorder.Body.String())
			if body != tc.want {
				t.Errorf("Want '%s', got '%s'", tc.want, body)
			}

			if tc.wantUrl != "" {
				location, err := responseRecorder.Result().Location()
				if err != nil {
					t.Errorf("err: %s", err.Error())
				}
				if location == nil {
					t.Error("no location header in response")
					return
				}
				if tc.wantUrl != location.String() {
					t.Errorf("Want '%s', got '%s'", tc.wantUrl, location.String())
				}
			}
		})
	}
}

type ImageStorageMemory struct {
	*storage.StorageMemory
	Opts  *storage.ImgProcessOpts
	Fpath string
}

func (s *ImageStorageMemory) ServeObject(bucket sst.Bucket, fpath string, opts *storage.ImgProcessOpts) (io.ReadCloser, *sst.ObjectInfo, error) {
	s.Opts = opts
	s.Fpath = fpath
	info := sst.ObjectInfo{
		Metadata: make(http.Header),
	}
	info.Metadata.Set("content-type", "image/jpeg")
	return io.NopCloser(strings.NewReader("hello world!")), &info, nil
}

func TestImageManipulation(t *testing.T) {
	logger := slog.Default()
	dbpool := NewPgsDb(logger)
	bucketName := GetAssetBucketName(dbpool.Users[0].GetID())

	tt := []ApiExample{
		{
			name:        "root-img",
			path:        "/app.jpg/s:500/rt:90",
			want:        "hello world!",
			status:      http.StatusOK,
			contentType: "image/jpeg",

			storage: map[string]map[string]string{
				bucketName: {
					"test/app.jpg": "hello world!",
				},
			},
		},
		{
			name:        "root-subdir-img",
			path:        "/subdir/app.jpg/rt:90/s:500",
			want:        "hello world!",
			status:      http.StatusOK,
			contentType: "image/jpeg",

			storage: map[string]map[string]string{
				bucketName: {
					"test/subdir/app.jpg": "hello world!",
				},
			},
		},
	}

	for _, tc := range tt {
		t.Run(tc.name, func(t *testing.T) {
			request := httptest.NewRequest("GET", dbpool.mkpath(tc.path), strings.NewReader(""))
			responseRecorder := httptest.NewRecorder()

			memst, _ := storage.NewStorageMemory(tc.storage)
			st := &ImageStorageMemory{
				StorageMemory: memst,
				Opts: &storage.ImgProcessOpts{
					Ratio: &storage.Ratio{},
				},
			}
			cfg := NewConfigSite(logger, dbpool, st)
			cfg.Domain = "pgs.test"
			router := NewWebRouter(logger, dbpool, st, cfg.Domain, cfg.TxtPrefix)
			router.ServeHTTP(responseRecorder, request)

			if responseRecorder.Code != tc.status {
				t.Errorf("Want status '%d', got '%d'", tc.status, responseRecorder.Code)
			}

			ct := responseRecorder.Header().Get("content-type")
			if ct != tc.contentType {
				t.Errorf("Want content type '%s', got '%s'", tc.contentType, ct)
			}

			body := strings.TrimSpace(responseRecorder.Body.String())
			if body != tc.want {
				t.Errorf("Want '%s', got '%s'", tc.want, body)
			}

			if st.Opts.Ratio.Width != 500 {
				t.Errorf("Want ratio width '500', got '%d'", st.Opts.Ratio.Width)
				return
			}

			if st.Opts.Rotate != 90 {
				t.Errorf("Want rotate '90', got '%d'", st.Opts.Rotate)
				return
			}
		})
	}
}