Bump Copyright to 2019
[src/xds/xds-server.git] / lib / xdsconfig / data.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 xdsconfig
19
20 import (
21         "encoding/xml"
22         "fmt"
23         "os"
24         "path/filepath"
25
26         common "gerrit.automotivelinux.org/gerrit/src/xds/xds-common.git/golib"
27         uuid "github.com/satori/go.uuid"
28         "github.com/syncthing/syncthing/lib/sync"
29 )
30
31 // xmlServerData contains persistent data stored/loaded by server
32 type xmlServerData struct {
33         XMLName xml.Name   `xml:"XDS-Server"`
34         Version string     `xml:"version,attr"`
35         Data    ServerData `xml:"server-data"`
36 }
37
38 // ServerData Hold server data saved in ServerData file
39 type ServerData struct {
40         ID string `xml:"id"`
41 }
42
43 var sdMutex = sync.NewMutex()
44
45 // ServerIDGet Get Server unique ID
46 func ServerIDGet() (string, error) {
47         var f string
48         var err error
49
50         d := ServerData{}
51         if f, err = ServerDataFilenameGet(); err != nil {
52                 return "", err
53         }
54         if err = serverDataRead(f, &d); err != nil || d.ID == "" {
55                 // Create a new uuid when not found
56                 d.ID = uuid.NewV1().String()
57                 if err := serverDataWrite(f, d); err != nil {
58                         return "", err
59                 }
60         }
61         return d.ID, nil
62 }
63
64 // serverDataRead reads data saved on disk
65 func serverDataRead(file string, data *ServerData) error {
66         if !common.Exists(file) {
67                 return fmt.Errorf("No folder config file found (%s)", file)
68         }
69
70         sdMutex.Lock()
71         defer sdMutex.Unlock()
72
73         fd, err := os.Open(file)
74         defer fd.Close()
75         if err != nil {
76                 return err
77         }
78
79         xsd := xmlServerData{}
80         err = xml.NewDecoder(fd).Decode(&xsd)
81         if err == nil {
82                 *data = xsd.Data
83         }
84         return err
85 }
86
87 // serverDataWrite writes persistant data to disk
88 func serverDataWrite(file string, data ServerData) error {
89         sdMutex.Lock()
90         defer sdMutex.Unlock()
91
92         dir := filepath.Dir(file)
93         if !common.Exists(dir) {
94                 if err := os.MkdirAll(dir, 0755); err != nil {
95                         return fmt.Errorf("Cannot create server data directory: %s", dir)
96                 }
97         }
98
99         fd, err := os.OpenFile(file, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0666)
100         defer fd.Close()
101         if err != nil {
102                 return err
103         }
104
105         xsd := &xmlServerData{
106                 Version: "1",
107                 Data:    data,
108         }
109
110         enc := xml.NewEncoder(fd)
111         enc.Indent("", "  ")
112         return enc.Encode(xsd)
113 }