281: Support toTimestamp template function (#282)

WillAnd31 · web-flow · commit 5445cd52997e · 2023-10-22T21:56:24.000+09:00
Co-authored-by: Will Anderson &lt;will.anderson@vivint.com&gt;
diff --git a/README.md b/README.md
@@ -160,6 +160,7 @@ functions](https://golang.org/pkg/text/template/#hdr-Functions)):
 | `extjson`       | `string`              | Parse the object as json and output colorized json                                |
 | `ppextjson`     | `string`              | Parse the object as json and output pretty-print colorized json                   |
 | `toRFC3339Nano` | `object`              | Parse timestamp (string, int, json.Number) and output it using RFC3339Nano format |
+| `toTimestamp`   | `object, string [, string]` | Parse timestamp (string, int, json.Number) and output it using the given layout in the timezone that is optionally given (defaults to UTC). |
 | `levelColor`    | `string`              | Print log level using appropriate color                                           |
 | `colorBlack`    | `string`              | Print text using black color                                                      |
 | `colorRed`      | `string`              | Print text using red color                                                        |
diff --git a/cmd/cmd.go b/cmd/cmd.go
@@ -488,6 +488,24 @@ func (o *options) generateTemplate() (*template.Template, error) {
 		"toUTC": func(ts any) time.Time {
 			return cast.ToTime(ts).UTC()
 		},
+		"toTimestamp": func(ts any, layout string, optionalTZ ...string) (string, error) {
+			t, parseErr := cast.ToTimeE(ts)
+			if parseErr != nil {
+				return "", parseErr
+			}
+
+			var tz string
+			if len(optionalTZ) > 0 {
+				tz = optionalTZ[0]
+			}
+
+			loc, loadErr := time.LoadLocation(tz)
+			if loadErr != nil {
+				return "", loadErr
+			}
+
+			return t.In(loc).Format(layout), nil
+		},
 		"color": func(color color.Color, text string) string {
 			return color.SprintFunc()(text)
 		},
diff --git a/cmd/cmd_test.go b/cmd/cmd_test.go
@@ -376,6 +376,28 @@ func TestOptionsGenerateTemplate(t *testing.T) {
 			"pod1 container1 [1970-01-01T00:02:03Z] INFO template message",
 			false,
 		},
+		{
+			"template-to-timestamp-with-timezone",
+			func() *options {
+				o := NewOptions(streams)
+				o.template = `{{ toTimestamp .Message "Jan 02 2006 15:04 MST" "US/Eastern" }}`
+				return o
+			}(),
+			`2024-01-01T05:00:00`,
+			`Jan 01 2024 00:00 EST`,
+			false,
+		},
+		{
+			"template-to-timestamp-without-timezone",
+			func() *options {
+				o := NewOptions(streams)
+				o.template = `{{ toTimestamp .Message "Jan 02 2006 15:04 MST" }}`
+				return o
+			}(),
+			`2024-01-01T05:00:00`,
+			`Jan 01 2024 05:00 UTC`,
+			false,
+		},
 	}
 
 	for _, tt := range tests {
