From bd5c4ad55e3d5183e9d72b2d1e3e1365bd7f4f32 Mon Sep 17 00:00:00 2001
From: Don <don.j.olmstead@gmail.com>
Date: Mon, 9 Sep 2019 11:06:21 -0700
Subject: [PATCH] Add parsing from the environment

---
 pkg/plugin/env.go   | 88 +++++++++++++++++++++++++++++++++++++++++++++
 pkg/plugin/types.go | 27 ++++++++++++++
 2 files changed, 115 insertions(+)

diff --git a/pkg/plugin/env.go b/pkg/plugin/env.go
index d3dca8b..33f9c7a 100644
--- a/pkg/plugin/env.go
+++ b/pkg/plugin/env.go
@@ -5,6 +5,13 @@
 
 package plugin
 
+import (
+	"os"
+	"strconv"
+	"strings"
+	"time"
+)
+
 // List of enviornment variables set by Drone when running a step within a
 // build stage.
 //
@@ -53,3 +60,84 @@ const (
 	// StepNumberEnvVar corresponds to Step.Number.
 	StepNumberEnvVar = "DRONE_STEP_NUMBER"
 )
+
+// StringEnvVar gets the environment variable's value.
+//
+// If the value is not present then this returns the empty string.
+func StringEnvVar(envVar string) string {
+	return envVarValue(envVar)
+}
+
+// StringSliceEnvVar gets the environment variable as a string slice.
+//
+// If the value is not present then this returns an empty slice.
+func StringSliceEnvVar(envVar string) []string {
+	return strings.Split(envVarValue(envVar), ",")
+}
+
+// IntEnvVar gets the environment variable as an int.
+//
+// If the value is not present then this returns 0.
+func IntEnvVar(envVar string) int {
+	return int(Int64EnvVar(envVar))
+}
+
+// Int64EnvVar gets the environment variable as an int64.
+//
+// If the value is not present then this returns 0.
+func Int64EnvVar(envVar string) int64 {
+	value, err := strconv.ParseInt(envVarValue(envVar), 0, 64)
+
+	if err != nil {
+		return 0
+	}
+
+	return value
+}
+
+// UintEnvVar gets the environment variable as an uint.
+//
+// If the value is not present then this returns 0.
+func UintEnvVar(envVar string) uint {
+	return uint(Uint64EnvVar(envVar))
+}
+
+// Uint64EnvVar gets the environment variable as an uint64.
+//
+// If the value is not present then this returns 0.
+func Uint64EnvVar(envVar string) uint64 {
+	value, err := strconv.ParseUint(envVarValue(envVar), 0, 64)
+
+	if err != nil {
+		return 0
+	}
+
+	return value
+}
+
+// BoolEnvVar gets the environment variable as a bool.
+//
+// If the value is not present then this returns false.
+func BoolEnvVar(envVar string) bool {
+	value, err := strconv.ParseBool(envVarValue(envVar))
+
+	if err != nil {
+		return false
+	}
+
+	return value
+}
+
+// TimeEnvVar gets the environment variable as a Time created from a unix
+// timestamp.
+//
+// If the value is not present then this returns time.Unix(0).
+func TimeEnvVar(envVar string) time.Time {
+	return time.Unix(Int64EnvVar(envVar), 0)
+}
+
+// envVarValue returns the first matched environment variable or the empty
+// string if none was found.
+func envVarValue(envVar string) string {
+	return os.Getenv(envVar)
+}
diff --git a/pkg/plugin/types.go b/pkg/plugin/types.go
index da94b0e..4c063dc 100644
--- a/pkg/plugin/types.go
+++ b/pkg/plugin/types.go
@@ -87,3 +87,30 @@ type (
 		Number int
 	}
 )
+
+// StageFromEnv creates a Stage from the environment variables used by Drone.
+func StageFromEnv() Stage {
+	return Stage{
+		Arch:      StringEnvVar(StageArchEnvVar),
+		DependsOn: StringSliceEnvVar(StageDependsOnEnvVar),
+		Finished:  TimeEnvVar(StageFinishedEnvVar),
+		Kind:      StringEnvVar(StageKindEnvVar),
+		Machine:   StringEnvVar(StageMachineEnvVar),
+		Name:      StringEnvVar(StageNameEnvVar),
+		Number:    IntEnvVar(StageNumberEnvVar),
+		OS:        StringEnvVar(StageOSEnvVar),
+		Started:   TimeEnvVar(StageStartedEnvVar),
+		Status:    StringEnvVar(StageStatusEnvVar),
+		Type:      StringEnvVar(StageTypeEnvVar),
+		Variant:   StringEnvVar(StageVariantEnvVar),
+		Version:   StringEnvVar(StageVersionEnvVar),
+	}
+}
+
+// StepFromEnv creates a Step from the environment variables used by Drone.
+func StepFromEnv() Step {
+	return Step{
+		Name:   StringEnvVar(StepNameEnvVar),
+		Number: IntEnvVar(StepNumberEnvVar),
+	}
+}