X-Git-Url: https://gerrit.automotivelinux.org/gerrit/gitweb?a=blobdiff_plain;f=lib%2Fagent%2Fprojects.go;h=f089882f2f65fa0018971fdd4cd6e310366278ef;hb=0a5ab2ca1ab89e6e667b7f9c7bdffdff036ca9b7;hp=39c120f4cccb1784306ad8971f1a423d5752eeb1;hpb=97ca1f277dc8b6973d6fa67add5593a9c395ce60;p=src%2Fxds%2Fxds-agent.git diff --git a/lib/agent/projects.go b/lib/agent/projects.go index 39c120f..f089882 100644 --- a/lib/agent/projects.go +++ b/lib/agent/projects.go @@ -3,8 +3,10 @@ package agent import ( "fmt" "log" + "strings" "time" + "github.com/iotbzh/xds-agent/lib/apiv1" "github.com/iotbzh/xds-agent/lib/syncthing" "github.com/syncthing/syncthing/lib/sync" ) @@ -14,16 +16,8 @@ type Projects struct { *Context SThg *st.SyncThing projects map[string]*IPROJECT - //SEB registerCB []RegisteredCB } -/* SEB -type RegisteredCB struct { - cb *EventCB - data *EventCBData -} -*/ - // Mutex to make add/delete atomic var pjMutex = sync.NewMutex() @@ -33,7 +27,6 @@ func NewProjects(ctx *Context, st *st.SyncThing) *Projects { Context: ctx, SThg: st, projects: make(map[string]*IPROJECT), - //registerCB: []RegisteredCB{}, } } @@ -51,26 +44,19 @@ func (p *Projects) Init(server *XdsServer) error { if svr.Disabled { continue } - xFlds := []FolderConfig{} - if err := svr.HTTPGet("/folders", &xFlds); err != nil { + xFlds := []XdsFolderConfig{} + if err := svr.GetFolders(&xFlds); err != nil { errMsg += fmt.Sprintf("Cannot retrieve folders config of XDS server ID %s : %v \n", svr.ID, err.Error()) continue } - p.Log.Debugf("Server %s, %d projects detected", svr.ID[:8], len(xFlds)) + p.Log.Debugf("Connected to XDS Server %s, %d projects detected", svr.ID, len(xFlds)) for _, prj := range xFlds { newP := svr.FolderToProject(prj) - if /*nPrj*/ _, err := p.createUpdate(newP, false, true); err != nil { + if _, err := p.createUpdate(newP, false, true); err != nil { errMsg += "Error while creating project id " + prj.ID + ": " + err.Error() + "\n " continue } - - /* FIXME emit EVTProjectChange event ? - if err := p.events.Emit(EVTProjectChange, *nPrj); err != nil { - p.Log.Warningf("Cannot notify project change: %v", err) - } - */ } - } p.Log.Infof("Number of loaded Projects: %d", len(p.projects)) @@ -81,6 +67,27 @@ func (p *Projects) Init(server *XdsServer) error { return nil } +// ResolveID Complete a Project ID (helper for user that can use partial ID value) +func (p *Projects) ResolveID(id string) (string, error) { + if id == "" { + return "", nil + } + + match := []string{} + for iid := range p.projects { + if strings.HasPrefix(iid, id) { + match = append(match, iid) + } + } + + if len(match) == 1 { + return match[0], nil + } else if len(match) == 0 { + return id, fmt.Errorf("Unknown id") + } + return id, fmt.Errorf("Multiple IDs found with provided prefix: " + id) +} + // Get returns the folder config or nil if not existing func (p *Projects) Get(id string) *IPROJECT { if id == "" { @@ -94,7 +101,7 @@ func (p *Projects) Get(id string) *IPROJECT { } // GetProjectArr returns the config of all folders as an array -func (p *Projects) GetProjectArr() []ProjectConfig { +func (p *Projects) GetProjectArr() []apiv1.ProjectConfig { pjMutex.Lock() defer pjMutex.Unlock() @@ -102,8 +109,8 @@ func (p *Projects) GetProjectArr() []ProjectConfig { } // GetProjectArrUnsafe Same as GetProjectArr without mutex protection -func (p *Projects) GetProjectArrUnsafe() []ProjectConfig { - conf := []ProjectConfig{} +func (p *Projects) GetProjectArrUnsafe() []apiv1.ProjectConfig { + conf := []apiv1.ProjectConfig{} for _, v := range p.projects { prj := (*v).GetProject() conf = append(conf, *prj) @@ -112,14 +119,14 @@ func (p *Projects) GetProjectArrUnsafe() []ProjectConfig { } // Add adds a new folder -func (p *Projects) Add(newF ProjectConfig) (*ProjectConfig, error) { +func (p *Projects) Add(newF apiv1.ProjectConfig) (*apiv1.ProjectConfig, error) { prj, err := p.createUpdate(newF, true, false) if err != nil { return prj, err } // Notify client with event - if err := p.events.Emit(EVTProjectAdd, *prj); err != nil { + if err := p.events.Emit(apiv1.EVTProjectAdd, *prj); err != nil { p.Log.Warningf("Cannot notify project deletion: %v", err) } @@ -127,7 +134,7 @@ func (p *Projects) Add(newF ProjectConfig) (*ProjectConfig, error) { } // CreateUpdate creates or update a folder -func (p *Projects) createUpdate(newF ProjectConfig, create bool, initial bool) (*ProjectConfig, error) { +func (p *Projects) createUpdate(newF apiv1.ProjectConfig, create bool, initial bool) (*apiv1.ProjectConfig, error) { var err error pjMutex.Lock() @@ -159,32 +166,35 @@ func (p *Projects) createUpdate(newF ProjectConfig, create bool, initial bool) ( var fld IPROJECT switch newF.Type { // SYNCTHING - case TypeCloudSync: + case apiv1.TypeCloudSync: if p.SThg != nil { - /*SEB fld = f.SThg.NewFolderST(f.Conf)*/ fld = NewProjectST(p.Context, svr) } else { return nil, fmt.Errorf("Cloud Sync project not supported") } // PATH MAP - case TypePathMap: + case apiv1.TypePathMap: fld = NewProjectPathMap(p.Context, svr) default: return nil, fmt.Errorf("Unsupported folder type") } - var newPrj *ProjectConfig + var newPrj *apiv1.ProjectConfig if create { // Add project on server if newPrj, err = fld.Add(newF); err != nil { - newF.Status = StatusErrorConfig - log.Printf("ERROR Adding folder: %v\n", err) + newF.Status = apiv1.StatusErrorConfig + log.Printf("ERROR Adding project: %v\n", err) return newPrj, err } } else { // Just update project config - newPrj = fld.SetProject(newF) + if newPrj, err = fld.UpdateProject(newF); err != nil { + newF.Status = apiv1.StatusErrorConfig + log.Printf("ERROR Updating project: %v\n", err) + return newPrj, err + } } // Sanity check @@ -207,16 +217,16 @@ func (p *Projects) createUpdate(newF ProjectConfig, create bool, initial bool) ( } // Delete deletes a specific folder -func (p *Projects) Delete(id string) (ProjectConfig, error) { +func (p *Projects) Delete(id string) (apiv1.ProjectConfig, error) { var err error pjMutex.Lock() defer pjMutex.Unlock() - fld := ProjectConfig{} + fld := apiv1.ProjectConfig{} fc, exist := p.projects[id] if !exist { - return fld, fmt.Errorf("unknown id") + return fld, fmt.Errorf("Unknown id") } prj := (*fc).GetProject() @@ -228,7 +238,7 @@ func (p *Projects) Delete(id string) (ProjectConfig, error) { delete(p.projects, id) // Notify client with event - if err := p.events.Emit(EVTProjectDelete, *prj); err != nil { + if err := p.events.Emit(apiv1.EVTProjectDelete, *prj); err != nil { p.Log.Warningf("Cannot notify project deletion: %v", err) }