c5318de68e53a8d4b4bc95692e8cebfb93daebbd
[src/xds/xds-server.git] / lib / xdsserver / folder-pathmap.go
1 package xdsserver
2
3 import (
4         "fmt"
5         "io/ioutil"
6         "os"
7         "path/filepath"
8         "strings"
9
10         common "github.com/iotbzh/xds-common/golib"
11         "github.com/iotbzh/xds-server/lib/xsapiv1"
12         uuid "github.com/satori/go.uuid"
13 )
14
15 // IFOLDER interface implementation for native/path mapping folders
16
17 // PathMap .
18 type PathMap struct {
19         *Context
20         config xsapiv1.FolderConfig
21 }
22
23 // NewFolderPathMap Create a new instance of PathMap
24 func NewFolderPathMap(ctx *Context) *PathMap {
25         f := PathMap{
26                 Context: ctx,
27                 config: xsapiv1.FolderConfig{
28                         Status: xsapiv1.StatusDisable,
29                 },
30         }
31         return &f
32 }
33
34 // NewUID Get a UUID
35 func (f *PathMap) NewUID(suffix string) string {
36         uuid := uuid.NewV1().String()
37         if len(suffix) > 0 {
38                 uuid += "_" + suffix
39         }
40         return uuid
41 }
42
43 // Add a new folder
44 func (f *PathMap) Add(cfg xsapiv1.FolderConfig) (*xsapiv1.FolderConfig, error) {
45         if cfg.DataPathMap.ServerPath == "" {
46                 return nil, fmt.Errorf("ServerPath must be set")
47         }
48
49         // Use shareRootDir if ServerPath is a relative path
50         dir := cfg.DataPathMap.ServerPath
51         if !filepath.IsAbs(dir) {
52                 dir = filepath.Join(f.Config.FileConf.ShareRootDir, dir)
53         }
54
55         // Sanity check
56         if !common.Exists(dir) {
57                 // try to create if not existing
58                 if err := os.MkdirAll(dir, 0755); err != nil {
59                         return nil, fmt.Errorf("Cannot create ServerPath directory: %s", dir)
60                 }
61         }
62         if !common.Exists(dir) {
63                 return nil, fmt.Errorf("ServerPath directory is not accessible: %s", dir)
64         }
65
66         f.config = cfg
67         f.config.RootPath = dir
68         f.config.DataPathMap.ServerPath = dir
69         f.config.IsInSync = true
70
71         // Verify file created by XDS agent when needed
72         if cfg.DataPathMap.CheckFile != "" {
73                 errMsg := "ServerPath sanity check error (%d): %v"
74                 ckFile := f.ConvPathCli2Svr(cfg.DataPathMap.CheckFile)
75                 if !common.Exists(ckFile) {
76                         return nil, fmt.Errorf(errMsg, 1, "file not present")
77                 }
78                 if cfg.DataPathMap.CheckContent != "" {
79                         fd, err := os.OpenFile(ckFile, os.O_APPEND|os.O_RDWR, 0600)
80                         if err != nil {
81                                 return nil, fmt.Errorf(errMsg, 2, err)
82                         }
83                         defer fd.Close()
84
85                         // Check specific message written by agent
86                         content, err := ioutil.ReadAll(fd)
87                         if err != nil {
88                                 return nil, fmt.Errorf(errMsg, 3, err)
89                         }
90                         if string(content) != cfg.DataPathMap.CheckContent {
91                                 return nil, fmt.Errorf(errMsg, 4, "file content differ")
92                         }
93
94                         // Write a specific message that will be check back on agent side
95                         msg := "Pathmap checked message written by xds-server ID: " + f.Config.ServerUID + "\n"
96                         if n, err := fd.WriteString(msg); n != len(msg) || err != nil {
97                                 return nil, fmt.Errorf(errMsg, 5, err)
98                         }
99                 }
100         }
101
102         f.config.Status = xsapiv1.StatusEnable
103
104         return &f.config, nil
105 }
106
107 // GetConfig Get public part of folder config
108 func (f *PathMap) GetConfig() xsapiv1.FolderConfig {
109         return f.config
110 }
111
112 // GetFullPath returns the full path of a directory (from server POV)
113 func (f *PathMap) GetFullPath(dir string) string {
114         if &dir == nil {
115                 return f.config.DataPathMap.ServerPath
116         }
117         return filepath.Join(f.config.DataPathMap.ServerPath, dir)
118 }
119
120 // ConvPathCli2Svr Convert path from Client to Server
121 func (f *PathMap) ConvPathCli2Svr(s string) string {
122         if f.config.ClientPath != "" && f.config.DataPathMap.ServerPath != "" {
123                 return strings.Replace(s,
124                         f.config.ClientPath,
125                         f.config.DataPathMap.ServerPath,
126                         -1)
127         }
128         return s
129 }
130
131 // ConvPathSvr2Cli Convert path from Server to Client
132 func (f *PathMap) ConvPathSvr2Cli(s string) string {
133         if f.config.ClientPath != "" && f.config.DataPathMap.ServerPath != "" {
134                 return strings.Replace(s,
135                         f.config.DataPathMap.ServerPath,
136                         f.config.ClientPath,
137                         -1)
138         }
139         return s
140 }
141
142 // Remove a folder
143 func (f *PathMap) Remove() error {
144         // nothing to do
145         return nil
146 }
147
148 // Update update some fields of a folder
149 func (f *PathMap) Update(cfg xsapiv1.FolderConfig) (*xsapiv1.FolderConfig, error) {
150         if f.config.ID != cfg.ID {
151                 return nil, fmt.Errorf("Invalid id")
152         }
153         f.config = cfg
154         return &f.config, nil
155 }
156
157 // RegisterEventChange requests registration for folder change event
158 func (f *PathMap) RegisterEventChange(cb *FolderEventCB, data *FolderEventCBData) error {
159         return nil
160 }
161
162 // UnRegisterEventChange remove registered callback
163 func (f *PathMap) UnRegisterEventChange() error {
164         return nil
165 }
166
167 // Sync Force folder files synchronization
168 func (f *PathMap) Sync() error {
169         return nil
170 }
171
172 // IsInSync Check if folder files are in-sync
173 func (f *PathMap) IsInSync() (bool, error) {
174         return true, nil
175 }