9 "github.com/franciscocpg/reflectme"
10 "github.com/iotbzh/xds-agent/lib/apiv1"
11 "github.com/iotbzh/xds-agent/lib/syncthing"
12 "github.com/syncthing/syncthing/lib/sync"
15 // Projects Represent a an XDS Projects
16 type Projects struct {
19 projects map[string]*IPROJECT
22 // Mutex to make add/delete atomic
23 var pjMutex = sync.NewMutex()
25 // NewProjects Create a new instance of Project Model
26 func NewProjects(ctx *Context, st *st.SyncThing) *Projects {
30 projects: make(map[string]*IPROJECT),
34 // Init Load Projects configuration
35 func (p *Projects) Init(server *XdsServer) error {
36 svrList := make(map[string]*XdsServer)
37 // If server not set, load for all servers
39 svrList = p.xdsServers
41 svrList[server.ID] = server
44 for _, svr := range svrList {
48 xFlds := []XdsFolderConfig{}
49 if err := svr.GetFolders(&xFlds); err != nil {
50 errMsg += fmt.Sprintf("Cannot retrieve folders config of XDS server ID %s : %v \n", svr.ID, err.Error())
53 p.Log.Debugf("Connected to XDS Server %s, %d projects detected", svr.ID, len(xFlds))
54 for _, prj := range xFlds {
55 newP := svr.FolderToProject(prj)
56 if _, err := p.createUpdate(newP, false, true); err != nil {
57 errMsg += "Error while creating project id " + prj.ID + ": " + err.Error() + "\n "
63 p.Log.Infof("Number of loaded Projects: %d", len(p.projects))
66 return fmt.Errorf(errMsg)
71 // ResolveID Complete a Project ID (helper for user that can use partial ID value)
72 func (p *Projects) ResolveID(id string) (string, error) {
78 for iid := range p.projects {
79 if strings.HasPrefix(iid, id) {
80 match = append(match, iid)
86 } else if len(match) == 0 {
87 return id, fmt.Errorf("Unknown id")
89 return id, fmt.Errorf("Multiple IDs found with provided prefix: " + id)
92 // Get returns the folder config or nil if not existing
93 func (p *Projects) Get(id string) *IPROJECT {
97 fc, exist := p.projects[id]
104 // GetProjectArr returns the config of all folders as an array
105 func (p *Projects) GetProjectArr() []apiv1.ProjectConfig {
107 defer pjMutex.Unlock()
109 return p.GetProjectArrUnsafe()
112 // GetProjectArrUnsafe Same as GetProjectArr without mutex protection
113 func (p *Projects) GetProjectArrUnsafe() []apiv1.ProjectConfig {
114 conf := []apiv1.ProjectConfig{}
115 for _, v := range p.projects {
116 prj := (*v).GetProject()
117 conf = append(conf, *prj)
122 // Add adds a new folder
123 func (p *Projects) Add(newF apiv1.ProjectConfig, fromSid string) (*apiv1.ProjectConfig, error) {
124 prj, err := p.createUpdate(newF, true, false)
129 // Notify client with event
130 if err := p.events.Emit(apiv1.EVTProjectAdd, *prj, fromSid); err != nil {
131 p.Log.Warningf("Cannot notify project deletion: %v", err)
137 // CreateUpdate creates or update a folder
138 func (p *Projects) createUpdate(newF apiv1.ProjectConfig, create bool, initial bool) (*apiv1.ProjectConfig, error) {
142 defer pjMutex.Unlock()
145 if _, exist := p.projects[newF.ID]; create && exist {
146 return nil, fmt.Errorf("ID already exists")
148 if newF.ClientPath == "" {
149 return nil, fmt.Errorf("ClientPath must be set")
151 if newF.ServerID == "" {
152 return nil, fmt.Errorf("Server ID must be set")
156 if svr, exist = p.xdsServers[newF.ServerID]; !exist {
157 return nil, fmt.Errorf("Unknown Server ID %s", newF.ServerID)
160 // Check type supported
161 b, exist := svr.ServerConfig.SupportedSharing[string(newF.Type)]
163 return nil, fmt.Errorf("Server doesn't support project type %s", newF.Type)
166 // Create a new folder object
170 case apiv1.TypeCloudSync:
172 fld = NewProjectST(p.Context, svr)
174 return nil, fmt.Errorf("Cloud Sync project not supported")
178 case apiv1.TypePathMap:
179 fld = NewProjectPathMap(p.Context, svr)
181 return nil, fmt.Errorf("Unsupported folder type")
184 var newPrj *apiv1.ProjectConfig
186 // Add project on server
187 if newPrj, err = fld.Add(newF); err != nil {
188 newF.Status = apiv1.StatusErrorConfig
189 log.Printf("ERROR Adding project: %v\n", err)
193 // Just update project config
194 if newPrj, err = fld.Setup(newF); err != nil {
195 newF.Status = apiv1.StatusErrorConfig
196 log.Printf("ERROR Updating project: %v\n", err)
203 log.Printf("ERROR project ID empty: %v", newF)
204 return newPrj, fmt.Errorf("Project ID empty")
207 // Add to folders list
208 p.projects[newPrj.ID] = &fld
210 // Force sync after creation
211 // (need to defer to be sure that WS events will arrive after HTTP creation reply)
213 time.Sleep(time.Millisecond * 500)
220 // Delete deletes a specific folder
221 func (p *Projects) Delete(id, fromSid string) (apiv1.ProjectConfig, error) {
225 defer pjMutex.Unlock()
227 fld := apiv1.ProjectConfig{}
228 fc, exist := p.projects[id]
230 return fld, fmt.Errorf("Unknown id")
233 prj := (*fc).GetProject()
235 if err = (*fc).Delete(); err != nil {
239 delete(p.projects, id)
241 // Notify client with event
242 if err := p.events.Emit(apiv1.EVTProjectDelete, *prj, fromSid); err != nil {
243 p.Log.Warningf("Cannot notify project deletion: %v", err)
249 // ForceSync Force the synchronization of a folder
250 func (p *Projects) ForceSync(id string) error {
253 return fmt.Errorf("Unknown id")
258 // IsProjectInSync Returns true when folder is in sync
259 func (p *Projects) IsProjectInSync(id string) (bool, error) {
262 return false, fmt.Errorf("Unknown id")
264 return (*fc).IsInSync()
267 // Update Update some field of a project
268 func (p *Projects) Update(id string, prj apiv1.ProjectConfig, fromSid string) (*apiv1.ProjectConfig, error) {
271 defer pjMutex.Unlock()
273 fc, exist := p.projects[id]
275 return nil, fmt.Errorf("Unknown id")
278 // Copy current in a new object to change nothing in case of an error rises
279 newFld := apiv1.ProjectConfig{}
280 reflectme.Copy((*fc).GetProject(), &newFld)
282 // Only update some fields
284 for _, fieldName := range apiv1.ProjectConfigUpdatableFields {
285 valNew, err := reflectme.GetField(prj, fieldName)
287 valCur, err := reflectme.GetField(newFld, fieldName)
288 if err == nil && valNew != valCur {
289 err = reflectme.SetField(&newFld, fieldName, valNew)
302 upPrj, err := (*fc).Update(newFld)
307 // Notify client with event
308 if err := p.events.Emit(apiv1.EVTProjectChange, *upPrj, fromSid); err != nil {
309 p.Log.Warningf("Cannot notify project change: %v", err)