From 6f14f57ae4b303e486b689c0c24c376eacd4eabd Mon Sep 17 00:00:00 2001
From: Dave Henderson <dhenderson@gmail.com>
Date: Tue, 31 Oct 2017 10:26:21 -0400
Subject: Add time.ParseLocal function

Signed-off-by: Dave Henderson <dhenderson@gmail.com>
---
 docs/content/functions/time.md | 67 ++++++++++++++++++++++++++++++++++++++++--
 funcs/time.go                  | 14 +++++++++
 test/integration/time.bats     | 12 ++++++++
 3 files changed, 91 insertions(+), 2 deletions(-)

diff --git a/docs/content/functions/time.md b/docs/content/functions/time.md
index 079f0ece..2bd342a2 100644
--- a/docs/content/functions/time.md
+++ b/docs/content/functions/time.md
@@ -10,6 +10,8 @@ few of the functions return a `time.Time` value. All of the
 [`time.Time` functions](https://golang.org/pkg/time/#Time) can then be used to
 convert, adjust, or format the time in your template.
 
+### Reference time
+
 An important difference between this and many other time/date utilities is how
 parsing and formatting is accomplished. Instead of relying solely on pre-defined
 formats, or having a complex system of variables, formatting is accomplished by
@@ -109,17 +111,78 @@ A number of pre-defined layouts are provided as constants, defined
 Just like [`time.Now`](#time-now), this is usually used in conjunction with
 other functions.
 
+_Note: In the absence of a time zone indicator, `time.Parse` returns a time in UTC._
+
 ### Usage
 ```go
 time.Parse layout timestamp
 ```
 
+### Arguments
+
+| name   | description |
+|--------|-------|
+| `layout` | The layout string to parse with|
+| `timestamp` | The timestamp to parse |
+
+### Examples
+
+Usage with [`Format`](https://golang.org/pkg/time/#Time.Format):
+```console
+$ gomplate -i '{{ (time.Parse "2006-01-02" "1993-10-23").Format "Monday January 2, 2006 MST" }}'
+Saturday October 23, 1993 UTC
+```
+
+## `time.ParseLocal`
+
+Same as [`time.Parse`](#time-parse), except that in the absence of a time zone
+indicator, the timestamp wil be parsed in the local timezone.
+
+### Usage
+```go
+time.ParseLocal layout timestamp
+```
+
+### Arguments
+
+| name   | description |
+|--------|-------|
+| `layout` | The layout string to parse with|
+| `timestamp` | The timestamp to parse |
+
+### Examples
+
+Usage with [`Format`](https://golang.org/pkg/time/#Time.Format):
+```console
+$ bin/gomplate -i '{{ (time.ParseLocal time.Kitchen "6:00AM").Format "15:04 MST" }}'
+06:00 EST
+```
+
+## `time.ParseInLocation`
+
+Same as [`time.Parse`](#time-parse), except that the time is parsed in the given location's time zone.
+
+This wraps [`time.ParseInLocation`](https://golang.org/pkg/time/#ParseInLocation).
+
+### Usage
+```go
+time.ParseInLocation layout location timestamp
+```
+
+### Arguments
+
+| name   | description |
+|--------|-------|
+| `layout` | The layout string to parse with |
+| `location` | The location to parse in |
+| `timestamp` | The timestamp to parse |
+
 ### Examples
 
 Usage with [`Format`](https://golang.org/pkg/time/#Time.Format):
 ```console
-$ gomplate -i '{{ (time.Parse "2006-01-02" "1993-10-23").Format "Monday January 2, 2006" }}'
-Saturday October 23, 1993
+$ gomplate -i '{{ (time.ParseInLocation time.Kitchen "Africa/Luanda" "6:00AM").Format "15:04 MST" }}'
+06:00 LMT
 ```
 
 ## `time.Unix`
diff --git a/funcs/time.go b/funcs/time.go
index c4e03deb..d3f55121 100644
--- a/funcs/time.go
+++ b/funcs/time.go
@@ -78,6 +78,20 @@ func (f *TimeFuncs) Parse(layout, value string) (gotime.Time, error) {
 	return gotime.Parse(layout, value)
 }
 
+// ParseLocal -
+func (f *TimeFuncs) ParseLocal(layout, value string) (gotime.Time, error) {
+	return gotime.ParseInLocation(layout, value, gotime.Local)
+}
+
+// ParseInLocation -
+func (f *TimeFuncs) ParseInLocation(layout, location, value string) (gotime.Time, error) {
+	loc, err := gotime.LoadLocation(location)
+	if err != nil {
+		return gotime.Time{}, err
+	}
+	return gotime.ParseInLocation(layout, value, loc)
+}
+
 // Now -
 func (f *TimeFuncs) Now() gotime.Time {
 	return gotime.Now()
diff --git a/test/integration/time.bats b/test/integration/time.bats
index 3d91d2f6..83441f60 100644
--- a/test/integration/time.bats
+++ b/test/integration/time.bats
@@ -27,6 +27,18 @@ load helper
   [[ "${output}" == "2009-02-13 23 +0000" ]]
 }
 
+@test "'(time.ParseLocal).Format'" {
+  TZ=Africa/Luanda gomplate -i "{{ (time.ParseLocal time.Kitchen \"6:00AM\").Format \"15:04 MST\" }}"
+  [ "$status" -eq 0 ]
+  [[ "${output}" == "06:00 LMT" ]]
+}
+
+@test "'(time.ParseInLocation).Format'" {
+  gomplate -i "{{ (time.ParseInLocation time.Kitchen \"Africa/Luanda\" \"6:00AM\").Format \"15:04 MST\" }}"
+  [ "$status" -eq 0 ]
+  [[ "${output}" == "06:00 LMT" ]]
+}
+
 @test "'(time.Unix).UTC.Format' int" {
   gomplate -i '{{ (time.Unix 1234567890).UTC.Format "2006-01-02 15 -0700" }}'
   [ "$status" -eq 0 ]
-- 
cgit v1.2.3