Adding timezone formatting option

Fixes #315.

Author: kenshaw

env/types.go

@@ -117,6 +117,10 @@ var pvarNames = []varName{
 		"time",
 		`format used to display time/date column values (default "RFC3339Nano")`,
 	},
+	{
+		"timezone",
+		`the timezone to display dates in (default '')`,
+	},
 	{
 		"title",
 		"set the table title for subsequently printed tables",
@@ -294,6 +298,7 @@ func init() {
 		"recordsep_zero":           "off",
 		"tableattr":                "",
 		"time":                     "RFC3339Nano",
+		"timezone":                 "",
 		"title":                    "",
 		"tuples_only":              "off",
 		"unicode_border_linestyle": "single",
@@ -476,7 +481,7 @@ func Ptoggle(name, extra string) (string, error) {
 			pvars[name] = "aligned"
 		}
 	case "linestyle":
-	case "csv_fieldsep", "fieldsep", "null", "recordsep", "time", "locale":
+	case "csv_fieldsep", "fieldsep", "null", "recordsep", "time", "timezone", "locale":
 	case "tableattr", "title":
 		pvars[name] = ""
 	case "unicode_border_linestyle", "unicode_column_linestyle", "unicode_header_linestyle":
@@ -526,6 +531,11 @@ func Pset(name, value string) (string, error) {
 		pvars[name] = value
 	case "csv_fieldsep", "fieldsep", "null", "recordsep", "tableattr", "time", "title", "locale":
 		pvars[name] = value
+	case "timezone":
+		if _, err := time.LoadLocation(value); err != nil {
+			return "", text.ErrInvalidTimezoneLocation
+		}
+		pvars[name] = value
 	case "unicode_border_linestyle", "unicode_column_linestyle", "unicode_header_linestyle":
 		if !borderRE.MatchString(value) {
 			return "", text.ErrInvalidFormatBorderLineStyle

go.mod

@@ -58,7 +58,7 @@ require (
 	github.com/uber/athenadriver v1.1.15
 	github.com/vertica/vertica-sql-go v1.3.3
 	github.com/xo/dburl v0.22.0
-	github.com/xo/tblfmt v0.13.0
+	github.com/xo/tblfmt v0.13.1
 	github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e
 	github.com/ydb-platform/ydb-go-sdk/v3 v3.61.2
 	github.com/yookoala/realpath v1.0.0

go.sum

@@ -1039,8 +1039,8 @@ github.com/xeipuuv/gojsonschema v1.2.0/go.mod h1:anYRn/JVcOK2ZgGU+IjEV4nwlhoK5sQ
 github.com/xo/dburl v0.22.0 h1:sO5WLm2ywMzyiLxEcLBlw5AyKvdR5hirq9U7s3fCoeM=
 github.com/xo/dburl v0.22.0/go.mod h1:B7/G9FGungw6ighV8xJNwWYQPMfn3gsi2sn5SE8Bzco=
 github.com/xo/tblfmt v0.0.0-20190609041254-28c54ec42ce8/go.mod h1:3U5kKQdIhwACye7ml3acccHmjGExY9WmUGU7rnDWgv0=
-github.com/xo/tblfmt v0.13.0 h1:mh0ilx9rKZNoMqGwvyMpQv3MED+X6te6ZqB+BzSypck=
-github.com/xo/tblfmt v0.13.0/go.mod h1:BLPC+dRy68cgSK/mPgQRfFQ/xLg231Fyic178ybjB34=
+github.com/xo/tblfmt v0.13.1 h1:lhQAH5OSPhKCEvmh4suPe52woLuF2npX2TLB91CiPZ4=
+github.com/xo/tblfmt v0.13.1/go.mod h1:BLPC+dRy68cgSK/mPgQRfFQ/xLg231Fyic178ybjB34=
 github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e h1:JVG44RsyaB9T2KIHavMF/ppJZNG9ZpyihvCd0w101no=
 github.com/xo/terminfo v0.0.0-20220910002029-abceb7e1c41e/go.mod h1:RbqR21r5mrJuqunuUZ/Dhy/avygyECGrLceyNeo4LiM=
 github.com/xwb1989/sqlparser v0.0.0-20180606152119-120387863bf2 h1:zzrxE1FKn5ryBNl9eKOeqQ58Y/Qpo3Q9QNxKHX5uzzQ=

text/errors.go

@@ -55,6 +55,8 @@ var (
 	ErrInvalidFormatLineStyle = errors.New(`\pset: allowed line styles are ascii, old-ascii, unicode`)
 	// ErrInvalidFormatBorderLineStyle is the invalid format border line style error.
 	ErrInvalidFormatBorderLineStyle = errors.New(`\pset: allowed Unicode border line styles are single, double`)
+	// ErrInvalidTimezoneLocation is the invalid timezone location error.
+	ErrInvalidTimezoneLocation = errors.New(`\pset: invalid timezone location`)
 	// ErrInvalidQuotedString is the invalid quoted string error.
 	ErrInvalidQuotedString = errors.New(`invalid quoted string`)
 	// ErrInvalidFormatOption is the invalid format option error.