lib/agent/project-st.go

   1 /*
   2  * Copyright (C) 2017-2018 "IoT.bzh"
   3  * Author Sebastien Douheret <sebastien@iot.bzh>
   4  *
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *   http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  */
  17
  18 package agent
  19
  20 import (
  21         "encoding/json"
  22         "fmt"
  23
  24         st "gerrit.automotivelinux.org/gerrit/src/xds/xds-agent.git/lib/syncthing"
  25         "gerrit.automotivelinux.org/gerrit/src/xds/xds-agent.git/lib/xaapiv1"
  26         "gerrit.automotivelinux.org/gerrit/src/xds/xds-server.git/lib/xsapiv1"
  27 )
  28
  29 // IPROJECT interface implementation for syncthing projects
  30
  31 // STProject .
  32 type STProject struct {
  33         *Context
  34         server   *XdsServer
  35         folder   *xsapiv1.FolderConfig
  36         eventIDs []int
  37 }
  38
  39 // NewProjectST Create a new instance of STProject
  40 func NewProjectST(ctx *Context, svr *XdsServer) *STProject {
  41         p := STProject{
  42                 Context: ctx,
  43                 server:  svr,
  44                 folder:  &xsapiv1.FolderConfig{},
  45         }
  46         return &p
  47 }
  48
  49 // Add a new project
  50 func (p *STProject) Add(cfg xaapiv1.ProjectConfig) (*xaapiv1.ProjectConfig, error) {
  51         var err error
  52
  53         // Add project/folder into XDS Server
  54         err = p.server.FolderAdd(p.server.ProjectToFolder(cfg), p.folder)
  55         if err != nil {
  56                 return nil, err
  57         }
  58         svrPrj := p.GetProject()
  59
  60         // Declare project into local Syncthing
  61         id, err := p.SThg.FolderChange(st.FolderChangeArg{
  62                 ID:           svrPrj.ID,
  63                 Label:        svrPrj.Label,
  64                 RelativePath: cfg.ClientPath,
  65                 SyncThingID:  p.server.ServerConfig.Builder.SyncThingID,
  66         })
  67         if err != nil {
  68                 return nil, err
  69         }
  70
  71         locPrj, err := p.SThg.FolderConfigGet(id)
  72         if err != nil {
  73                 svrPrj.Status = xaapiv1.StatusErrorConfig
  74                 return nil, err
  75         }
  76         if svrPrj.ID != locPrj.ID {
  77                 p.Log.Errorf("Project ID in XDSServer and local ST differ: %s != %s", svrPrj.ID, locPrj.ID)
  78         }
  79
  80         // Use Setup function to setup remains fields
  81         return p.Setup(*svrPrj)
  82 }
  83
  84 // Delete a project
  85 func (p *STProject) Delete() error {
  86         errSvr := p.server.FolderDelete(p.folder.ID)
  87         errLoc := p.SThg.FolderDelete(p.folder.ID)
  88         if errSvr != nil {
  89                 return errSvr
  90         }
  91         return errLoc
  92 }
  93
  94 // GetProject Get public part of project config
  95 func (p *STProject) GetProject() *xaapiv1.ProjectConfig {
  96         prj := p.server.FolderToProject(*p.folder)
  97         prj.ServerID = p.server.ID
  98         return &prj
  99 }
 100
 101 // Setup Setup local project config
 102 func (p *STProject) Setup(prj xaapiv1.ProjectConfig) (*xaapiv1.ProjectConfig, error) {
 103         // Update folder
 104         p.folder = p.server.ProjectToFolder(prj)
 105         svrPrj := p.GetProject()
 106
 107         // Register events to update folder status
 108         // Register to XDS Server events
 109         if _, err := p.server.EventOn(xsapiv1.EVTFolderStateChange, "", p._cbServerFolderChanged); err != nil {
 110                 p.Log.Errorf("XDS Server EventOn '%s' failed: %v", xsapiv1.EVTFolderStateChange, err)
 111                 return svrPrj, err
 112         }
 113
 114         // Register to Local Syncthing events
 115         for _, evName := range []string{st.EventStateChanged, st.EventFolderPaused} {
 116                 evID, err := p.SThg.Events.Register(evName, p._cbLocalSTEvents, svrPrj.ID, nil)
 117                 if err != nil {
 118                         return nil, err
 119                 }
 120                 p.eventIDs = append(p.eventIDs, evID)
 121         }
 122
 123         return svrPrj, nil
 124 }
 125
 126 // Update Update some field of a project
 127 func (p *STProject) Update(prj xaapiv1.ProjectConfig) (*xaapiv1.ProjectConfig, error) {
 128
 129         if p.folder.ID != prj.ID {
 130                 return nil, fmt.Errorf("Invalid id")
 131         }
 132
 133         err := p.server.FolderUpdate(p.server.ProjectToFolder(prj), p.folder)
 134         if err != nil {
 135                 return nil, err
 136         }
 137
 138         return p.GetProject(), nil
 139 }
 140
 141 // GetServer Get the XdsServer that holds this project
 142 func (p *STProject) GetServer() *XdsServer {
 143         return p.server
 144 }
 145
 146 // Sync Force project files synchronization
 147 func (p *STProject) Sync() error {
 148         if err := p.server.FolderSync(p.folder.ID); err != nil {
 149                 return err
 150         }
 151         return p.SThg.FolderScan(p.folder.ID, "")
 152 }
 153
 154 // IsInSync Check if project files are in-sync
 155 func (p *STProject) IsInSync() (bool, error) {
 156         // Should be up-to-date by callbacks (see below)
 157         return p.folder.IsInSync, nil
 158 }
 159
 160 /**
 161 ** Private functions
 162 ***/
 163
 164 // callback use to update (XDS Server) folder IsInSync status
 165
 166 func (p *STProject) _cbServerFolderChanged(pData interface{}, data interface{}) error {
 167         evt := xsapiv1.EventMsg{}
 168         d, err := json.Marshal(data)
 169         if err != nil {
 170                 p.Log.Errorf("Cannot marshal XDS Server event folder-change err=%v", err)
 171                 return err
 172         }
 173         if err = json.Unmarshal(d, &evt); err != nil {
 174                 p.Log.Errorf("Cannot unmarshal XDS Server event folder-change err=%v", err)
 175                 return err
 176         }
 177
 178         fld, err := evt.DecodeFolderConfig()
 179         if err != nil {
 180                 p.Log.Errorf("Cannot decode FolderChanged event: %v", data)
 181         }
 182
 183         // Only process event that concerns this project/folder ID
 184         if p.folder.ID != fld.ID {
 185                 return nil
 186         }
 187
 188         if fld.IsInSync != p.folder.DataCloudSync.STSvrIsInSync ||
 189                 fld.Status != p.folder.DataCloudSync.STSvrStatus {
 190
 191                 p.folder.DataCloudSync.STSvrIsInSync = fld.IsInSync
 192                 p.folder.DataCloudSync.STSvrStatus = fld.Status
 193
 194                 if err := p.events.Emit(xaapiv1.EVTProjectChange, p.server.FolderToProject(*p.folder), ""); err != nil {
 195                         p.Log.Warningf("Cannot notify project change (from server): %v", err)
 196                 }
 197         }
 198         return nil
 199 }
 200
 201 // callback use to update IsInSync status
 202 func (p *STProject) _cbLocalSTEvents(ev st.Event, data *st.EventsCBData) {
 203
 204         inSync := p.folder.DataCloudSync.STLocIsInSync
 205         sts := p.folder.DataCloudSync.STLocStatus
 206         prevSync := inSync
 207         prevStatus := sts
 208
 209         switch ev.Type {
 210
 211         case st.EventStateChanged:
 212                 to := ev.Data["to"]
 213                 switch to {
 214                 case "scanning", "syncing":
 215                         sts = xaapiv1.StatusSyncing
 216                 case "idle":
 217                         sts = xaapiv1.StatusEnable
 218                 }
 219                 inSync = (to == "idle")
 220
 221         case st.EventFolderPaused:
 222                 if sts == xaapiv1.StatusEnable {
 223                         sts = xaapiv1.StatusPause
 224                 }
 225                 inSync = false
 226         }
 227
 228         if prevSync != inSync || prevStatus != sts {
 229
 230                 p.folder.DataCloudSync.STLocIsInSync = inSync
 231                 p.folder.DataCloudSync.STLocStatus = sts
 232
 233                 if err := p.events.Emit(xaapiv1.EVTProjectChange, p.server.FolderToProject(*p.folder), ""); err != nil {
 234                         p.Log.Warningf("Cannot notify project change (local): %v", err)
 235                 }
 236         }
 237 }