From 7c1f07702aa0e60e3c8ec2150354b466bd2f6fa7 Mon Sep 17 00:00:00 2001
From: Lev Berman <ldmberman@gmail.com>
Date: Wed, 30 Mar 2016 13:07:55 +0300
Subject: [PATCH] #48 - Extract snapshot IDs from the server links & include
 them in the output

---
 models/server/entities.go | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/models/server/entities.go b/models/server/entities.go
index 60832a0..c068d8c 100644
--- a/models/server/entities.go
+++ b/models/server/entities.go
@@ -1,6 +1,9 @@
 package server
 
 import (
+	"encoding/json"
+	"strings"
+
 	"github.com/centurylinkcloud/clc-go-cli/models"
 	"github.com/centurylinkcloud/clc-go-cli/models/alert"
 	"github.com/centurylinkcloud/clc-go-cli/models/customfields"
@@ -56,5 +59,41 @@ type IPAddresses struct {
 
 type Snapshot struct {
 	Name  string
+	Id    string
 	Links []models.LinkEntity
 }
+
+func (s *Snapshot) UnmarshalJSON(data []byte) error {
+	// Here we are getting the snapshot ID out of the server links. Looks
+	// rather hacky but there doesn't seem to be another way for retrieving it.
+	id := func() string {
+		m := struct {
+			Links []struct {
+				Rel  string
+				Href string
+			}
+		}{}
+		if err := json.Unmarshal(data, &m); err != nil {
+			return ""
+		}
+		for _, link := range m.Links {
+			if link.Rel != "self" {
+				continue
+			}
+			tokens := strings.Split(link.Href, "/")
+			return tokens[len(tokens)-1]
+		}
+		return ""
+	}()
+
+	if id != "" {
+		s.Id = id
+	}
+
+	type SimpleSnapshot Snapshot
+	return json.Unmarshal(data, &struct {
+		*SimpleSnapshot
+	}{
+		SimpleSnapshot: (*SimpleSnapshot)(s),
+	})
+}