9 "github.com/iotbzh/xds-agent/lib/apiv1"
10 "github.com/iotbzh/xds-agent/lib/syncthing"
11 "github.com/syncthing/syncthing/lib/sync"
14 // Projects Represent a an XDS Projects
15 type Projects struct {
18 projects map[string]*IPROJECT
21 // Mutex to make add/delete atomic
22 var pjMutex = sync.NewMutex()
24 // NewProjects Create a new instance of Project Model
25 func NewProjects(ctx *Context, st *st.SyncThing) *Projects {
29 projects: make(map[string]*IPROJECT),
33 // Init Load Projects configuration
34 func (p *Projects) Init(server *XdsServer) error {
35 svrList := make(map[string]*XdsServer)
36 // If server not set, load for all servers
38 svrList = p.xdsServers
40 svrList[server.ID] = server
43 for _, svr := range svrList {
47 xFlds := []XdsFolderConfig{}
48 if err := svr.GetFolders(&xFlds); err != nil {
49 errMsg += fmt.Sprintf("Cannot retrieve folders config of XDS server ID %s : %v \n", svr.ID, err.Error())
52 p.Log.Debugf("Connected to XDS Server %s, %d projects detected", svr.ID, len(xFlds))
53 for _, prj := range xFlds {
54 newP := svr.FolderToProject(prj)
55 if _, err := p.createUpdate(newP, false, true); err != nil {
56 errMsg += "Error while creating project id " + prj.ID + ": " + err.Error() + "\n "
62 p.Log.Infof("Number of loaded Projects: %d", len(p.projects))
65 return fmt.Errorf(errMsg)
70 // ResolveID Complete a Project ID (helper for user that can use partial ID value)
71 func (p *Projects) ResolveID(id string) (string, error) {
77 for iid := range p.projects {
78 if strings.HasPrefix(iid, id) {
79 match = append(match, iid)
85 } else if len(match) == 0 {
86 return id, fmt.Errorf("Unknown id")
88 return id, fmt.Errorf("Multiple IDs found with provided prefix: " + id)
91 // Get returns the folder config or nil if not existing
92 func (p *Projects) Get(id string) *IPROJECT {
96 fc, exist := p.projects[id]
103 // GetProjectArr returns the config of all folders as an array
104 func (p *Projects) GetProjectArr() []apiv1.ProjectConfig {
106 defer pjMutex.Unlock()
108 return p.GetProjectArrUnsafe()
111 // GetProjectArrUnsafe Same as GetProjectArr without mutex protection
112 func (p *Projects) GetProjectArrUnsafe() []apiv1.ProjectConfig {
113 conf := []apiv1.ProjectConfig{}
114 for _, v := range p.projects {
115 prj := (*v).GetProject()
116 conf = append(conf, *prj)
121 // Add adds a new folder
122 func (p *Projects) Add(newF apiv1.ProjectConfig) (*apiv1.ProjectConfig, error) {
123 prj, err := p.createUpdate(newF, true, false)
128 // Notify client with event
129 if err := p.events.Emit(apiv1.EVTProjectAdd, *prj); err != nil {
130 p.Log.Warningf("Cannot notify project deletion: %v", err)
136 // CreateUpdate creates or update a folder
137 func (p *Projects) createUpdate(newF apiv1.ProjectConfig, create bool, initial bool) (*apiv1.ProjectConfig, error) {
141 defer pjMutex.Unlock()
144 if _, exist := p.projects[newF.ID]; create && exist {
145 return nil, fmt.Errorf("ID already exists")
147 if newF.ClientPath == "" {
148 return nil, fmt.Errorf("ClientPath must be set")
150 if newF.ServerID == "" {
151 return nil, fmt.Errorf("Server ID must be set")
155 if svr, exist = p.xdsServers[newF.ServerID]; !exist {
156 return nil, fmt.Errorf("Unknown Server ID %s", newF.ServerID)
159 // Check type supported
160 b, exist := svr.ServerConfig.SupportedSharing[string(newF.Type)]
162 return nil, fmt.Errorf("Server doesn't support project type %s", newF.Type)
165 // Create a new folder object
169 case apiv1.TypeCloudSync:
171 fld = NewProjectST(p.Context, svr)
173 return nil, fmt.Errorf("Cloud Sync project not supported")
177 case apiv1.TypePathMap:
178 fld = NewProjectPathMap(p.Context, svr)
180 return nil, fmt.Errorf("Unsupported folder type")
183 var newPrj *apiv1.ProjectConfig
185 // Add project on server
186 if newPrj, err = fld.Add(newF); err != nil {
187 newF.Status = apiv1.StatusErrorConfig
188 log.Printf("ERROR Adding project: %v\n", err)
192 // Just update project config
193 if newPrj, err = fld.UpdateProject(newF); err != nil {
194 newF.Status = apiv1.StatusErrorConfig
195 log.Printf("ERROR Updating project: %v\n", err)
202 log.Printf("ERROR project ID empty: %v", newF)
203 return newPrj, fmt.Errorf("Project ID empty")
206 // Add to folders list
207 p.projects[newPrj.ID] = &fld
209 // Force sync after creation
210 // (need to defer to be sure that WS events will arrive after HTTP creation reply)
212 time.Sleep(time.Millisecond * 500)
219 // Delete deletes a specific folder
220 func (p *Projects) Delete(id string) (apiv1.ProjectConfig, error) {
224 defer pjMutex.Unlock()
226 fld := apiv1.ProjectConfig{}
227 fc, exist := p.projects[id]
229 return fld, fmt.Errorf("Unknown id")
232 prj := (*fc).GetProject()
234 if err = (*fc).Delete(); err != nil {
238 delete(p.projects, id)
240 // Notify client with event
241 if err := p.events.Emit(apiv1.EVTProjectDelete, *prj); err != nil {
242 p.Log.Warningf("Cannot notify project deletion: %v", err)
248 // ForceSync Force the synchronization of a folder
249 func (p *Projects) ForceSync(id string) error {
252 return fmt.Errorf("Unknown id")
257 // IsProjectInSync Returns true when folder is in sync
258 func (p *Projects) IsProjectInSync(id string) (bool, error) {
261 return false, fmt.Errorf("Unknown id")
263 return (*fc).IsInSync()