lib/agent/project-pathmap.go

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