homepage: refactor and fix overrides not being applied

internal/api/v1/favicon/favicon.go

@@ -86,8 +86,8 @@ func GetFavIcon(w http.ResponseWriter, req *http.Request) {
 	}
 
 	var result *fetchResult
-	hp := r.HomepageConfig()
-	if !hp.IsEmpty() && hp.Icon != nil {
+	hp := r.HomepageItem()
+	if hp.Icon != nil {
 		if hp.Icon.IconSource == homepage.IconSourceRelative {
 			result = findIcon(r, req, hp.Icon.Value)
 		} else {

internal/homepage/homepage.go

@@ -1,13 +1,15 @@
 package homepage
 
 import (
+	"encoding/json"
+
+	config "github.com/yusing/go-proxy/internal/config/types"
 	"github.com/yusing/go-proxy/internal/utils"
 )
 
 type (
-	//nolint:recvcheck
-	Categories map[string]Category
-	Category   []*Item
+	Homepage map[string]Category
+	Category []*Item
 
 	ItemConfig struct {
 		Show         bool           `json:"show"`
@@ -22,11 +24,8 @@ type (
 	Item struct {
 		*ItemConfig
 
-		Alias    string `json:"alias"` // proxy alias
-		Provider string `json:"provider"`
-		URL      string `json:"url"` // alias + domain
-
-		IsUnset bool `json:"-"`
+		Alias    string
+		Provider string
 	}
 )
 
@@ -38,33 +37,34 @@ func init() {
 	})
 }
 
-func NewItem(alias string) *Item {
-	return &Item{
-		ItemConfig: &ItemConfig{
-			Show: true,
-		},
-		Alias:   alias,
-		IsUnset: true,
-	}
-}
-
-func NewHomePageConfig() Categories {
-	return Categories(make(map[string]Category))
+func (cfg *ItemConfig) GetOverride(alias string) *ItemConfig {
+	return overrideConfigInstance.GetOverride(alias, cfg)
 }
 
-func (item *Item) IsEmpty() bool {
-	return item == nil || item.IsUnset || item.ItemConfig == nil
-}
-
-func (item *Item) ApplyOverride() *Item {
-	return overrideConfigInstance.ApplyOverride(item)
-}
-
-func (c *Categories) Clear() {
-	*c = make(Categories)
+func (item *Item) MarshalJSON() ([]byte, error) {
+	godoxyCfg := config.GetInstance().Value()
+	// use first domain as base domain
+	domains := godoxyCfg.MatchDomains
+	var url *string
+	if len(domains) > 0 {
+		url = new(string)
+		*url = item.Alias + domains[0]
+	}
+	return json.Marshal(map[string]any{
+		"show":          item.Show,
+		"alias":         item.Alias,
+		"provider":      item.Provider,
+		"url":           url,
+		"name":          item.Name,
+		"icon":          item.Icon,
+		"category":      item.Category,
+		"description":   item.Description,
+		"sort_order":    item.SortOrder,
+		"widget_config": item.WidgetConfig,
+	})
 }
 
-func (c Categories) Add(item *Item) {
+func (c Homepage) Add(item *Item) {
 	if c[item.Category] == nil {
 		c[item.Category] = make(Category, 0)
 	}

internal/homepage/homepage_test.go

@@ -20,7 +20,7 @@ func TestOverrideItem(t *testing.T) {
 			Category: "App",
 		},
 	}
-	override := &ItemConfig{
+	want := &ItemConfig{
 		Show:     true,
 		Name:     "Bar",
 		Category: "Test",
@@ -30,7 +30,7 @@ func TestOverrideItem(t *testing.T) {
 		},
 	}
 	overrides := GetOverrideConfig()
-	overrides.OverrideItem(a.Alias, override)
-	overridden := a.ApplyOverride()
-	ExpectDeepEqual(t, overridden.ItemConfig, override)
+	overrides.OverrideItem(a.Alias, want)
+	got := a.GetOverride(a.Alias)
+	ExpectDeepEqual(t, got, want)
 }

internal/homepage/override_config.go

@@ -64,16 +64,17 @@ func (c *OverrideConfig) OverrideItems(items map[string]*ItemConfig) {
 	}
 }
 
-func (c *OverrideConfig) ApplyOverride(item *Item) *Item {
+func (c *OverrideConfig) GetOverride(alias string, item *ItemConfig) *ItemConfig {
 	c.mu.RLock()
 	defer c.mu.RUnlock()
-	itemOverride, hasOverride := c.ItemOverrides[item.Alias]
+	itemOverride, hasOverride := c.ItemOverrides[alias]
 	if hasOverride {
-		item.ItemConfig = itemOverride
-		item.IsUnset = false
+		return itemOverride
 	}
-	if show, ok := c.ItemVisibility[item.Alias]; ok {
-		item.Show = show
+	if show, ok := c.ItemVisibility[alias]; ok {
+		clone := *item
+		clone.Show = show
+		return &clone
 	}
 	return item
 }

internal/route/reverse_proxy.go

@@ -225,7 +225,7 @@ func (r *ReveseProxyRoute) addToLoadBalancer(parent task.Parent) {
 		linked = l.(*ReveseProxyRoute)
 		lb = linked.loadBalancer
 		lb.UpdateConfigIfNeeded(cfg)
-		if linked.Homepage.IsEmpty() && !r.Homepage.IsEmpty() {
+		if linked.Homepage == nil {
 			linked.Homepage = r.Homepage
 		}
 	} else {

internal/route/route.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"strings"
 
+	"github.com/docker/docker/api/types/container"
 	"github.com/yusing/go-proxy/agent/pkg/agent"
 	"github.com/yusing/go-proxy/internal"
 	"github.com/yusing/go-proxy/internal/docker"
@@ -15,7 +16,6 @@ import (
 	"github.com/yusing/go-proxy/internal/utils/strutils"
 	"github.com/yusing/go-proxy/internal/watcher/health"
 
-	dockertypes "github.com/docker/docker/api/types"
 	"github.com/yusing/go-proxy/internal/common"
 	config "github.com/yusing/go-proxy/internal/config/types"
 	"github.com/yusing/go-proxy/internal/net/gphttp/accesslog"
@@ -41,7 +41,7 @@ type (
 		HealthCheck  *health.HealthCheckConfig  `json:"healthcheck,omitempty"`
 		LoadBalance  *loadbalance.Config        `json:"load_balance,omitempty"`
 		Middlewares  map[string]docker.LabelMap `json:"middlewares,omitempty"`
-		Homepage     *homepage.Item             `json:"homepage,omitempty"`
+		Homepage     *homepage.ItemConfig       `json:"homepage,omitempty"`
 		AccessLog    *accesslog.Config          `json:"access_log,omitempty"`
 
 		Metadata `deserialize:"-"`
@@ -190,8 +190,16 @@ func (r *Route) LoadBalanceConfig() *loadbalance.Config {
 	return r.LoadBalance
 }
 
-func (r *Route) HomepageConfig() *homepage.Item {
-	return r.Homepage
+func (r *Route) HomepageConfig() *homepage.ItemConfig {
+	return r.Homepage.GetOverride(r.Alias)
+}
+
+func (r *Route) HomepageItem() *homepage.Item {
+	return &homepage.Item{
+		Alias:      r.Alias,
+		Provider:   r.Provider,
+		ItemConfig: r.HomepageConfig(),
+	}
 }
 
 func (r *Route) ContainerInfo() *docker.Container {
@@ -370,14 +378,12 @@ func (r *Route) FinalizeHomepageConfig() {
 
 	isDocker := r.Container != nil
 
-	if r.Homepage.IsEmpty() {
-		r.Homepage = homepage.NewItem(r.Alias)
+	if r.Homepage == nil {
+		r.Homepage = &homepage.ItemConfig{Show: true}
 	}
+	r.Homepage = r.Homepage.GetOverride(r.Alias)
 
 	hp := r.Homepage
-	hp.Alias = r.Alias
-	hp.Provider = r.Provider
-	hp = hp.ApplyOverride()
 
 	var key string
 	if hp.Name == "" {
@@ -424,7 +430,7 @@ func (r *Route) FinalizeHomepageConfig() {
 	}
 }
 
-func lowestPort(ports map[int]dockertypes.Port) (res int) {
+func lowestPort(ports map[int]container.Port) (res int) {
 	cmp := (uint16)(65535)
 	for port, v := range ports {
 		if v.PrivatePort < cmp {

internal/route/routes/routequery/query.go

@@ -64,33 +64,33 @@ func HomepageCategories() []string {
 	check := make(map[string]struct{})
 	categories := make([]string, 0)
 	routes.GetHTTPRoutes().RangeAll(func(alias string, r route.HTTPRoute) {
-		homepage := r.HomepageConfig()
-		if homepage.IsEmpty() || homepage.Category == "" {
+		item := r.HomepageConfig()
+		if item == nil || item.Category == "" {
 			return
 		}
-		if _, ok := check[homepage.Category]; ok {
+		if _, ok := check[item.Category]; ok {
 			return
 		}
-		check[homepage.Category] = struct{}{}
-		categories = append(categories, homepage.Category)
+		check[item.Category] = struct{}{}
+		categories = append(categories, item.Category)
 	})
 	return categories
 }
 
-func HomepageConfig(categoryFilter, providerFilter string) homepage.Categories {
-	hpCfg := homepage.NewHomePageConfig()
+func HomepageConfig(categoryFilter, providerFilter string) homepage.Homepage {
+	hp := make(homepage.Homepage)
 
 	routes.GetHTTPRoutes().RangeAll(func(alias string, r route.HTTPRoute) {
-		item := r.HomepageConfig()
-		if providerFilter != "" && item.Provider != providerFilter {
+		if providerFilter != "" && r.ProviderName() != providerFilter {
 			return
 		}
+		item := r.HomepageItem()
 		if categoryFilter != "" && item.Category != categoryFilter {
 			return
 		}
-		hpCfg.Add(item)
+		hp.Add(item)
 	})
-	return hpCfg
+	return hp
 }
 
 func RoutesByAlias(typeFilter ...route.RouteType) map[string]route.Route {

internal/route/types/route.go

@@ -29,7 +29,8 @@ type (
 		IdlewatcherConfig() *idlewatcher.Config
 		HealthCheckConfig() *health.HealthCheckConfig
 		LoadBalanceConfig() *loadbalance.Config
-		HomepageConfig() *homepage.Item
+		HomepageConfig() *homepage.ItemConfig
+		HomepageItem() *homepage.Item
 		ContainerInfo() *docker.Container
 
 		Agent() *agent.AgentConfig