Compare commits
3 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| e1e7aa64ca | |||
| 703b2fafb0 | |||
| 69a1ab86c2 |
+38
-6
@@ -3,30 +3,62 @@ package main
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"log/slog"
|
||||
"net/http"
|
||||
"os"
|
||||
"strings"
|
||||
|
||||
"gitea.leeworks.dev/0xwheatyz/gitea-mobile/internal/config"
|
||||
"gitea.leeworks.dev/0xwheatyz/gitea-mobile/internal/handlers"
|
||||
"gitea.leeworks.dev/0xwheatyz/gitea-mobile/internal/middleware"
|
||||
)
|
||||
|
||||
func main() {
|
||||
addr := os.Getenv("LISTEN_ADDR")
|
||||
if addr == "" {
|
||||
addr = ":8080"
|
||||
// Set up structured logging.
|
||||
slog.SetDefault(slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{
|
||||
Level: slog.LevelInfo,
|
||||
})))
|
||||
|
||||
cfg, err := config.Load()
|
||||
if err != nil {
|
||||
log.Fatalf("configuration error: %v", err)
|
||||
}
|
||||
|
||||
// Determine if cookies should be marked Secure (disable for local dev).
|
||||
secureCookies := !strings.HasPrefix(cfg.ListenAddr, "localhost") &&
|
||||
!strings.HasPrefix(cfg.ListenAddr, "127.0.0.1")
|
||||
|
||||
mux := http.NewServeMux()
|
||||
|
||||
// Health endpoint (no auth required).
|
||||
mux.HandleFunc("GET /health", func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
fmt.Fprintln(w, "ok")
|
||||
})
|
||||
|
||||
// Settings handler.
|
||||
settingsHandler := &handlers.SettingsHandler{
|
||||
SessionSecret: cfg.SessionSecret,
|
||||
SecureCookies: secureCookies,
|
||||
}
|
||||
mux.HandleFunc("/settings", settingsHandler.ServeHTTP)
|
||||
|
||||
// Static file server.
|
||||
mux.Handle("/static/", http.StripPrefix("/static/", http.FileServer(http.Dir("static"))))
|
||||
|
||||
// Placeholder dashboard (will be replaced in issue #4/#5).
|
||||
mux.HandleFunc("GET /", func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
||||
fmt.Fprintln(w, "<h1>Gitea Mobile</h1><p>Coming soon.</p>")
|
||||
fmt.Fprintln(w, "<h1>Gitea Mobile</h1><p>Dashboard coming soon.</p>")
|
||||
})
|
||||
|
||||
log.Printf("listening on %s", addr)
|
||||
if err := http.ListenAndServe(addr, mux); err != nil {
|
||||
// Apply middleware chain: logging -> auth.
|
||||
var handler http.Handler = mux
|
||||
handler = middleware.Auth(cfg.SessionSecret)(handler)
|
||||
handler = middleware.Logging()(handler)
|
||||
|
||||
slog.Info("server starting", "addr", cfg.ListenAddr, "gitea_url", cfg.GiteaURL)
|
||||
if err := http.ListenAndServe(cfg.ListenAddr, handler); err != nil {
|
||||
log.Fatalf("server error: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -0,0 +1,104 @@
|
||||
package auth
|
||||
|
||||
import (
|
||||
"crypto/hmac"
|
||||
"crypto/sha256"
|
||||
"encoding/base64"
|
||||
"errors"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
const (
|
||||
cookieName = "gitea_token"
|
||||
cookieMaxAge = 30 * 24 * 60 * 60 // 30 days in seconds
|
||||
)
|
||||
|
||||
var (
|
||||
ErrInvalidSignature = errors.New("invalid cookie signature")
|
||||
ErrMalformedCookie = errors.New("malformed cookie value")
|
||||
)
|
||||
|
||||
// SetTokenCookie stores a Gitea API token in a signed HTTP-only cookie.
|
||||
func SetTokenCookie(w http.ResponseWriter, token string, secret string, secure bool) {
|
||||
signed := sign(token, secret)
|
||||
|
||||
http.SetCookie(w, &http.Cookie{
|
||||
Name: cookieName,
|
||||
Value: signed,
|
||||
Path: "/",
|
||||
MaxAge: cookieMaxAge,
|
||||
HttpOnly: true,
|
||||
Secure: secure,
|
||||
SameSite: http.SameSiteStrictMode,
|
||||
Expires: time.Now().Add(30 * 24 * time.Hour),
|
||||
})
|
||||
}
|
||||
|
||||
// ClearTokenCookie removes the token cookie.
|
||||
func ClearTokenCookie(w http.ResponseWriter, secure bool) {
|
||||
http.SetCookie(w, &http.Cookie{
|
||||
Name: cookieName,
|
||||
Value: "",
|
||||
Path: "/",
|
||||
MaxAge: -1,
|
||||
HttpOnly: true,
|
||||
Secure: secure,
|
||||
SameSite: http.SameSiteStrictMode,
|
||||
})
|
||||
}
|
||||
|
||||
// GetToken extracts and verifies the Gitea API token from the request cookie.
|
||||
// Returns the token string or an error if the cookie is missing or invalid.
|
||||
func GetToken(r *http.Request, secret string) (string, error) {
|
||||
cookie, err := r.Cookie(cookieName)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
token, err := verify(cookie.Value, secret)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
return token, nil
|
||||
}
|
||||
|
||||
// sign creates a signed cookie value: base64(token).base64(hmac-sha256(token))
|
||||
func sign(token string, secret string) string {
|
||||
encodedToken := base64.URLEncoding.EncodeToString([]byte(token))
|
||||
mac := computeHMAC(encodedToken, secret)
|
||||
return fmt.Sprintf("%s.%s", encodedToken, mac)
|
||||
}
|
||||
|
||||
// verify checks the HMAC signature and returns the original token.
|
||||
func verify(signed string, secret string) (string, error) {
|
||||
parts := strings.SplitN(signed, ".", 2)
|
||||
if len(parts) != 2 {
|
||||
return "", ErrMalformedCookie
|
||||
}
|
||||
|
||||
encodedToken := parts[0]
|
||||
providedMAC := parts[1]
|
||||
expectedMAC := computeHMAC(encodedToken, secret)
|
||||
|
||||
if !hmac.Equal([]byte(providedMAC), []byte(expectedMAC)) {
|
||||
return "", ErrInvalidSignature
|
||||
}
|
||||
|
||||
tokenBytes, err := base64.URLEncoding.DecodeString(encodedToken)
|
||||
if err != nil {
|
||||
return "", ErrMalformedCookie
|
||||
}
|
||||
|
||||
return string(tokenBytes), nil
|
||||
}
|
||||
|
||||
// computeHMAC generates a base64-encoded HMAC-SHA256 of the given data.
|
||||
func computeHMAC(data string, secret string) string {
|
||||
h := hmac.New(sha256.New, []byte(secret))
|
||||
h.Write([]byte(data))
|
||||
return base64.URLEncoding.EncodeToString(h.Sum(nil))
|
||||
}
|
||||
@@ -0,0 +1,89 @@
|
||||
package auth
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
)
|
||||
|
||||
const testSecret = "test-secret-that-is-at-least-32-chars-long"
|
||||
|
||||
func TestSignAndVerify(t *testing.T) {
|
||||
token := "abc123-gitea-token"
|
||||
signed := sign(token, testSecret)
|
||||
|
||||
got, err := verify(signed, testSecret)
|
||||
if err != nil {
|
||||
t.Fatalf("verify failed: %v", err)
|
||||
}
|
||||
if got != token {
|
||||
t.Errorf("got %q, want %q", got, token)
|
||||
}
|
||||
}
|
||||
|
||||
func TestVerify_InvalidSignature(t *testing.T) {
|
||||
token := "abc123-gitea-token"
|
||||
signed := sign(token, testSecret)
|
||||
|
||||
_, err := verify(signed, "wrong-secret-that-is-at-least-32-chars")
|
||||
if err != ErrInvalidSignature {
|
||||
t.Errorf("expected ErrInvalidSignature, got %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestVerify_MalformedCookie(t *testing.T) {
|
||||
_, err := verify("no-dot-separator", testSecret)
|
||||
if err != ErrMalformedCookie {
|
||||
t.Errorf("expected ErrMalformedCookie, got %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSetAndGetToken(t *testing.T) {
|
||||
token := "my-gitea-api-token"
|
||||
|
||||
// Create a response recorder to capture the Set-Cookie header.
|
||||
w := httptest.NewRecorder()
|
||||
SetTokenCookie(w, token, testSecret, false)
|
||||
|
||||
// Extract the cookie from the response.
|
||||
resp := w.Result()
|
||||
cookies := resp.Cookies()
|
||||
if len(cookies) == 0 {
|
||||
t.Fatal("expected a cookie to be set")
|
||||
}
|
||||
|
||||
// Create a new request with the cookie.
|
||||
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
||||
req.AddCookie(cookies[0])
|
||||
|
||||
got, err := GetToken(req, testSecret)
|
||||
if err != nil {
|
||||
t.Fatalf("GetToken failed: %v", err)
|
||||
}
|
||||
if got != token {
|
||||
t.Errorf("got %q, want %q", got, token)
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetToken_NoCookie(t *testing.T) {
|
||||
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
||||
|
||||
_, err := GetToken(req, testSecret)
|
||||
if err == nil {
|
||||
t.Fatal("expected error for missing cookie")
|
||||
}
|
||||
}
|
||||
|
||||
func TestClearTokenCookie(t *testing.T) {
|
||||
w := httptest.NewRecorder()
|
||||
ClearTokenCookie(w, false)
|
||||
|
||||
resp := w.Result()
|
||||
cookies := resp.Cookies()
|
||||
if len(cookies) == 0 {
|
||||
t.Fatal("expected a cookie to be set")
|
||||
}
|
||||
if cookies[0].MaxAge != -1 {
|
||||
t.Errorf("MaxAge = %d, want -1", cookies[0].MaxAge)
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,50 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
)
|
||||
|
||||
// Config holds application configuration loaded from environment variables.
|
||||
type Config struct {
|
||||
// GiteaURL is the base URL of the Gitea instance.
|
||||
GiteaURL string
|
||||
|
||||
// GiteaToken is the default API token (optional; users can set their own via cookie).
|
||||
GiteaToken string
|
||||
|
||||
// ListenAddr is the server listen address.
|
||||
ListenAddr string
|
||||
|
||||
// SessionSecret is the HMAC key for signing session cookies.
|
||||
SessionSecret string
|
||||
}
|
||||
|
||||
// Load reads configuration from environment variables.
|
||||
// Returns an error if required variables are missing.
|
||||
func Load() (*Config, error) {
|
||||
cfg := &Config{
|
||||
GiteaURL: os.Getenv("GITEA_URL"),
|
||||
GiteaToken: os.Getenv("GITEA_TOKEN"),
|
||||
ListenAddr: os.Getenv("LISTEN_ADDR"),
|
||||
SessionSecret: os.Getenv("SESSION_SECRET"),
|
||||
}
|
||||
|
||||
if cfg.ListenAddr == "" {
|
||||
cfg.ListenAddr = ":8080"
|
||||
}
|
||||
|
||||
if cfg.GiteaURL == "" {
|
||||
return nil, fmt.Errorf("GITEA_URL environment variable is required")
|
||||
}
|
||||
|
||||
if cfg.SessionSecret == "" {
|
||||
return nil, fmt.Errorf("SESSION_SECRET environment variable is required")
|
||||
}
|
||||
|
||||
if len(cfg.SessionSecret) < 32 {
|
||||
return nil, fmt.Errorf("SESSION_SECRET must be at least 32 characters")
|
||||
}
|
||||
|
||||
return cfg, nil
|
||||
}
|
||||
@@ -0,0 +1,89 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"os"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestLoad_Success(t *testing.T) {
|
||||
os.Setenv("GITEA_URL", "https://gitea.example.com")
|
||||
os.Setenv("SESSION_SECRET", "test-secret-that-is-at-least-32-chars-long")
|
||||
os.Setenv("LISTEN_ADDR", ":9090")
|
||||
os.Setenv("GITEA_TOKEN", "test-token")
|
||||
defer func() {
|
||||
os.Unsetenv("GITEA_URL")
|
||||
os.Unsetenv("SESSION_SECRET")
|
||||
os.Unsetenv("LISTEN_ADDR")
|
||||
os.Unsetenv("GITEA_TOKEN")
|
||||
}()
|
||||
|
||||
cfg, err := Load()
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
|
||||
if cfg.GiteaURL != "https://gitea.example.com" {
|
||||
t.Errorf("GiteaURL = %q, want %q", cfg.GiteaURL, "https://gitea.example.com")
|
||||
}
|
||||
if cfg.ListenAddr != ":9090" {
|
||||
t.Errorf("ListenAddr = %q, want %q", cfg.ListenAddr, ":9090")
|
||||
}
|
||||
if cfg.GiteaToken != "test-token" {
|
||||
t.Errorf("GiteaToken = %q, want %q", cfg.GiteaToken, "test-token")
|
||||
}
|
||||
}
|
||||
|
||||
func TestLoad_DefaultListenAddr(t *testing.T) {
|
||||
os.Setenv("GITEA_URL", "https://gitea.example.com")
|
||||
os.Setenv("SESSION_SECRET", "test-secret-that-is-at-least-32-chars-long")
|
||||
os.Unsetenv("LISTEN_ADDR")
|
||||
defer func() {
|
||||
os.Unsetenv("GITEA_URL")
|
||||
os.Unsetenv("SESSION_SECRET")
|
||||
}()
|
||||
|
||||
cfg, err := Load()
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
|
||||
if cfg.ListenAddr != ":8080" {
|
||||
t.Errorf("ListenAddr = %q, want %q", cfg.ListenAddr, ":8080")
|
||||
}
|
||||
}
|
||||
|
||||
func TestLoad_MissingGiteaURL(t *testing.T) {
|
||||
os.Unsetenv("GITEA_URL")
|
||||
os.Setenv("SESSION_SECRET", "test-secret-that-is-at-least-32-chars-long")
|
||||
defer os.Unsetenv("SESSION_SECRET")
|
||||
|
||||
_, err := Load()
|
||||
if err == nil {
|
||||
t.Fatal("expected error for missing GITEA_URL")
|
||||
}
|
||||
}
|
||||
|
||||
func TestLoad_MissingSessionSecret(t *testing.T) {
|
||||
os.Setenv("GITEA_URL", "https://gitea.example.com")
|
||||
os.Unsetenv("SESSION_SECRET")
|
||||
defer os.Unsetenv("GITEA_URL")
|
||||
|
||||
_, err := Load()
|
||||
if err == nil {
|
||||
t.Fatal("expected error for missing SESSION_SECRET")
|
||||
}
|
||||
}
|
||||
|
||||
func TestLoad_ShortSessionSecret(t *testing.T) {
|
||||
os.Setenv("GITEA_URL", "https://gitea.example.com")
|
||||
os.Setenv("SESSION_SECRET", "tooshort")
|
||||
defer func() {
|
||||
os.Unsetenv("GITEA_URL")
|
||||
os.Unsetenv("SESSION_SECRET")
|
||||
}()
|
||||
|
||||
_, err := Load()
|
||||
if err == nil {
|
||||
t.Fatal("expected error for short SESSION_SECRET")
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,625 @@
|
||||
// Package gitea provides an aggregation layer over the Gitea API,
|
||||
// supporting concurrent fetching across multiple organizations and repos
|
||||
// with in-memory caching.
|
||||
package gitea
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"sort"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
// Client wraps the Gitea API with aggregation and caching capabilities.
|
||||
type Client struct {
|
||||
baseURL string
|
||||
httpClient *http.Client
|
||||
|
||||
mu sync.RWMutex
|
||||
cache map[string]*cacheEntry
|
||||
|
||||
// maxConcurrent controls the semaphore size for parallel API calls.
|
||||
maxConcurrent int
|
||||
// cacheTTL controls how long cache entries remain valid.
|
||||
cacheTTL time.Duration
|
||||
}
|
||||
|
||||
type cacheEntry struct {
|
||||
data interface{}
|
||||
expiresAt time.Time
|
||||
}
|
||||
|
||||
// Org represents a Gitea organization.
|
||||
type Org struct {
|
||||
Name string `json:"username"`
|
||||
FullName string `json:"full_name"`
|
||||
Description string `json:"description"`
|
||||
AvatarURL string `json:"avatar_url"`
|
||||
}
|
||||
|
||||
// Repo represents a Gitea repository.
|
||||
type Repo struct {
|
||||
ID int64 `json:"id"`
|
||||
Name string `json:"name"`
|
||||
FullName string `json:"full_name"`
|
||||
Description string `json:"description"`
|
||||
Owner struct {
|
||||
Login string `json:"login"`
|
||||
} `json:"owner"`
|
||||
HTMLURL string `json:"html_url"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
}
|
||||
|
||||
// Issue represents a Gitea issue.
|
||||
type Issue struct {
|
||||
ID int64 `json:"id"`
|
||||
Number int64 `json:"number"`
|
||||
Title string `json:"title"`
|
||||
Body string `json:"body"`
|
||||
State string `json:"state"`
|
||||
Labels []struct {
|
||||
ID int64 `json:"id"`
|
||||
Name string `json:"name"`
|
||||
Color string `json:"color"`
|
||||
} `json:"labels"`
|
||||
Assignee *struct {
|
||||
Login string `json:"login"`
|
||||
AvatarURL string `json:"avatar_url"`
|
||||
} `json:"assignee"`
|
||||
Assignees []struct {
|
||||
Login string `json:"login"`
|
||||
AvatarURL string `json:"avatar_url"`
|
||||
} `json:"assignees"`
|
||||
HTMLURL string `json:"html_url"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
RepoOwner string `json:"-"` // populated after fetch
|
||||
RepoName string `json:"-"` // populated after fetch
|
||||
}
|
||||
|
||||
// PullRequest represents a Gitea pull request.
|
||||
type PullRequest struct {
|
||||
ID int64 `json:"id"`
|
||||
Number int64 `json:"number"`
|
||||
Title string `json:"title"`
|
||||
Body string `json:"body"`
|
||||
State string `json:"state"`
|
||||
Labels []struct {
|
||||
ID int64 `json:"id"`
|
||||
Name string `json:"name"`
|
||||
Color string `json:"color"`
|
||||
} `json:"labels"`
|
||||
User *struct {
|
||||
Login string `json:"login"`
|
||||
AvatarURL string `json:"avatar_url"`
|
||||
} `json:"user"`
|
||||
Mergeable bool `json:"mergeable"`
|
||||
HTMLURL string `json:"html_url"`
|
||||
DiffURL string `json:"diff_url"`
|
||||
Additions int `json:"additions"`
|
||||
Deletions int `json:"deletions"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
UpdatedAt time.Time `json:"updated_at"`
|
||||
RepoOwner string `json:"-"` // populated after fetch
|
||||
RepoName string `json:"-"` // populated after fetch
|
||||
}
|
||||
|
||||
// TriageItem represents an item in the triage queue.
|
||||
type TriageItem struct {
|
||||
Type string // "issue" or "pull"
|
||||
RepoOwner string
|
||||
RepoName string
|
||||
Number int64
|
||||
Title string
|
||||
HTMLURL string
|
||||
Labels []string
|
||||
UpdatedAt time.Time
|
||||
}
|
||||
|
||||
// NewClient creates a new Gitea API client.
|
||||
func NewClient(baseURL string) *Client {
|
||||
return &Client{
|
||||
baseURL: strings.TrimRight(baseURL, "/"),
|
||||
httpClient: &http.Client{
|
||||
Timeout: 30 * time.Second,
|
||||
},
|
||||
cache: make(map[string]*cacheEntry),
|
||||
maxConcurrent: 5,
|
||||
cacheTTL: 30 * time.Second,
|
||||
}
|
||||
}
|
||||
|
||||
// doRequest performs an authenticated HTTP request to the Gitea API.
|
||||
func (c *Client) doRequest(ctx context.Context, token, method, path string, body io.Reader) (*http.Response, error) {
|
||||
url := c.baseURL + "/api/v1" + path
|
||||
|
||||
req, err := http.NewRequestWithContext(ctx, method, url, body)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("creating request: %w", err)
|
||||
}
|
||||
|
||||
req.Header.Set("Authorization", "token "+token)
|
||||
req.Header.Set("Accept", "application/json")
|
||||
if body != nil {
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
}
|
||||
|
||||
resp, err := c.httpClient.Do(req)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("executing request: %w", err)
|
||||
}
|
||||
|
||||
if resp.StatusCode >= 400 {
|
||||
defer resp.Body.Close()
|
||||
respBody, _ := io.ReadAll(resp.Body)
|
||||
return nil, fmt.Errorf("API error %d: %s", resp.StatusCode, string(respBody))
|
||||
}
|
||||
|
||||
return resp, nil
|
||||
}
|
||||
|
||||
// getFromCache returns cached data if still valid.
|
||||
func (c *Client) getFromCache(key string) (interface{}, bool) {
|
||||
c.mu.RLock()
|
||||
defer c.mu.RUnlock()
|
||||
|
||||
entry, ok := c.cache[key]
|
||||
if !ok || time.Now().After(entry.expiresAt) {
|
||||
return nil, false
|
||||
}
|
||||
return entry.data, true
|
||||
}
|
||||
|
||||
// setCache stores data in cache with TTL.
|
||||
func (c *Client) setCache(key string, data interface{}) {
|
||||
c.mu.Lock()
|
||||
defer c.mu.Unlock()
|
||||
|
||||
c.cache[key] = &cacheEntry{
|
||||
data: data,
|
||||
expiresAt: time.Now().Add(c.cacheTTL),
|
||||
}
|
||||
}
|
||||
|
||||
// invalidateCache removes entries matching the given prefix.
|
||||
func (c *Client) invalidateCache(prefix string) {
|
||||
c.mu.Lock()
|
||||
defer c.mu.Unlock()
|
||||
|
||||
for k := range c.cache {
|
||||
if strings.HasPrefix(k, prefix) {
|
||||
delete(c.cache, k)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// InvalidateAll clears the entire cache (called on write operations).
|
||||
func (c *Client) InvalidateAll() {
|
||||
c.mu.Lock()
|
||||
defer c.mu.Unlock()
|
||||
c.cache = make(map[string]*cacheEntry)
|
||||
}
|
||||
|
||||
// ListOrgs returns the organizations the authenticated user belongs to.
|
||||
func (c *Client) ListOrgs(ctx context.Context, token string) ([]Org, error) {
|
||||
cacheKey := "orgs"
|
||||
if cached, ok := c.getFromCache(cacheKey); ok {
|
||||
return cached.([]Org), nil
|
||||
}
|
||||
|
||||
resp, err := c.doRequest(ctx, token, http.MethodGet, "/user/orgs?limit=50", nil)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("listing orgs: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
var orgs []Org
|
||||
if err := json.NewDecoder(resp.Body).Decode(&orgs); err != nil {
|
||||
return nil, fmt.Errorf("decoding orgs: %w", err)
|
||||
}
|
||||
|
||||
c.setCache(cacheKey, orgs)
|
||||
return orgs, nil
|
||||
}
|
||||
|
||||
// ListOrgRepos returns all repositories for a given organization.
|
||||
func (c *Client) ListOrgRepos(ctx context.Context, token, org string) ([]Repo, error) {
|
||||
cacheKey := fmt.Sprintf("repos-%s", org)
|
||||
if cached, ok := c.getFromCache(cacheKey); ok {
|
||||
return cached.([]Repo), nil
|
||||
}
|
||||
|
||||
var allRepos []Repo
|
||||
page := 1
|
||||
for {
|
||||
path := fmt.Sprintf("/orgs/%s/repos?limit=50&page=%d", org, page)
|
||||
resp, err := c.doRequest(ctx, token, http.MethodGet, path, nil)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("listing repos for %s: %w", org, err)
|
||||
}
|
||||
|
||||
var repos []Repo
|
||||
if err := json.NewDecoder(resp.Body).Decode(&repos); err != nil {
|
||||
resp.Body.Close()
|
||||
return nil, fmt.Errorf("decoding repos: %w", err)
|
||||
}
|
||||
resp.Body.Close()
|
||||
|
||||
if len(repos) == 0 {
|
||||
break
|
||||
}
|
||||
allRepos = append(allRepos, repos...)
|
||||
if len(repos) < 50 {
|
||||
break
|
||||
}
|
||||
page++
|
||||
}
|
||||
|
||||
c.setCache(cacheKey, allRepos)
|
||||
return allRepos, nil
|
||||
}
|
||||
|
||||
// ListOrgsAndRepos returns a map of org name to repos for all orgs the user belongs to.
|
||||
func (c *Client) ListOrgsAndRepos(ctx context.Context, token string) (map[string][]Repo, error) {
|
||||
orgs, err := c.ListOrgs(ctx, token)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
result := make(map[string][]Repo)
|
||||
var mu sync.Mutex
|
||||
sem := make(chan struct{}, c.maxConcurrent)
|
||||
var wg sync.WaitGroup
|
||||
var firstErr error
|
||||
|
||||
for _, org := range orgs {
|
||||
wg.Add(1)
|
||||
go func(orgName string) {
|
||||
defer wg.Done()
|
||||
sem <- struct{}{}
|
||||
defer func() { <-sem }()
|
||||
|
||||
repos, err := c.ListOrgRepos(ctx, token, orgName)
|
||||
if err != nil {
|
||||
mu.Lock()
|
||||
if firstErr == nil {
|
||||
firstErr = err
|
||||
}
|
||||
mu.Unlock()
|
||||
return
|
||||
}
|
||||
|
||||
mu.Lock()
|
||||
result[orgName] = repos
|
||||
mu.Unlock()
|
||||
}(org.Name)
|
||||
}
|
||||
|
||||
wg.Wait()
|
||||
|
||||
if firstErr != nil {
|
||||
return nil, firstErr
|
||||
}
|
||||
return result, nil
|
||||
}
|
||||
|
||||
// ListAllIssues fetches all open issues across all repos in the given orgs,
|
||||
// using concurrent requests with a semaphore.
|
||||
func (c *Client) ListAllIssues(ctx context.Context, token string, orgs []string) ([]Issue, error) {
|
||||
cacheKey := fmt.Sprintf("issues-%s", strings.Join(orgs, ","))
|
||||
if cached, ok := c.getFromCache(cacheKey); ok {
|
||||
return cached.([]Issue), nil
|
||||
}
|
||||
|
||||
// First, collect all repos for the given orgs.
|
||||
var allRepos []Repo
|
||||
for _, org := range orgs {
|
||||
repos, err := c.ListOrgRepos(ctx, token, org)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("listing repos for %s: %w", org, err)
|
||||
}
|
||||
allRepos = append(allRepos, repos...)
|
||||
}
|
||||
|
||||
// Fan out issue fetching across repos.
|
||||
var allIssues []Issue
|
||||
var mu sync.Mutex
|
||||
sem := make(chan struct{}, c.maxConcurrent)
|
||||
var wg sync.WaitGroup
|
||||
var firstErr error
|
||||
|
||||
for _, repo := range allRepos {
|
||||
wg.Add(1)
|
||||
go func(r Repo) {
|
||||
defer wg.Done()
|
||||
sem <- struct{}{}
|
||||
defer func() { <-sem }()
|
||||
|
||||
path := fmt.Sprintf("/repos/%s/issues?state=open&type=issues&limit=50", r.FullName)
|
||||
resp, err := c.doRequest(ctx, token, http.MethodGet, path, nil)
|
||||
if err != nil {
|
||||
mu.Lock()
|
||||
if firstErr == nil {
|
||||
firstErr = fmt.Errorf("fetching issues for %s: %w", r.FullName, err)
|
||||
}
|
||||
mu.Unlock()
|
||||
return
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
var issues []Issue
|
||||
if err := json.NewDecoder(resp.Body).Decode(&issues); err != nil {
|
||||
mu.Lock()
|
||||
if firstErr == nil {
|
||||
firstErr = fmt.Errorf("decoding issues for %s: %w", r.FullName, err)
|
||||
}
|
||||
mu.Unlock()
|
||||
return
|
||||
}
|
||||
|
||||
// Tag each issue with repo info.
|
||||
for i := range issues {
|
||||
issues[i].RepoOwner = r.Owner.Login
|
||||
issues[i].RepoName = r.Name
|
||||
}
|
||||
|
||||
mu.Lock()
|
||||
allIssues = append(allIssues, issues...)
|
||||
mu.Unlock()
|
||||
}(repo)
|
||||
}
|
||||
|
||||
wg.Wait()
|
||||
|
||||
if firstErr != nil {
|
||||
return nil, firstErr
|
||||
}
|
||||
|
||||
// Sort by updated time, newest first.
|
||||
sort.Slice(allIssues, func(i, j int) bool {
|
||||
return allIssues[i].UpdatedAt.After(allIssues[j].UpdatedAt)
|
||||
})
|
||||
|
||||
c.setCache(cacheKey, allIssues)
|
||||
return allIssues, nil
|
||||
}
|
||||
|
||||
// ListAllPullRequests fetches all open PRs across all repos in the given orgs.
|
||||
func (c *Client) ListAllPullRequests(ctx context.Context, token string, orgs []string) ([]PullRequest, error) {
|
||||
cacheKey := fmt.Sprintf("pulls-%s", strings.Join(orgs, ","))
|
||||
if cached, ok := c.getFromCache(cacheKey); ok {
|
||||
return cached.([]PullRequest), nil
|
||||
}
|
||||
|
||||
var allRepos []Repo
|
||||
for _, org := range orgs {
|
||||
repos, err := c.ListOrgRepos(ctx, token, org)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("listing repos for %s: %w", org, err)
|
||||
}
|
||||
allRepos = append(allRepos, repos...)
|
||||
}
|
||||
|
||||
var allPRs []PullRequest
|
||||
var mu sync.Mutex
|
||||
sem := make(chan struct{}, c.maxConcurrent)
|
||||
var wg sync.WaitGroup
|
||||
var firstErr error
|
||||
|
||||
for _, repo := range allRepos {
|
||||
wg.Add(1)
|
||||
go func(r Repo) {
|
||||
defer wg.Done()
|
||||
sem <- struct{}{}
|
||||
defer func() { <-sem }()
|
||||
|
||||
path := fmt.Sprintf("/repos/%s/pulls?state=open&limit=50", r.FullName)
|
||||
resp, err := c.doRequest(ctx, token, http.MethodGet, path, nil)
|
||||
if err != nil {
|
||||
mu.Lock()
|
||||
if firstErr == nil {
|
||||
firstErr = fmt.Errorf("fetching PRs for %s: %w", r.FullName, err)
|
||||
}
|
||||
mu.Unlock()
|
||||
return
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
var prs []PullRequest
|
||||
if err := json.NewDecoder(resp.Body).Decode(&prs); err != nil {
|
||||
mu.Lock()
|
||||
if firstErr == nil {
|
||||
firstErr = fmt.Errorf("decoding PRs for %s: %w", r.FullName, err)
|
||||
}
|
||||
mu.Unlock()
|
||||
return
|
||||
}
|
||||
|
||||
for i := range prs {
|
||||
prs[i].RepoOwner = r.Owner.Login
|
||||
prs[i].RepoName = r.Name
|
||||
}
|
||||
|
||||
mu.Lock()
|
||||
allPRs = append(allPRs, prs...)
|
||||
mu.Unlock()
|
||||
}(repo)
|
||||
}
|
||||
|
||||
wg.Wait()
|
||||
|
||||
if firstErr != nil {
|
||||
return nil, firstErr
|
||||
}
|
||||
|
||||
sort.Slice(allPRs, func(i, j int) bool {
|
||||
return allPRs[i].UpdatedAt.After(allPRs[j].UpdatedAt)
|
||||
})
|
||||
|
||||
c.setCache(cacheKey, allPRs)
|
||||
return allPRs, nil
|
||||
}
|
||||
|
||||
// GetTriageQueue returns unassigned issues and PRs needing review, sorted by priority.
|
||||
func (c *Client) GetTriageQueue(ctx context.Context, token string, orgs []string) ([]TriageItem, error) {
|
||||
issues, err := c.ListAllIssues(ctx, token, orgs)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("fetching issues for triage: %w", err)
|
||||
}
|
||||
|
||||
prs, err := c.ListAllPullRequests(ctx, token, orgs)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("fetching PRs for triage: %w", err)
|
||||
}
|
||||
|
||||
var queue []TriageItem
|
||||
|
||||
// Add unassigned issues.
|
||||
for _, issue := range issues {
|
||||
if issue.Assignee == nil && len(issue.Assignees) == 0 {
|
||||
var labels []string
|
||||
for _, l := range issue.Labels {
|
||||
labels = append(labels, l.Name)
|
||||
}
|
||||
queue = append(queue, TriageItem{
|
||||
Type: "issue",
|
||||
RepoOwner: issue.RepoOwner,
|
||||
RepoName: issue.RepoName,
|
||||
Number: issue.Number,
|
||||
Title: issue.Title,
|
||||
HTMLURL: issue.HTMLURL,
|
||||
Labels: labels,
|
||||
UpdatedAt: issue.UpdatedAt,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
// Add PRs (all open PRs may need review attention).
|
||||
for _, pr := range prs {
|
||||
var labels []string
|
||||
for _, l := range pr.Labels {
|
||||
labels = append(labels, l.Name)
|
||||
}
|
||||
queue = append(queue, TriageItem{
|
||||
Type: "pull",
|
||||
RepoOwner: pr.RepoOwner,
|
||||
RepoName: pr.RepoName,
|
||||
Number: pr.Number,
|
||||
Title: pr.Title,
|
||||
HTMLURL: pr.HTMLURL,
|
||||
Labels: labels,
|
||||
UpdatedAt: pr.UpdatedAt,
|
||||
})
|
||||
}
|
||||
|
||||
// Sort by priority labels (P1 > P2 > P3 > no priority), then by updated time.
|
||||
sort.Slice(queue, func(i, j int) bool {
|
||||
pi := priorityScore(queue[i].Labels)
|
||||
pj := priorityScore(queue[j].Labels)
|
||||
if pi != pj {
|
||||
return pi < pj // lower score = higher priority
|
||||
}
|
||||
return queue[i].UpdatedAt.After(queue[j].UpdatedAt)
|
||||
})
|
||||
|
||||
return queue, nil
|
||||
}
|
||||
|
||||
// CreateIssue creates a new issue in the specified repository.
|
||||
func (c *Client) CreateIssue(ctx context.Context, token, owner, repo, title, body string, labels []int64) (*Issue, error) {
|
||||
payload := map[string]interface{}{
|
||||
"title": title,
|
||||
"body": body,
|
||||
}
|
||||
if len(labels) > 0 {
|
||||
payload["labels"] = labels
|
||||
}
|
||||
|
||||
jsonData, err := json.Marshal(payload)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("marshaling issue: %w", err)
|
||||
}
|
||||
|
||||
path := fmt.Sprintf("/repos/%s/%s/issues", owner, repo)
|
||||
resp, err := c.doRequest(ctx, token, http.MethodPost, path, strings.NewReader(string(jsonData)))
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("creating issue: %w", err)
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
var issue Issue
|
||||
if err := json.NewDecoder(resp.Body).Decode(&issue); err != nil {
|
||||
return nil, fmt.Errorf("decoding created issue: %w", err)
|
||||
}
|
||||
|
||||
issue.RepoOwner = owner
|
||||
issue.RepoName = repo
|
||||
|
||||
c.InvalidateAll() // Invalidate cache after write.
|
||||
return &issue, nil
|
||||
}
|
||||
|
||||
// ApplyLabel adds a label to an issue.
|
||||
func (c *Client) ApplyLabel(ctx context.Context, token, owner, repo string, index int64, labelIDs []int64) error {
|
||||
payload := map[string]interface{}{
|
||||
"labels": labelIDs,
|
||||
}
|
||||
|
||||
jsonData, err := json.Marshal(payload)
|
||||
if err != nil {
|
||||
return fmt.Errorf("marshaling labels: %w", err)
|
||||
}
|
||||
|
||||
path := fmt.Sprintf("/repos/%s/%s/issues/%d/labels", owner, repo, index)
|
||||
resp, err := c.doRequest(ctx, token, http.MethodPost, path, strings.NewReader(string(jsonData)))
|
||||
if err != nil {
|
||||
return fmt.Errorf("applying labels: %w", err)
|
||||
}
|
||||
resp.Body.Close()
|
||||
|
||||
c.InvalidateAll()
|
||||
return nil
|
||||
}
|
||||
|
||||
// SubmitReview submits a review on a pull request.
|
||||
func (c *Client) SubmitReview(ctx context.Context, token, owner, repo string, index int64, reviewType, body string) error {
|
||||
payload := map[string]interface{}{
|
||||
"event": reviewType, // "APPROVED", "REQUEST_CHANGES", "COMMENT"
|
||||
"body": body,
|
||||
}
|
||||
|
||||
jsonData, err := json.Marshal(payload)
|
||||
if err != nil {
|
||||
return fmt.Errorf("marshaling review: %w", err)
|
||||
}
|
||||
|
||||
path := fmt.Sprintf("/repos/%s/%s/pulls/%d/reviews", owner, repo, index)
|
||||
resp, err := c.doRequest(ctx, token, http.MethodPost, path, strings.NewReader(string(jsonData)))
|
||||
if err != nil {
|
||||
return fmt.Errorf("submitting review: %w", err)
|
||||
}
|
||||
resp.Body.Close()
|
||||
|
||||
c.InvalidateAll()
|
||||
return nil
|
||||
}
|
||||
|
||||
// priorityScore returns a numeric score for sorting (lower = higher priority).
|
||||
func priorityScore(labels []string) int {
|
||||
for _, l := range labels {
|
||||
switch l {
|
||||
case "P1":
|
||||
return 1
|
||||
case "P2":
|
||||
return 2
|
||||
case "P3":
|
||||
return 3
|
||||
}
|
||||
}
|
||||
return 4 // no priority label
|
||||
}
|
||||
@@ -0,0 +1,215 @@
|
||||
package gitea
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestNewClient(t *testing.T) {
|
||||
c := NewClient("https://gitea.example.com")
|
||||
if c.baseURL != "https://gitea.example.com" {
|
||||
t.Errorf("baseURL = %q, want %q", c.baseURL, "https://gitea.example.com")
|
||||
}
|
||||
if c.maxConcurrent != 5 {
|
||||
t.Errorf("maxConcurrent = %d, want 5", c.maxConcurrent)
|
||||
}
|
||||
if c.cacheTTL != 30*time.Second {
|
||||
t.Errorf("cacheTTL = %v, want 30s", c.cacheTTL)
|
||||
}
|
||||
}
|
||||
|
||||
func TestNewClient_TrailingSlash(t *testing.T) {
|
||||
c := NewClient("https://gitea.example.com/")
|
||||
if c.baseURL != "https://gitea.example.com" {
|
||||
t.Errorf("baseURL = %q, want trailing slash removed", c.baseURL)
|
||||
}
|
||||
}
|
||||
|
||||
func TestCache(t *testing.T) {
|
||||
c := NewClient("https://gitea.example.com")
|
||||
|
||||
// Cache miss.
|
||||
_, ok := c.getFromCache("key1")
|
||||
if ok {
|
||||
t.Error("expected cache miss")
|
||||
}
|
||||
|
||||
// Cache set and hit.
|
||||
c.setCache("key1", "value1")
|
||||
val, ok := c.getFromCache("key1")
|
||||
if !ok {
|
||||
t.Fatal("expected cache hit")
|
||||
}
|
||||
if val.(string) != "value1" {
|
||||
t.Errorf("got %q, want %q", val, "value1")
|
||||
}
|
||||
|
||||
// Invalidate.
|
||||
c.invalidateCache("key")
|
||||
_, ok = c.getFromCache("key1")
|
||||
if ok {
|
||||
t.Error("expected cache miss after invalidation")
|
||||
}
|
||||
}
|
||||
|
||||
func TestCacheExpiry(t *testing.T) {
|
||||
c := NewClient("https://gitea.example.com")
|
||||
c.cacheTTL = 1 * time.Millisecond
|
||||
|
||||
c.setCache("key1", "value1")
|
||||
time.Sleep(5 * time.Millisecond)
|
||||
|
||||
_, ok := c.getFromCache("key1")
|
||||
if ok {
|
||||
t.Error("expected cache miss after TTL expiry")
|
||||
}
|
||||
}
|
||||
|
||||
func TestInvalidateAll(t *testing.T) {
|
||||
c := NewClient("https://gitea.example.com")
|
||||
c.setCache("key1", "value1")
|
||||
c.setCache("key2", "value2")
|
||||
|
||||
c.InvalidateAll()
|
||||
|
||||
_, ok1 := c.getFromCache("key1")
|
||||
_, ok2 := c.getFromCache("key2")
|
||||
if ok1 || ok2 {
|
||||
t.Error("expected all cache entries to be invalidated")
|
||||
}
|
||||
}
|
||||
|
||||
func TestPriorityScore(t *testing.T) {
|
||||
tests := []struct {
|
||||
labels []string
|
||||
want int
|
||||
}{
|
||||
{[]string{"P1", "bug"}, 1},
|
||||
{[]string{"P2"}, 2},
|
||||
{[]string{"P3", "enhancement"}, 3},
|
||||
{[]string{"bug", "enhancement"}, 4},
|
||||
{nil, 4},
|
||||
}
|
||||
|
||||
for _, tt := range tests {
|
||||
got := priorityScore(tt.labels)
|
||||
if got != tt.want {
|
||||
t.Errorf("priorityScore(%v) = %d, want %d", tt.labels, got, tt.want)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func TestListOrgs(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
if r.URL.Path != "/api/v1/user/orgs" {
|
||||
t.Errorf("unexpected path: %s", r.URL.Path)
|
||||
http.NotFound(w, r)
|
||||
return
|
||||
}
|
||||
if r.Header.Get("Authorization") != "token test-token" {
|
||||
t.Error("missing or wrong Authorization header")
|
||||
}
|
||||
|
||||
orgs := []Org{
|
||||
{Name: "org1", FullName: "Organization 1"},
|
||||
{Name: "org2", FullName: "Organization 2"},
|
||||
}
|
||||
json.NewEncoder(w).Encode(orgs)
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
c := NewClient(server.URL)
|
||||
orgs, err := c.ListOrgs(context.Background(), "test-token")
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if len(orgs) != 2 {
|
||||
t.Fatalf("got %d orgs, want 2", len(orgs))
|
||||
}
|
||||
if orgs[0].Name != "org1" {
|
||||
t.Errorf("orgs[0].Name = %q, want %q", orgs[0].Name, "org1")
|
||||
}
|
||||
}
|
||||
|
||||
func TestListOrgs_Cached(t *testing.T) {
|
||||
callCount := 0
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
callCount++
|
||||
json.NewEncoder(w).Encode([]Org{{Name: "org1"}})
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
c := NewClient(server.URL)
|
||||
|
||||
// First call should hit the server.
|
||||
_, err := c.ListOrgs(context.Background(), "test-token")
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
|
||||
// Second call should use cache.
|
||||
_, err = c.ListOrgs(context.Background(), "test-token")
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
|
||||
if callCount != 1 {
|
||||
t.Errorf("server called %d times, want 1 (cached)", callCount)
|
||||
}
|
||||
}
|
||||
|
||||
func TestListOrgRepos(t *testing.T) {
|
||||
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
repos := []Repo{
|
||||
{ID: 1, Name: "repo1", FullName: "org1/repo1"},
|
||||
{ID: 2, Name: "repo2", FullName: "org1/repo2"},
|
||||
}
|
||||
json.NewEncoder(w).Encode(repos)
|
||||
}))
|
||||
defer server.Close()
|
||||
|
||||
c := NewClient(server.URL)
|
||||
repos, err := c.ListOrgRepos(context.Background(), "test-token", "org1")
|
||||
if err != nil {
|
||||
t.Fatalf("unexpected error: %v", err)
|
||||
}
|
||||
if len(repos) != 2 {
|
||||
t.Fatalf("got %d repos, want 2", len(repos))
|
||||
}
|
||||
}
|
||||
|
||||
func TestGetTriageQueue_Sorting(t *testing.T) {
|
||||
queue := []TriageItem{
|
||||
{Title: "low", Labels: []string{"P3"}, UpdatedAt: time.Now()},
|
||||
{Title: "high", Labels: []string{"P1"}, UpdatedAt: time.Now()},
|
||||
{Title: "medium", Labels: []string{"P2"}, UpdatedAt: time.Now()},
|
||||
{Title: "none", Labels: nil, UpdatedAt: time.Now()},
|
||||
}
|
||||
|
||||
// Apply the same sort as GetTriageQueue.
|
||||
sortTriageQueue(queue)
|
||||
|
||||
expected := []string{"high", "medium", "low", "none"}
|
||||
for i, item := range queue {
|
||||
if item.Title != expected[i] {
|
||||
t.Errorf("queue[%d].Title = %q, want %q", i, item.Title, expected[i])
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// sortTriageQueue is a test helper applying the same sort as GetTriageQueue.
|
||||
func sortTriageQueue(queue []TriageItem) {
|
||||
for i := 0; i < len(queue); i++ {
|
||||
for j := i + 1; j < len(queue); j++ {
|
||||
pi := priorityScore(queue[i].Labels)
|
||||
pj := priorityScore(queue[j].Labels)
|
||||
if pj < pi || (pj == pi && queue[j].UpdatedAt.After(queue[i].UpdatedAt)) {
|
||||
queue[i], queue[j] = queue[j], queue[i]
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,177 @@
|
||||
package handlers
|
||||
|
||||
import (
|
||||
"html/template"
|
||||
"net/http"
|
||||
"strings"
|
||||
|
||||
"gitea.leeworks.dev/0xwheatyz/gitea-mobile/internal/auth"
|
||||
"gitea.leeworks.dev/0xwheatyz/gitea-mobile/internal/middleware"
|
||||
)
|
||||
|
||||
var settingsTemplate = template.Must(template.New("settings").Parse(`<!DOCTYPE html>
|
||||
<html lang="en">
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0, viewport-fit=cover">
|
||||
<title>Settings — Gitea Mobile</title>
|
||||
<style>
|
||||
* { box-sizing: border-box; margin: 0; padding: 0; }
|
||||
body {
|
||||
font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, sans-serif;
|
||||
background: #0d1117; color: #e6edf3;
|
||||
padding: 1rem;
|
||||
padding-top: max(1rem, env(safe-area-inset-top));
|
||||
}
|
||||
h1 { font-size: 1.5rem; margin-bottom: 1rem; }
|
||||
.card {
|
||||
background: #161b22; border: 1px solid #30363d; border-radius: 8px;
|
||||
padding: 1rem; margin-bottom: 1rem;
|
||||
}
|
||||
label { display: block; font-size: 0.875rem; color: #8b949e; margin-bottom: 0.5rem; }
|
||||
input[type="text"], input[type="password"] {
|
||||
width: 100%; padding: 0.5rem; font-size: 1rem;
|
||||
background: #0d1117; border: 1px solid #30363d; border-radius: 6px;
|
||||
color: #e6edf3; margin-bottom: 1rem;
|
||||
}
|
||||
input:focus { outline: none; border-color: #58a6ff; }
|
||||
button {
|
||||
width: 100%; padding: 0.75rem; font-size: 1rem; font-weight: 600;
|
||||
background: #238636; color: #fff; border: none; border-radius: 6px;
|
||||
cursor: pointer;
|
||||
}
|
||||
button:active { background: #2ea043; }
|
||||
.message {
|
||||
padding: 0.75rem; border-radius: 6px; margin-bottom: 1rem;
|
||||
font-size: 0.875rem;
|
||||
}
|
||||
.message.success { background: #0d2818; border: 1px solid #238636; color: #3fb950; }
|
||||
.message.error { background: #2d1117; border: 1px solid #da3633; color: #f85149; }
|
||||
.message.info { background: #0c1d2e; border: 1px solid #1f6feb; color: #58a6ff; }
|
||||
.hint { font-size: 0.75rem; color: #8b949e; margin-top: 0.25rem; margin-bottom: 1rem; }
|
||||
.status { font-size: 0.875rem; color: #8b949e; }
|
||||
.status .connected { color: #3fb950; }
|
||||
.logout-btn {
|
||||
background: #21262d; border: 1px solid #30363d; margin-top: 0.5rem;
|
||||
}
|
||||
.logout-btn:active { background: #30363d; }
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<h1>Settings</h1>
|
||||
|
||||
{{if .Message}}
|
||||
<div class="message {{.MessageType}}">{{.Message}}</div>
|
||||
{{end}}
|
||||
|
||||
{{if .HasToken}}
|
||||
<div class="card">
|
||||
<p class="status">Status: <span class="connected">Connected</span></p>
|
||||
<p class="hint">A Gitea API token is configured.</p>
|
||||
<form method="POST" action="/settings">
|
||||
<input type="hidden" name="action" value="logout">
|
||||
<button type="submit" class="logout-btn">Remove Token</button>
|
||||
</form>
|
||||
</div>
|
||||
{{end}}
|
||||
|
||||
<div class="card">
|
||||
<form method="POST" action="/settings">
|
||||
<input type="hidden" name="action" value="save">
|
||||
<label for="token">Gitea API Token</label>
|
||||
<input type="password" id="token" name="token" placeholder="Enter your Gitea API token" required>
|
||||
<p class="hint">Generate a token at your Gitea instance under Settings → Applications.</p>
|
||||
<button type="submit">{{if .HasToken}}Update Token{{else}}Save Token{{end}}</button>
|
||||
</form>
|
||||
</div>
|
||||
|
||||
{{if .HasToken}}
|
||||
<p style="text-align:center; margin-top:1rem;">
|
||||
<a href="/" style="color:#58a6ff; text-decoration:none;">Back to Dashboard</a>
|
||||
</p>
|
||||
{{end}}
|
||||
</body>
|
||||
</html>`))
|
||||
|
||||
// SettingsHandler handles GET and POST requests for the settings page.
|
||||
type SettingsHandler struct {
|
||||
SessionSecret string
|
||||
SecureCookies bool
|
||||
}
|
||||
|
||||
type settingsData struct {
|
||||
HasToken bool
|
||||
Message string
|
||||
MessageType string // "success", "error", "info"
|
||||
}
|
||||
|
||||
// ServeHTTP handles the settings page.
|
||||
func (h *SettingsHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
|
||||
switch r.Method {
|
||||
case http.MethodGet:
|
||||
h.handleGet(w, r)
|
||||
case http.MethodPost:
|
||||
h.handlePost(w, r)
|
||||
default:
|
||||
http.Error(w, "method not allowed", http.StatusMethodNotAllowed)
|
||||
}
|
||||
}
|
||||
|
||||
func (h *SettingsHandler) handleGet(w http.ResponseWriter, r *http.Request) {
|
||||
hasToken := false
|
||||
if token := middleware.TokenFromContext(r.Context()); token != "" {
|
||||
hasToken = true
|
||||
} else if _, err := auth.GetToken(r, h.SessionSecret); err == nil {
|
||||
hasToken = true
|
||||
}
|
||||
|
||||
data := settingsData{HasToken: hasToken}
|
||||
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
||||
settingsTemplate.Execute(w, data)
|
||||
}
|
||||
|
||||
func (h *SettingsHandler) handlePost(w http.ResponseWriter, r *http.Request) {
|
||||
if err := r.ParseForm(); err != nil {
|
||||
h.renderWithMessage(w, r, "Failed to parse form.", "error")
|
||||
return
|
||||
}
|
||||
|
||||
action := r.FormValue("action")
|
||||
|
||||
switch action {
|
||||
case "logout":
|
||||
auth.ClearTokenCookie(w, h.SecureCookies)
|
||||
h.renderWithMessage(w, r, "Token removed successfully.", "success")
|
||||
return
|
||||
|
||||
case "save":
|
||||
token := strings.TrimSpace(r.FormValue("token"))
|
||||
if token == "" {
|
||||
h.renderWithMessage(w, r, "Token cannot be empty.", "error")
|
||||
return
|
||||
}
|
||||
|
||||
auth.SetTokenCookie(w, token, h.SessionSecret, h.SecureCookies)
|
||||
// After saving, redirect to dashboard.
|
||||
http.Redirect(w, r, "/", http.StatusSeeOther)
|
||||
return
|
||||
|
||||
default:
|
||||
h.renderWithMessage(w, r, "Unknown action.", "error")
|
||||
}
|
||||
}
|
||||
|
||||
func (h *SettingsHandler) renderWithMessage(w http.ResponseWriter, r *http.Request, msg, msgType string) {
|
||||
hasToken := false
|
||||
if _, err := auth.GetToken(r, h.SessionSecret); err == nil {
|
||||
hasToken = true
|
||||
}
|
||||
|
||||
data := settingsData{
|
||||
HasToken: hasToken,
|
||||
Message: msg,
|
||||
MessageType: msgType,
|
||||
}
|
||||
w.Header().Set("Content-Type", "text/html; charset=utf-8")
|
||||
settingsTemplate.Execute(w, data)
|
||||
}
|
||||
@@ -0,0 +1,54 @@
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"context"
|
||||
"log/slog"
|
||||
"net/http"
|
||||
|
||||
"gitea.leeworks.dev/0xwheatyz/gitea-mobile/internal/auth"
|
||||
)
|
||||
|
||||
// contextKey is a private type for context keys in this package.
|
||||
type contextKey string
|
||||
|
||||
const (
|
||||
// TokenContextKey is the context key for the Gitea API token.
|
||||
TokenContextKey contextKey = "gitea_token"
|
||||
)
|
||||
|
||||
// TokenFromContext extracts the Gitea API token from the request context.
|
||||
func TokenFromContext(ctx context.Context) string {
|
||||
token, _ := ctx.Value(TokenContextKey).(string)
|
||||
return token
|
||||
}
|
||||
|
||||
// Auth returns middleware that checks for a valid token cookie.
|
||||
// Unauthenticated requests are redirected to the settings page.
|
||||
// The /health, /settings, and /static/ paths are exempt from auth.
|
||||
func Auth(sessionSecret string) func(http.Handler) http.Handler {
|
||||
return func(next http.Handler) http.Handler {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
// Skip auth for exempt paths.
|
||||
path := r.URL.Path
|
||||
if path == "/health" || path == "/settings" || hasPrefix(path, "/static/") {
|
||||
next.ServeHTTP(w, r)
|
||||
return
|
||||
}
|
||||
|
||||
token, err := auth.GetToken(r, sessionSecret)
|
||||
if err != nil || token == "" {
|
||||
slog.Debug("unauthenticated request, redirecting to settings", "path", path, "error", err)
|
||||
http.Redirect(w, r, "/settings", http.StatusSeeOther)
|
||||
return
|
||||
}
|
||||
|
||||
// Inject token into request context.
|
||||
ctx := context.WithValue(r.Context(), TokenContextKey, token)
|
||||
next.ServeHTTP(w, r.WithContext(ctx))
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
func hasPrefix(s, prefix string) bool {
|
||||
return len(s) >= len(prefix) && s[:len(prefix)] == prefix
|
||||
}
|
||||
@@ -0,0 +1,85 @@
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"net/http/httptest"
|
||||
"testing"
|
||||
|
||||
"gitea.leeworks.dev/0xwheatyz/gitea-mobile/internal/auth"
|
||||
)
|
||||
|
||||
const testSecret = "test-secret-that-is-at-least-32-chars-long"
|
||||
|
||||
func TestAuth_HealthBypass(t *testing.T) {
|
||||
handler := Auth(testSecret)(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
}))
|
||||
|
||||
req := httptest.NewRequest(http.MethodGet, "/health", nil)
|
||||
w := httptest.NewRecorder()
|
||||
handler.ServeHTTP(w, req)
|
||||
|
||||
if w.Code != http.StatusOK {
|
||||
t.Errorf("status = %d, want %d", w.Code, http.StatusOK)
|
||||
}
|
||||
}
|
||||
|
||||
func TestAuth_SettingsBypass(t *testing.T) {
|
||||
handler := Auth(testSecret)(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
}))
|
||||
|
||||
req := httptest.NewRequest(http.MethodGet, "/settings", nil)
|
||||
w := httptest.NewRecorder()
|
||||
handler.ServeHTTP(w, req)
|
||||
|
||||
if w.Code != http.StatusOK {
|
||||
t.Errorf("status = %d, want %d", w.Code, http.StatusOK)
|
||||
}
|
||||
}
|
||||
|
||||
func TestAuth_RedirectWithoutToken(t *testing.T) {
|
||||
handler := Auth(testSecret)(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.WriteHeader(http.StatusOK)
|
||||
}))
|
||||
|
||||
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
||||
w := httptest.NewRecorder()
|
||||
handler.ServeHTTP(w, req)
|
||||
|
||||
if w.Code != http.StatusSeeOther {
|
||||
t.Errorf("status = %d, want %d", w.Code, http.StatusSeeOther)
|
||||
}
|
||||
if loc := w.Header().Get("Location"); loc != "/settings" {
|
||||
t.Errorf("Location = %q, want %q", loc, "/settings")
|
||||
}
|
||||
}
|
||||
|
||||
func TestAuth_PassWithToken(t *testing.T) {
|
||||
called := false
|
||||
handler := Auth(testSecret)(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
called = true
|
||||
token := TokenFromContext(r.Context())
|
||||
if token != "my-token" {
|
||||
t.Errorf("token = %q, want %q", token, "my-token")
|
||||
}
|
||||
w.WriteHeader(http.StatusOK)
|
||||
}))
|
||||
|
||||
// Set a token cookie.
|
||||
cookieW := httptest.NewRecorder()
|
||||
auth.SetTokenCookie(cookieW, "my-token", testSecret, false)
|
||||
cookie := cookieW.Result().Cookies()[0]
|
||||
|
||||
req := httptest.NewRequest(http.MethodGet, "/", nil)
|
||||
req.AddCookie(cookie)
|
||||
w := httptest.NewRecorder()
|
||||
handler.ServeHTTP(w, req)
|
||||
|
||||
if !called {
|
||||
t.Error("next handler was not called")
|
||||
}
|
||||
if w.Code != http.StatusOK {
|
||||
t.Errorf("status = %d, want %d", w.Code, http.StatusOK)
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,38 @@
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"log/slog"
|
||||
"net/http"
|
||||
"time"
|
||||
)
|
||||
|
||||
// responseWriter wraps http.ResponseWriter to capture the status code.
|
||||
type responseWriter struct {
|
||||
http.ResponseWriter
|
||||
statusCode int
|
||||
}
|
||||
|
||||
func (rw *responseWriter) WriteHeader(code int) {
|
||||
rw.statusCode = code
|
||||
rw.ResponseWriter.WriteHeader(code)
|
||||
}
|
||||
|
||||
// Logging returns middleware that logs each HTTP request with structured logging.
|
||||
func Logging() func(http.Handler) http.Handler {
|
||||
return func(next http.Handler) http.Handler {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
start := time.Now()
|
||||
rw := &responseWriter{ResponseWriter: w, statusCode: http.StatusOK}
|
||||
|
||||
next.ServeHTTP(rw, r)
|
||||
|
||||
slog.Info("http request",
|
||||
"method", r.Method,
|
||||
"path", r.URL.Path,
|
||||
"status", rw.statusCode,
|
||||
"duration", time.Since(start).String(),
|
||||
"remote", r.RemoteAddr,
|
||||
)
|
||||
})
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user