summaryrefslogtreecommitdiff
path: root/service/static/service.go
diff options
context:
space:
mode:
Diffstat (limited to 'service/static/service.go')
-rw-r--r--service/static/service.go120
1 files changed, 120 insertions, 0 deletions
diff --git a/service/static/service.go b/service/static/service.go
new file mode 100644
index 00000000..916c18a2
--- /dev/null
+++ b/service/static/service.go
@@ -0,0 +1,120 @@
+package static
+
+import (
+ "github.com/sirupsen/logrus"
+ "net/http"
+ "os"
+ "path"
+ "strings"
+ rrttp "github.com/spiral/roadrunner/service/http"
+ "github.com/spiral/roadrunner/service"
+)
+
+// Name contains default service name.
+const Name = "static-server"
+
+// Service serves static files. Potentially convert into middleware?
+type Service struct {
+ // Logger is associated debug and error logger. Can be empty.
+ Logger *logrus.Logger
+
+ // server configuration (location, forbidden files and etc)
+ cfg *Config
+
+ // root is initiated http directory
+ root http.Dir
+
+ // let's service stay running
+ done chan interface{}
+}
+
+// Configure must return configure service and return true if service hasStatus enabled. Must return error in case of
+// misconfiguration. Services must not be used without proper configuration pushed first.
+func (s *Service) Configure(cfg service.Config, c service.Container) (enabled bool, err error) {
+ config := &Config{}
+ if err := cfg.Unmarshal(config); err != nil {
+ return false, err
+ }
+
+ if !config.Enable {
+ return false, nil
+ }
+
+ if err := config.Valid(); err != nil {
+ return false, err
+ }
+
+ s.cfg = config
+ s.root = http.Dir(s.cfg.Dir)
+
+ // registering as middleware
+ if h, ok := c.Get(rrttp.Name); ok >= service.StatusConfigured {
+ if h, ok := h.(*rrttp.Service); ok {
+ h.Add(s)
+ }
+ } else {
+ if s.Logger != nil {
+ s.Logger.Warningf("no http service found")
+ }
+ }
+
+ return true, nil
+}
+
+// Serve serves the service.
+func (s *Service) Serve() error {
+ s.done = make(chan interface{})
+ <-s.done
+
+ return nil
+}
+
+// Stop stops the service.
+func (s *Service) Stop() {
+ //todo: this is not safe (TODO CHECK IT?)
+ close(s.done)
+}
+
+// Handle must return true if request/response pair is handled withing the middleware.
+func (s *Service) Handle(w http.ResponseWriter, r *http.Request) bool {
+ fPath := r.URL.Path
+ if !strings.HasPrefix(fPath, "/") {
+ fPath = "/" + fPath
+ }
+ fPath = path.Clean(fPath)
+
+ if s.cfg.Forbids(fPath) {
+ if s.Logger != nil {
+ s.Logger.Warningf("attempt to access forbidden file %s", fPath)
+ }
+ return false
+ }
+
+ f, err := s.root.Open(fPath)
+ if err != nil {
+ if !os.IsNotExist(err) {
+ if s.Logger != nil {
+ s.Logger.Error(err)
+ }
+ }
+
+ return false
+ }
+ defer f.Close()
+
+ d, err := f.Stat()
+ if err != nil {
+ if s.Logger != nil {
+ s.Logger.Error(err)
+ }
+ return false
+ }
+
+ // do not Handle directories
+ if d.IsDir() {
+ return false
+ }
+
+ http.ServeContent(w, r, d.Name(), d.ModTime(), f)
+ return true
+}