Add new functions parsing with location

Author: noah

parser.go

@@ -4,6 +4,7 @@ import (
 	"fmt"
 	"strconv"
 	"strings"
+	"time"
 )
 
 type (
@@ -55,6 +56,36 @@ var (
 	}
 )
 
+func MustParseInLocation(expr string, locName string) *Schedule {
+	loc, err := time.LoadLocation(locName)
+	if err != nil {
+		panic(err)
+	}
+
+	schdule, err := Parse(expr)
+	if err != nil {
+		panic(err)
+	}
+
+	schdule.Location = loc
+	return schdule
+}
+
+func ParseInLocation(expr string, locName string) (*Schedule, error) {
+	loc, err := time.LoadLocation(locName)
+	if err != nil {
+		return nil, err
+	}
+
+	schdule, err := Parse(expr)
+	if err != nil {
+		return nil, err
+	}
+
+	schdule.Location = loc
+	return schdule, nil
+}
+
 func MustParse(expr string) *Schedule {
 	s, err := Parse(expr)
 	if err != nil {

schedule.go

@@ -8,13 +8,17 @@ type (
 	Schedule struct {
 		Minute, Hour, Dom, Month, Dow bitset
 
-		// TODO: support the timezone option
-		// Location *time.Location
+		Location *time.Location
 	}
 )
 
 func (s *Schedule) Next(t time.Time) time.Time {
 	loc := time.UTC
+	if s.Location != nil {
+		loc = s.Location
+	}
+
+	t.In(loc)
 
 	added := false
 
@@ -93,6 +97,11 @@ L:
 
 func (s *Schedule) Prev(t time.Time) time.Time {
 	loc := time.UTC
+	if s.Location != nil {
+		loc = s.Location
+	}
+
+	t.In(loc)
 
 	subtracted := false