diff --git a/pkg/cluster/cluster_scan.go b/pkg/cluster/cluster_scan.go index 4b453d4..4886af7 100644 --- a/pkg/cluster/cluster_scan.go +++ b/pkg/cluster/cluster_scan.go @@ -12,6 +12,10 @@ const ( // ZoneRegionsFileName indicates the file containing // region names as references ZoneRegionsFileName = "regions" + + // RegionClusterTokenFileName contains the kubernetes + // token of the cluster this region represents + RegionClusterTokenFileName = "k8s_token" ) func (m *Cluster) scan(opts *ScanOptions) error { @@ -186,6 +190,8 @@ func (z *Zone) scan() error { switch { case name == ZoneRegionsFileName: err = z.loadRegions() + case name == RegionClusterTokenFileName: + err = z.loadClusterToken() case e.IsDir(): err = z.scanSubdirectory(name) default: @@ -218,6 +224,28 @@ func (z *Zone) loadRegions() error { return err } +func (z *Zone) loadClusterToken() error { + filename := path.Join(z.Name, RegionClusterTokenFileName) + lines, err := z.zones.ReadLines(filename) + if err == nil { + var token string + + for _, s := range lines { + if s != "" { + token = s + break + } + } + + err = z.zones.setRegionClusterToken(z.Name, token) + if err != nil { + err = core.Wrap(err, filename) + } + } + + return err +} + func (z *Zone) scanSubdirectory(name string) error { m := &Machine{ zone: z, diff --git a/pkg/cluster/regions.go b/pkg/cluster/regions.go index a79ced5..e2215a5 100644 --- a/pkg/cluster/regions.go +++ b/pkg/cluster/regions.go @@ -16,6 +16,7 @@ type Region struct { zones []*Zone Name string + Cluster *string `json:",omitempty" yaml:",omitempty"` Regions []string `json:",omitempty" yaml:",omitempty"` } @@ -106,6 +107,26 @@ func (m *Cluster) setRegionZones(name string, zones ...*Zone) { }) } +func (m *Cluster) setRegionClusterToken(name string, token string) error { + for i := range m.Regions { + r := &m.Regions[i] + + if r.Name == name { + // found + r.Cluster = &token + return nil + } + } + + // new + m.Regions = append(m.Regions, Region{ + m: m, + Name: name, + Cluster: &token, + }) + return nil +} + func (m *Cluster) appendRegionRegions(name string, subs ...string) { for i := range m.Regions { r := &m.Regions[i] @@ -210,9 +231,25 @@ func (z *Zone) syncZoneRegions() error { // SyncRegions writes to the file system the regions covered // by this meta-region func (r *Region) SyncRegions() error { + if err := r.syncRegionsFile(); err != nil { + return err + } + + return r.syncClusterFile() +} + +func (r *Region) mkdir() error { + return r.m.MkdirAll(r.Name) +} + +func (r *Region) syncRegionsFile() error { + var err error + name := filepath.Join(r.Name, ZoneRegionsFileName) - if len(r.Regions) > 0 { + if len(r.Regions) == 0 { + err = r.m.RemoveFile(name) + } else if err = r.mkdir(); err == nil { var buf bytes.Buffer for _, s := range r.Regions { @@ -220,12 +257,29 @@ func (r *Region) SyncRegions() error { _, _ = buf.WriteRune('\n') } - if err := r.m.MkdirAll(r.Name); err != nil { - return err + err = r.m.WriteStringFile(buf.String(), name) + } + + return err +} + +func (r *Region) syncClusterFile() error { + var err error + + name := filepath.Join(r.Name, RegionClusterTokenFileName) + + if r.Cluster == nil { + err = r.m.RemoveFile(name) + } else if err = r.mkdir(); err == nil { + var buf bytes.Buffer + + _, _ = buf.WriteString(*r.Cluster) + if buf.Len() > 0 { + _, _ = buf.WriteRune('\n') } - return r.m.WriteStringFile(buf.String(), name) + err = r.m.WriteStringFile(buf.String(), name) } - return r.m.RemoveFile(name) + return err }