From fb384bfa492d282e4b4488e683fa3096a26c46c9 Mon Sep 17 00:00:00 2001
From: Stefan Prodan <stefan.prodan@gmail.com>
Date: Sat, 6 Jan 2018 12:45:32 +0200
Subject: [PATCH] pod labels and annotations

---
 pkg/server/response.go |  7 +++---
 pkg/server/server.go   | 49 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 53 insertions(+), 3 deletions(-)

diff --git a/pkg/server/response.go b/pkg/server/response.go
index e0d2986..0522d0b 100644
--- a/pkg/server/response.go
+++ b/pkg/server/response.go
@@ -1,7 +1,8 @@
 package server
 
 type Response struct {
-	Environment []string `json:"environment"`
-	Runtime map[string]string `json:"runtime"`
+	Environment []string          `json:"environment"`
+	Runtime     map[string]string `json:"runtime"`
+	Labels      []string          `json:"labels"`
+	Annotations []string          `json:"annotations"`
 }
-
diff --git a/pkg/server/server.go b/pkg/server/server.go
index 30be990..39d051e 100644
--- a/pkg/server/server.go
+++ b/pkg/server/server.go
@@ -1,11 +1,14 @@
 package server
 
 import (
+	"io/ioutil"
 	"net/http"
 	"os"
+	"path/filepath"
 	"runtime"
 	"strconv"
 
+	"github.com/pkg/errors"
 	"gopkg.in/yaml.v2"
 )
 
@@ -37,9 +40,23 @@ func (s *Server) index(w http.ResponseWriter, r *http.Request) {
 	}
 	runtime["hostname"], _ = os.Hostname()
 
+	labels, err := readFiles("/etc/podinfod/metadata/labels")
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		w.Write([]byte(err.Error()))
+	}
+
+	annotations, err := readFiles("/etc/podinfod/metadata/annotations")
+	if err != nil {
+		w.WriteHeader(http.StatusInternalServerError)
+		w.Write([]byte(err.Error()))
+	}
+
 	resp := &Response{
 		Environment: os.Environ(),
 		Runtime:     runtime,
+		Labels:      labels,
+		Annotations: annotations,
 	}
 
 	d, err := yaml.Marshal(resp)
@@ -62,3 +79,35 @@ func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 
 	s.mux.ServeHTTP(w, r)
 }
+
+func readFiles(dir string) ([]string, error) {
+	files := []string{}
+	err := filepath.Walk(dir, func(path string, f os.FileInfo, err error) error {
+		files = append(files, path)
+
+		return nil
+	})
+	if err != nil {
+		return nil, errors.Wrapf(err, "Reading from %v failed", dir)
+	}
+	list := make([]string, 0)
+	for _, path := range files {
+		data, err := ioutil.ReadFile(path)
+		if err != nil {
+			return nil, errors.Wrapf(err, "Reading %v failed", path)
+		}
+		content := string(data)
+		duplicate := false
+		for _, p := range list {
+			if p == content {
+				duplicate = true
+				break
+			}
+		}
+		if duplicate {
+			continue
+		}
+		list = append(list, content)
+	}
+	return list, nil
+}