2015-11-01 15:35:01 +00:00
|
|
|
package provider
|
2015-09-07 13:25:13 +00:00
|
|
|
|
|
|
|
import (
|
2015-09-07 15:39:22 +00:00
|
|
|
"os"
|
|
|
|
"path/filepath"
|
|
|
|
"strings"
|
2015-09-24 15:16:13 +00:00
|
|
|
|
|
|
|
"github.com/BurntSushi/toml"
|
|
|
|
log "github.com/Sirupsen/logrus"
|
2015-11-01 15:35:01 +00:00
|
|
|
"github.com/emilevauge/traefik/types"
|
2015-09-24 15:16:13 +00:00
|
|
|
"gopkg.in/fsnotify.v1"
|
2015-09-07 13:25:13 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
type FileProvider struct {
|
2015-09-12 13:10:03 +00:00
|
|
|
Watch bool
|
2015-09-07 15:39:22 +00:00
|
|
|
Filename string
|
2015-09-07 13:25:13 +00:00
|
|
|
}
|
|
|
|
|
2015-11-01 15:35:01 +00:00
|
|
|
func (provider *FileProvider) Provide(configurationChan chan<- types.ConfigMessage) error {
|
2015-09-07 13:25:13 +00:00
|
|
|
watcher, err := fsnotify.NewWatcher()
|
|
|
|
if err != nil {
|
2015-09-11 14:37:13 +00:00
|
|
|
log.Error("Error creating file watcher", err)
|
2015-10-01 10:04:25 +00:00
|
|
|
return err
|
2015-09-07 13:25:13 +00:00
|
|
|
}
|
|
|
|
|
2015-09-07 15:39:22 +00:00
|
|
|
file, err := os.Open(provider.Filename)
|
2015-09-07 13:25:13 +00:00
|
|
|
if err != nil {
|
2015-09-11 14:37:13 +00:00
|
|
|
log.Error("Error opening file", err)
|
2015-10-01 10:04:25 +00:00
|
|
|
return err
|
2015-09-07 13:25:13 +00:00
|
|
|
}
|
2015-09-07 15:39:22 +00:00
|
|
|
defer file.Close()
|
2015-09-07 13:25:13 +00:00
|
|
|
|
2015-10-03 14:50:53 +00:00
|
|
|
if provider.Watch {
|
|
|
|
// Process events
|
|
|
|
go func() {
|
|
|
|
defer watcher.Close()
|
|
|
|
for {
|
|
|
|
select {
|
|
|
|
case event := <-watcher.Events:
|
|
|
|
if strings.Contains(event.Name, file.Name()) {
|
|
|
|
log.Debug("File event:", event)
|
|
|
|
configuration := provider.LoadFileConfig(file.Name())
|
|
|
|
if configuration != nil {
|
2015-11-01 15:35:01 +00:00
|
|
|
configurationChan <- types.ConfigMessage{"file", configuration}
|
2015-10-03 14:50:53 +00:00
|
|
|
}
|
2015-09-07 21:25:07 +00:00
|
|
|
}
|
2015-10-03 14:50:53 +00:00
|
|
|
case error := <-watcher.Errors:
|
|
|
|
log.Error("Watcher event error", error)
|
2015-09-07 15:39:22 +00:00
|
|
|
}
|
2015-09-07 13:25:13 +00:00
|
|
|
}
|
2015-10-03 14:50:53 +00:00
|
|
|
}()
|
2015-09-07 16:10:33 +00:00
|
|
|
err = watcher.Add(filepath.Dir(file.Name()))
|
2015-10-03 14:50:53 +00:00
|
|
|
if err != nil {
|
|
|
|
log.Error("Error adding file watcher", err)
|
|
|
|
return err
|
|
|
|
}
|
2015-09-07 15:39:22 +00:00
|
|
|
}
|
2015-09-07 13:25:13 +00:00
|
|
|
|
2015-09-07 22:15:14 +00:00
|
|
|
configuration := provider.LoadFileConfig(file.Name())
|
2015-11-01 15:35:01 +00:00
|
|
|
configurationChan <- types.ConfigMessage{"file", configuration}
|
2015-10-01 10:04:25 +00:00
|
|
|
return nil
|
2015-09-07 13:25:13 +00:00
|
|
|
}
|
|
|
|
|
2015-11-01 15:35:01 +00:00
|
|
|
func (provider *FileProvider) LoadFileConfig(filename string) *types.Configuration {
|
|
|
|
configuration := new(types.Configuration)
|
2015-09-07 22:15:14 +00:00
|
|
|
if _, err := toml.DecodeFile(filename, configuration); err != nil {
|
2015-09-11 14:37:13 +00:00
|
|
|
log.Error("Error reading file:", err)
|
2015-09-07 13:25:13 +00:00
|
|
|
return nil
|
|
|
|
}
|
2015-09-07 22:15:14 +00:00
|
|
|
return configuration
|
2015-09-12 13:10:03 +00:00
|
|
|
}
|