From ccaeb4eec86e6c7d063753a5deacaef107b567df Mon Sep 17 00:00:00 2001
From: Tom Withers <thomas.withers@digital.justice.gov.uk>
Date: Fri, 19 Jun 2020 14:21:36 +0100
Subject: [PATCH] Add ability to get expired workspaces

---
 main.go | 60 +++++++++++++++++++++++++++++++++++++++++++++++----------
 1 file changed, 50 insertions(+), 10 deletions(-)

diff --git a/main.go b/main.go
index 0ea7db1..3ae105c 100644
--- a/main.go
+++ b/main.go
@@ -42,15 +42,15 @@ func main()  {
 		flag.Usage()
 	}
 
-	if workspaceName == "" {
-		fmt.Println("Error: Workspace not passed")
-		flag.Usage()
-	} else {
-		PutWorkspace(&workspaceName, &awsAccountId, &awsIAMRoleName)
+	if expiredWorkspaces {
+		GetExpiredWorkspaces(&awsAccountId, &awsIAMRoleName)
 	}
 
-	if expiredWorkspaces {
-		GetExpiredWorkspaces()
+	if workspaceName != "" && expiredWorkspaces == true{
+		PutWorkspace(&workspaceName, &awsAccountId, &awsIAMRoleName)
+	} else {
+		fmt.Println("Error: Workspace not passed")
+		flag.Usage()
 	}
 }
 
@@ -100,6 +100,46 @@ func PutWorkspace(workspace *string, accountId *string, iamRoleName *string) {
 	fmt.Println("Successfully added '" + item.WorkspaceName + "' with TTL " + strconv.FormatInt(item.ExpiresTTL, 10) + " for workspace cleanup")
 }
 
-func GetExpiredWorkspaces() {
-	fmt.Println("Not yet implemented")
-}
\ No newline at end of file
+func GetExpiredWorkspaces(accountId *string, iamRoleName *string) {
+	type Item struct {
+		WorkspaceName string
+	}
+
+	sess, err := session.NewSession()
+	if err != nil {
+		log.Fatalln(err)
+	}
+
+	RoleArn := "arn:aws:iam::" + *accountId + ":role/" + *iamRoleName
+
+	creds := stscreds.NewCredentials(sess, RoleArn)
+	awsConfig := aws.Config{Credentials: creds, Region: aws.String("eu-west-1")}
+
+	svc := dynamodb.New(sess, &awsConfig)
+
+	params := &dynamodb.ScanInput{
+		TableName: aws.String("WorkspaceCleanup"),
+	}
+
+	result, err := svc.Scan(params)
+	if err != nil {
+		exitWithError(fmt.Errorf("failed to make Query API call, %v", err))
+	}
+
+	items := []Item{}
+
+	err = dynamodbattribute.UnmarshalListOfMaps(result.Items, &items)
+	if err != nil {
+		exitWithError(fmt.Errorf("failed to unmarshal Query result items, %v", err))
+	}
+
+	for i, item := range items {
+		fmt.Print(item.WorkspaceName, " ")
+		i++
+	}
+}
+
+func exitWithError(err error) {
+	fmt.Fprintln(os.Stderr, err)
+	os.Exit(1)
+}