Use go module as dependency tool instead of glide
[src/xds/xds-agent.git] / 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 }