|
|
@ -5,15 +5,16 @@ import (
|
|
|
|
"flag"
|
|
|
|
"flag"
|
|
|
|
"fmt"
|
|
|
|
"fmt"
|
|
|
|
"io/ioutil"
|
|
|
|
"io/ioutil"
|
|
|
|
"log"
|
|
|
|
|
|
|
|
"net/http"
|
|
|
|
"net/http"
|
|
|
|
"net/url"
|
|
|
|
"net/url"
|
|
|
|
"os"
|
|
|
|
"os"
|
|
|
|
"os/user"
|
|
|
|
"os/user"
|
|
|
|
"path/filepath"
|
|
|
|
"path/filepath"
|
|
|
|
"strings"
|
|
|
|
"strings"
|
|
|
|
|
|
|
|
"syscall"
|
|
|
|
"time"
|
|
|
|
"time"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"golang.org/x/net/context"
|
|
|
|
"golang.org/x/net/context"
|
|
|
|
"golang.org/x/oauth2"
|
|
|
|
"golang.org/x/oauth2"
|
|
|
|
"golang.org/x/oauth2/google"
|
|
|
|
"golang.org/x/oauth2/google"
|
|
|
@ -21,13 +22,23 @@ import (
|
|
|
|
)
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
var (
|
|
|
|
var (
|
|
|
|
nowTime = time.Now()
|
|
|
|
nowTime = time.Now()
|
|
|
|
oneYear, _ = time.ParseDuration("365d")
|
|
|
|
oneYearString = fmt.Sprintf("%dh", 365*24)
|
|
|
|
clientSecretFlag = flag.String("s", "client_secret.json", "client secret json file")
|
|
|
|
oneYear, _ = time.ParseDuration(oneYearString)
|
|
|
|
calendarIDFlag = flag.String("i", "primary", "calendar ID to search")
|
|
|
|
clientSecretFlag = flag.String(
|
|
|
|
endTimeFlag = flag.String("e", nowTime.Format(time.RFC3339), "end time for search")
|
|
|
|
"s", getEnvDefault("CLIENT_SECRET", "client_secret.json"), "client secret json file")
|
|
|
|
timeWindowFlag = flag.Duration("t", oneYear, "time window duration from before end time")
|
|
|
|
calendarIDFlag = flag.String(
|
|
|
|
defaultLocationFlag = flag.String("l", "notset", "default location for where")
|
|
|
|
"i", getEnvDefault("CALENDAR", "primary"), "calendar ID to search")
|
|
|
|
|
|
|
|
endTimeFlag = flag.String(
|
|
|
|
|
|
|
|
"e", getEnvDefault("END_TIME", nowTime.Format(dateFmt)), "end time for search")
|
|
|
|
|
|
|
|
timeWindowFlag = flag.Duration(
|
|
|
|
|
|
|
|
"t", func() time.Duration {
|
|
|
|
|
|
|
|
stringVal := getEnvDefault("DURATION", oneYearString)
|
|
|
|
|
|
|
|
val, _ := time.ParseDuration(stringVal)
|
|
|
|
|
|
|
|
return val
|
|
|
|
|
|
|
|
}(), "time window duration from before end time")
|
|
|
|
|
|
|
|
defaultLocationFlag = flag.String(
|
|
|
|
|
|
|
|
"l", getEnvDefault("DEFAULT_LOCATION", "notset"), "default location for where")
|
|
|
|
|
|
|
|
|
|
|
|
locationAliases = map[string]string{
|
|
|
|
locationAliases = map[string]string{
|
|
|
|
"pittsburgh": "dan",
|
|
|
|
"pittsburgh": "dan",
|
|
|
@ -40,6 +51,15 @@ const (
|
|
|
|
dateFmt = "2006-01-02"
|
|
|
|
dateFmt = "2006-01-02"
|
|
|
|
)
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
func getEnvDefault(key, defVal string) string {
|
|
|
|
|
|
|
|
for _, k := range []string{"WHEREWHEN_" + key, key} {
|
|
|
|
|
|
|
|
if v, ok := syscall.Getenv(k); ok {
|
|
|
|
|
|
|
|
return v
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
return defVal
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
func main() {
|
|
|
|
func main() {
|
|
|
|
flag.Parse()
|
|
|
|
flag.Parse()
|
|
|
|
|
|
|
|
|
|
|
@ -47,23 +67,23 @@ func main() {
|
|
|
|
|
|
|
|
|
|
|
|
b, err := ioutil.ReadFile(*clientSecretFlag)
|
|
|
|
b, err := ioutil.ReadFile(*clientSecretFlag)
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalf("Unable to read client secret file: %v", err)
|
|
|
|
logrus.Fatalf("Unable to read client secret file: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
config, err := google.ConfigFromJSON(b, calendar.CalendarReadonlyScope)
|
|
|
|
config, err := google.ConfigFromJSON(b, calendar.CalendarReadonlyScope)
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalf("Unable to parse client secret file to config: %v", err)
|
|
|
|
logrus.Fatalf("Unable to parse client secret file to config: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
client := getClient(ctx, config)
|
|
|
|
client := getClient(ctx, config)
|
|
|
|
|
|
|
|
|
|
|
|
srv, err := calendar.New(client)
|
|
|
|
srv, err := calendar.New(client)
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalf("Unable to retrieve calendar Client %v", err)
|
|
|
|
logrus.Fatalf("Unable to retrieve calendar Client %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
endT, err := time.Parse(time.RFC3339, *endTimeFlag)
|
|
|
|
endT, err := time.Parse(dateFmt, *endTimeFlag)
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalf("Unable to parse end time %q: %v", *endTimeFlag, err)
|
|
|
|
logrus.Fatalf("Unable to parse end time %q: %v", *endTimeFlag, err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
maxResults := int64(2500)
|
|
|
|
maxResults := int64(2500)
|
|
|
@ -73,17 +93,17 @@ func main() {
|
|
|
|
SingleEvents(true).TimeMin(t).TimeMax(endT.Format(time.RFC3339)).
|
|
|
|
SingleEvents(true).TimeMin(t).TimeMax(endT.Format(time.RFC3339)).
|
|
|
|
MaxResults(maxResults).OrderBy("startTime").Do()
|
|
|
|
MaxResults(maxResults).OrderBy("startTime").Do()
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalf("Unable to retrieve next %d events. %v", maxResults, err)
|
|
|
|
logrus.Fatalf("Unable to retrieve next %d events. %v", maxResults, err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
log.Printf("found %d events", len(events.Items))
|
|
|
|
logrus.WithField("count", len(events.Items)).Infof("found events")
|
|
|
|
log.Printf("generating days between %v and %v", startT, endT)
|
|
|
|
logrus.Debugf("generating days between %v and %v", startT, endT)
|
|
|
|
|
|
|
|
|
|
|
|
days := generateDays(startT, endT)
|
|
|
|
days := generateDays(startT, endT)
|
|
|
|
foundEvents := []*lilEvent{}
|
|
|
|
foundEvents := []*lilEvent{}
|
|
|
|
defaultLocation := *defaultLocationFlag
|
|
|
|
defaultLocation := *defaultLocationFlag
|
|
|
|
|
|
|
|
|
|
|
|
log.Printf("generated %d days between %v and %v", len(days), startT, endT)
|
|
|
|
logrus.Debugf("generated %d days between %v and %v", len(days), startT, endT)
|
|
|
|
|
|
|
|
|
|
|
|
if len(events.Items) > 0 {
|
|
|
|
if len(events.Items) > 0 {
|
|
|
|
for _, i := range events.Items {
|
|
|
|
for _, i := range events.Items {
|
|
|
@ -95,13 +115,13 @@ func main() {
|
|
|
|
if i.Start.DateTime != "" {
|
|
|
|
if i.Start.DateTime != "" {
|
|
|
|
startT, err := time.Parse(time.RFC3339, i.Start.DateTime)
|
|
|
|
startT, err := time.Parse(time.RFC3339, i.Start.DateTime)
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("error parsing start date: %v", err)
|
|
|
|
logrus.Errorf("error parsing start date: %v", err)
|
|
|
|
continue
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
endT, err := time.Parse(time.RFC3339, i.End.DateTime)
|
|
|
|
endT, err := time.Parse(time.RFC3339, i.End.DateTime)
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("error parsing end date: %v", err)
|
|
|
|
logrus.Errorf("error parsing end date: %v", err)
|
|
|
|
continue
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
}
|
|
|
|
startDate = startT.Format(dateFmt)
|
|
|
|
startDate = startT.Format(dateFmt)
|
|
|
@ -137,21 +157,23 @@ func main() {
|
|
|
|
evt.Location = loc
|
|
|
|
evt.Location = loc
|
|
|
|
|
|
|
|
|
|
|
|
if evt.Start == evt.End {
|
|
|
|
if evt.Start == evt.End {
|
|
|
|
log.Printf("skipping same-day event on %s (%s)", evt.Start, evt.Summary)
|
|
|
|
logrus.Debugf("skipping same-day event on %s (%s)", evt.Start, evt.Summary)
|
|
|
|
continue
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
}
|
|
|
|
foundEvents = append(foundEvents, evt)
|
|
|
|
foundEvents = append(foundEvents, evt)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
} else {
|
|
|
|
log.Println("No events found")
|
|
|
|
logrus.Info("No events found")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
for _, day := range days {
|
|
|
|
for _, day := range days {
|
|
|
|
log.Printf("working on date %v", day)
|
|
|
|
logrus.Debugf("working on date %v", day)
|
|
|
|
|
|
|
|
|
|
|
|
for _, evt := range foundEvents {
|
|
|
|
for _, evt := range foundEvents {
|
|
|
|
log.Printf("checking if %s is in range %s %s", day.Date, evt.Start, evt.End)
|
|
|
|
logrus.Debugf("checking if %s is in range %s %s", day.Date, evt.Start, evt.End)
|
|
|
|
|
|
|
|
|
|
|
|
if evt.HasDate(day.Date) {
|
|
|
|
if evt.HasDate(day.Date) {
|
|
|
|
log.Printf("match found for %s in range %s %s, setting location to %s", day.Date, evt.Start, evt.End, evt.Location)
|
|
|
|
logrus.Debugf("match found for %s in range %s %s, setting location to %s", day.Date, evt.Start, evt.End, evt.Location)
|
|
|
|
day.Location = evt.Location
|
|
|
|
day.Location = evt.Location
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
@ -160,9 +182,10 @@ func main() {
|
|
|
|
for _, day := range days {
|
|
|
|
for _, day := range days {
|
|
|
|
asJson, err := json.Marshal(day)
|
|
|
|
asJson, err := json.Marshal(day)
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("ERROR: %v", err)
|
|
|
|
logrus.Error(err)
|
|
|
|
continue
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
fmt.Println(string(asJson))
|
|
|
|
fmt.Println(string(asJson))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
@ -170,7 +193,7 @@ func main() {
|
|
|
|
func getClient(ctx context.Context, config *oauth2.Config) *http.Client {
|
|
|
|
func getClient(ctx context.Context, config *oauth2.Config) *http.Client {
|
|
|
|
cacheFile, err := tokenCacheFile()
|
|
|
|
cacheFile, err := tokenCacheFile()
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalf("Unable to get path to cached credential file. %v", err)
|
|
|
|
logrus.Fatalf("Unable to get path to cached credential file. %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
tok, err := tokenFromFile(cacheFile)
|
|
|
|
tok, err := tokenFromFile(cacheFile)
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
@ -187,12 +210,12 @@ func getTokenFromWeb(config *oauth2.Config) *oauth2.Token {
|
|
|
|
|
|
|
|
|
|
|
|
var code string
|
|
|
|
var code string
|
|
|
|
if _, err := fmt.Scan(&code); err != nil {
|
|
|
|
if _, err := fmt.Scan(&code); err != nil {
|
|
|
|
log.Fatalf("Unable to read authorization code %v", err)
|
|
|
|
logrus.Fatalf("Unable to read authorization code %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
tok, err := config.Exchange(oauth2.NoContext, code)
|
|
|
|
tok, err := config.Exchange(oauth2.NoContext, code)
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalf("Unable to retrieve token from web %v", err)
|
|
|
|
logrus.Fatalf("Unable to retrieve token from web %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return tok
|
|
|
|
return tok
|
|
|
|
}
|
|
|
|
}
|
|
|
@ -223,7 +246,7 @@ func saveToken(file string, token *oauth2.Token) {
|
|
|
|
fmt.Printf("Saving credential file to: %s\n", file)
|
|
|
|
fmt.Printf("Saving credential file to: %s\n", file)
|
|
|
|
f, err := os.OpenFile(file, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0600)
|
|
|
|
f, err := os.OpenFile(file, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0600)
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
log.Fatalf("Unable to cache oauth token: %v", err)
|
|
|
|
logrus.Fatalf("Unable to cache oauth token: %v", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
defer f.Close()
|
|
|
|
defer f.Close()
|
|
|
|
json.NewEncoder(f).Encode(token)
|
|
|
|
json.NewEncoder(f).Encode(token)
|
|
|
@ -238,13 +261,13 @@ type lilEvent struct {
|
|
|
|
|
|
|
|
|
|
|
|
func (e *lilEvent) HasDate(d string) bool {
|
|
|
|
func (e *lilEvent) HasDate(d string) bool {
|
|
|
|
if e.Start == d {
|
|
|
|
if e.Start == d {
|
|
|
|
log.Printf("match for %s at %s", d, e.Start)
|
|
|
|
logrus.Debugf("match for %s at %s", d, e.Start)
|
|
|
|
return true
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
dT, err := time.Parse(dateFmt, d)
|
|
|
|
dT, err := time.Parse(dateFmt, d)
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("ERROR: %v", err)
|
|
|
|
logrus.Error(err)
|
|
|
|
return false
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
@ -257,7 +280,7 @@ func (e *lilEvent) dateRange() []time.Time {
|
|
|
|
for _, t := range []string{e.Start, e.End} {
|
|
|
|
for _, t := range []string{e.Start, e.End} {
|
|
|
|
parsedDate, err := time.Parse(dateFmt, t)
|
|
|
|
parsedDate, err := time.Parse(dateFmt, t)
|
|
|
|
if err != nil {
|
|
|
|
if err != nil {
|
|
|
|
log.Printf("ERROR: %v", err)
|
|
|
|
logrus.Error(err)
|
|
|
|
ret = append(ret, time.Time{})
|
|
|
|
ret = append(ret, time.Time{})
|
|
|
|
continue
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
}
|