diff options
Diffstat (limited to 'service/static/service.go')
-rw-r--r-- | service/static/service.go | 120 |
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 +} |