lib/agent/project-pathmap.go

   1 package agent
   2
   3 import (
   4         "fmt"
   5         "io/ioutil"
   6         "os"
   7         "strings"
   8
   9         common "github.com/iotbzh/xds-common/golib"
  10 )
  11
  12 // IPROJECT interface implementation for native/path mapping projects
  13
  14 // PathMap .
  15 type PathMap struct {
  16         *Context
  17         server *XdsServer
  18         folder *XdsFolderConfig
  19 }
  20
  21 // NewProjectPathMap Create a new instance of PathMap
  22 func NewProjectPathMap(ctx *Context, svr *XdsServer) *PathMap {
  23         p := PathMap{
  24                 Context: ctx,
  25                 server:  svr,
  26                 folder:  &XdsFolderConfig{},
  27         }
  28         return &p
  29 }
  30
  31 // Add a new project
  32 func (p *PathMap) Add(cfg ProjectConfig) (*ProjectConfig, error) {
  33         var err error
  34         var file *os.File
  35         errMsg := "ClientPath sanity check error (%d): %v"
  36
  37         // Sanity check to verify that we have RW permission and path-mapping is correct
  38         dir := cfg.ClientPath
  39         if !common.Exists(dir) {
  40                 // try to create if not existing
  41                 if err := os.MkdirAll(dir, 0755); err != nil {
  42                         return nil, fmt.Errorf("Cannot create ClientPath directory: %s", dir)
  43                 }
  44         }
  45         if !common.Exists(dir) {
  46                 return nil, fmt.Errorf("ClientPath directory is not accessible: %s", dir)
  47         }
  48         if file, err = ioutil.TempFile(dir, ".xds_pathmap_check"); err != nil {
  49                 return nil, fmt.Errorf(errMsg, 1, err)
  50         }
  51         // Write a specific message that will be check by server during folder add
  52         msg := "Pathmap checked message written by xds-agent ID: " + p.Config.AgentUID + "\n"
  53         if n, err := file.WriteString(msg); n != len(msg) || err != nil {
  54                 return nil, fmt.Errorf(errMsg, 2, err)
  55         }
  56         defer func() {
  57                 if file != nil {
  58                         os.Remove(file.Name())
  59                         file.Close()
  60                 }
  61         }()
  62
  63         // Convert to Xds folder
  64         fld := p.server.ProjectToFolder(cfg)
  65         fld.DataPathMap.CheckFile = file.Name()
  66         fld.DataPathMap.CheckContent = msg
  67
  68         // Send request to create folder on XDS server side
  69         err = p.server.FolderAdd(fld, p.folder)
  70         if err != nil {
  71                 return nil, fmt.Errorf("Folders mapping verification failure.\n%v", err)
  72         }
  73
  74         // 2nd part of sanity checker
  75         // check specific message added by XDS Server during folder add processing
  76         content, err := ioutil.ReadFile(file.Name())
  77         if err != nil {
  78                 return nil, fmt.Errorf(errMsg, 3, err)
  79         }
  80         if !strings.Contains(string(content),
  81                 "Pathmap checked message written by xds-server ID") {
  82                 return nil, fmt.Errorf(errMsg, 4, "file content differ")
  83         }
  84
  85         return p.GetProject(), nil
  86 }
  87
  88 // Delete a project
  89 func (p *PathMap) Delete() error {
  90         return p.server.FolderDelete(p.folder.ID)
  91 }
  92
  93 // GetProject Get public part of project config
  94 func (p *PathMap) GetProject() *ProjectConfig {
  95         prj := p.server.FolderToProject(*p.folder)
  96         prj.ServerID = p.server.ID
  97         return &prj
  98 }
  99
 100 // UpdateProject Set project config
 101 func (p *PathMap) UpdateProject(prj ProjectConfig) (*ProjectConfig, error) {
 102         p.folder = p.server.ProjectToFolder(prj)
 103         np := p.GetProject()
 104         if err := p.events.Emit(EVTProjectChange, np); err != nil {
 105                 return np, err
 106         }
 107         return np, nil
 108 }
 109
 110 // GetServer Get the XdsServer that holds this project
 111 func (p *PathMap) GetServer() *XdsServer {
 112         return p.server
 113 }
 114
 115 // Sync Force project files synchronization
 116 func (p *PathMap) Sync() error {
 117         return nil
 118 }
 119
 120 // IsInSync Check if project files are in-sync
 121 func (p *PathMap) IsInSync() (bool, error) {
 122         return true, nil
 123 }