147 lines
3.2 KiB
Go
147 lines
3.2 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"fmt"
|
|
"regexp"
|
|
"strings"
|
|
"net/http"
|
|
"io/ioutil"
|
|
)
|
|
|
|
import (
|
|
"github.com/BurntSushi/toml"
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/dustin/go-humanize"
|
|
)
|
|
|
|
type Resource struct {
|
|
Url string `toml:"url"`
|
|
Proxied bool `toml:"proxied",default:false`
|
|
Mime string `toml:"mime"`
|
|
}
|
|
func (self Resource) Get() ([]byte, error) {
|
|
return ioutil.ReadFile(self.Url[7:])
|
|
}
|
|
|
|
type ResourceDConfig struct {
|
|
Enabled bool `toml:"enabled"`
|
|
ListenURL string `toml:"listen_url"`
|
|
ProxyCacheMinSize string `toml:"proxy_cache_min_size",default:5MB`
|
|
}
|
|
|
|
type Config struct {
|
|
ResourceD ResourceDConfig `toml:"resourceD"`
|
|
Resource map[string]Resource `toml:"resource"`
|
|
}
|
|
|
|
func (self Config) Validate() int {
|
|
re, err := regexp.Compile(`^(file|http(s|))://`)
|
|
if err != nil { panic(err) }
|
|
|
|
for key, res := range self.Resource {
|
|
if ! re.MatchString(res.Url) {
|
|
panic(fmt.Sprintf("Resource %s has invalid URL: %s\nOnly file://, http:// and https:// URLs are allowed", key, res.Url))
|
|
}
|
|
}
|
|
|
|
return 0
|
|
}
|
|
func (self Resource) GetProxied() ([]byte, error) {
|
|
|
|
cached, exists := ProxyResourceCache[self.Url];
|
|
if exists {
|
|
return cached, nil
|
|
}
|
|
|
|
resp, err := http.Get(self.Url)
|
|
if err != nil { return make([]byte, 0, 0), err }
|
|
|
|
buf, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil { return make([]byte, 0, 0), err }
|
|
|
|
// cache only those that are less than 5 mb
|
|
if len(buf) > ProxyCacheMinSize {
|
|
ProxyResourceCache[self.Url] = buf
|
|
}
|
|
|
|
return buf, nil
|
|
}
|
|
|
|
var ProxyResourceCache map[string][]byte = make(map[string][]byte)
|
|
var ProxyCacheMinSize int
|
|
|
|
func main() {
|
|
var conf Config
|
|
|
|
data, err := ioutil.ReadFile("resourced.toml")
|
|
if err != nil { panic(err) }
|
|
|
|
a, err := toml.Decode(string(data), &conf)
|
|
if err != nil { panic(err) }
|
|
_ = a
|
|
|
|
cache_min, err := humanize.ParseBytes(conf.ResourceD.ProxyCacheMinSize)
|
|
if err != nil { panic(err) }
|
|
ProxyCacheMinSize = int(cache_min)
|
|
|
|
conf.Validate()
|
|
|
|
app := fiber.New(fiber.Config {
|
|
Prefork: true,
|
|
CaseSensitive: false,
|
|
StrictRouting: true,
|
|
ServerHeader: "",
|
|
AppName: "blek! File resourceD",
|
|
})
|
|
|
|
app.Get("/info/is_enabled", func (c *fiber.Ctx) error {
|
|
return c.JSON(conf.ResourceD.Enabled)
|
|
})
|
|
|
|
app.Use(func (c *fiber.Ctx) error {
|
|
if ! conf.ResourceD.Enabled {
|
|
return c.Status(fiber.StatusNotFound).SendString("ResourceD is disabled")
|
|
}
|
|
return c.Next()
|
|
})
|
|
|
|
app.Get("/:id", func (c *fiber.Ctx) error {
|
|
res, exists := conf.Resource[c.Params("id")]
|
|
if ! exists {
|
|
return c.Status(fiber.StatusNotFound).SendString("Resource not found")
|
|
}
|
|
|
|
if ! strings.HasPrefix(res.Url, "file://") {
|
|
|
|
if res.Proxied {
|
|
data, err := res.GetProxied()
|
|
if err != nil {
|
|
log.Fatalln(err)
|
|
// we failed, send a redirect instead
|
|
// (next executed line would be 115)
|
|
} else {
|
|
c.Response().Header.SetContentType(res.Mime)
|
|
c.Response().Header.SetContentLength(len(data))
|
|
return c.Send(data)
|
|
}
|
|
}
|
|
|
|
c.Location(res.Url)
|
|
c.Status(302)
|
|
return nil
|
|
}
|
|
|
|
data, err := res.Get()
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
c.Response().Header.SetContentType(res.Mime)
|
|
c.Response().Header.SetContentLength(len(data))
|
|
|
|
return c.Send(data)
|
|
})
|
|
|
|
log.Fatal(app.Listen(conf.ResourceD.ListenURL))
|
|
} |