mirror of
https://github.com/yusing/godoxy.git
synced 2025-05-20 04:42:33 +02:00
1a5f3735cf
* cleanup code for URL type * fix makefile for trace mode * refactor, merge Entry, RawEntry and Route into one. * Implement fileserver. * refactor: rename HTTPRoute to ReverseProxyRoute to avoid confusion * refactor: move metrics logger to middleware package - fix prometheus metrics for load balanced routes - route will now fail when health monitor fail to start * fix extra output of ls-* commands by defer initializaing stuff, speed up start time * add test for path traversal attack, small fix on FileServer.Start method * rename rule.on.bypass to pass * refactor and fixed map-to-map deserialization * updated route loading logic * schemas: add "add_prefix" option to modify_request middleware * updated route JSONMarshalling --------- Co-authored-by: yusing <yusing@6uo.me>
107 lines
2.7 KiB
Go
107 lines
2.7 KiB
Go
package homepage
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/yusing/go-proxy/internal/common"
|
|
"github.com/yusing/go-proxy/internal/logging"
|
|
"github.com/yusing/go-proxy/internal/task"
|
|
"github.com/yusing/go-proxy/internal/utils"
|
|
)
|
|
|
|
type OverrideConfig struct {
|
|
ItemOverrides map[string]*ItemConfig `json:"item_overrides"`
|
|
DisplayOrder map[string]int `json:"display_order"` // TODO: implement this
|
|
CategoryOrder map[string]int `json:"category_order"` // TODO: implement this
|
|
ItemVisibility map[string]bool `json:"item_visibility"`
|
|
mu sync.RWMutex
|
|
}
|
|
|
|
var overrideConfigInstance = &OverrideConfig{
|
|
ItemOverrides: make(map[string]*ItemConfig),
|
|
DisplayOrder: make(map[string]int),
|
|
CategoryOrder: make(map[string]int),
|
|
ItemVisibility: make(map[string]bool),
|
|
}
|
|
|
|
func InitOverridesConfig() {
|
|
overrideConfigInstance.mu.Lock()
|
|
defer overrideConfigInstance.mu.Unlock()
|
|
|
|
err := utils.LoadJSONIfExist(common.HomepageJSONConfigPath, overrideConfigInstance)
|
|
if err != nil {
|
|
logging.Error().Err(err).Msg("failed to load homepage overrides config")
|
|
} else {
|
|
logging.Info().Msgf("homepage overrides config loaded, %d items", len(overrideConfigInstance.ItemOverrides))
|
|
}
|
|
task.OnProgramExit("save_homepage_json_config", func() {
|
|
if len(overrideConfigInstance.ItemOverrides) == 0 {
|
|
return
|
|
}
|
|
if err := utils.SaveJSON(common.HomepageJSONConfigPath, overrideConfigInstance, 0o644); err != nil {
|
|
logging.Error().Err(err).Msg("failed to save homepage overrides config")
|
|
}
|
|
})
|
|
}
|
|
|
|
func GetOverrideConfig() *OverrideConfig {
|
|
return overrideConfigInstance
|
|
}
|
|
|
|
func (c *OverrideConfig) OverrideItem(alias string, override *ItemConfig) {
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
c.ItemOverrides[alias] = override
|
|
}
|
|
|
|
func (c *OverrideConfig) OverrideItems(items map[string]*ItemConfig) {
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
for key, value := range items {
|
|
c.ItemOverrides[key] = value
|
|
}
|
|
}
|
|
|
|
func (c *OverrideConfig) GetOverride(item *Item) *Item {
|
|
c.mu.RLock()
|
|
defer c.mu.RUnlock()
|
|
itemOverride, hasOverride := c.ItemOverrides[item.Alias]
|
|
if hasOverride {
|
|
clone := *item
|
|
clone.ItemConfig = itemOverride
|
|
clone.IsUnset = false
|
|
item = &clone
|
|
}
|
|
if show, ok := c.ItemVisibility[item.Alias]; ok {
|
|
if !hasOverride {
|
|
clone := *item
|
|
clone.Show = show
|
|
item = &clone
|
|
} else {
|
|
item.Show = show
|
|
}
|
|
}
|
|
return item
|
|
}
|
|
|
|
func (c *OverrideConfig) SetCategoryOrder(key string, value int) {
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
c.CategoryOrder[key] = value
|
|
}
|
|
|
|
func (c *OverrideConfig) UnhideItems(keys ...string) {
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
for _, key := range keys {
|
|
c.ItemVisibility[key] = true
|
|
}
|
|
}
|
|
|
|
func (c *OverrideConfig) HideItems(keys ...string) {
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
for _, key := range keys {
|
|
c.ItemVisibility[key] = false
|
|
}
|
|
}
|