faceprox/faceprox.go at main · geeksforsocialchange/faceprox · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
package main

import (
	"alicekaerast/faceprox/lib"
	_ "embed"
	"encoding/json"
	"fmt"
	ics "github.com/arran4/golang-ical"
	faceloader "github.com/geeksforsocialchange/faceloader/parser"
	"github.com/gorilla/mux"
	"github.com/patrickmn/go-cache"
	"log"
	"net/http"
	"time"
)

//go:embed static/event.html
var eventPage string

//go:embed static/index.html
var indexPage string

//go:embed static/events.html
var eventsPage string

var c *cache.Cache

var m *faceloader.MBasic

func main() {
	c = cache.New(5*time.Minute, 10*time.Minute)

	m = faceloader.NewMBasicClient()

	r := mux.NewRouter()
	r.HandleFunc("/", HomeHandler)
	r.HandleFunc("/events/{key}.json", EventDataHandler)
	r.HandleFunc("/events/{key}.ics", EventIcalHandler)
	r.HandleFunc("/events/{key}", EventHandler)
	r.HandleFunc("/page/{key}.json", PageDataHandler)
	r.HandleFunc("/page/{key}.ics", PageIcalHandler)
	r.HandleFunc("/page/{key}", PageHandler)

	srv := &http.Server{
		Handler: r,
		Addr:    "0.0.0.0:8000",
		// Good practice: enforce timeouts for servers you create!
		WriteTimeout: 15 * time.Second,
		ReadTimeout:  15 * time.Second,
	}

	log.Println("Listening on http://0.0.0.0:8000")
	log.Fatal(srv.ListenAndServe())
}

func HomeHandler(w http.ResponseWriter, r *http.Request) {
	fmt.Fprint(w, indexPage)
}

func EventHandler(w http.ResponseWriter, r *http.Request) {
	fmt.Fprint(w, eventPage)
}

func PageHandler(w http.ResponseWriter, r *http.Request) {
	fmt.Fprint(w, eventsPage)
}

func EventDataHandler(w http.ResponseWriter, r *http.Request) {
	vars := mux.Vars(r)
	url := fmt.Sprintf("https://mbasic.facebook.com/events/%v", vars["key"])
	result := lib.GetEvent(url, c, m)
	w.Header().Set("Content-Type", "application/json")
	json.NewEncoder(w).Encode(result)
}

func EventIcalHandler(w http.ResponseWriter, r *http.Request) {
	vars := mux.Vars(r)
	url := fmt.Sprintf("https://mbasic.facebook.com/events/%v", vars["key"])
	result := lib.GetEvent(url, c, m)
	ics, err := faceloader.InterfaceToIcal(result)
	if err != nil {
		log.Println(err)
	}
	w.Header().Set("Content-Type", "calendar/text")
	fmt.Fprint(w, ics.Serialize())
}

func PageIcalHandler(w http.ResponseWriter, r *http.Request) {
	vars := mux.Vars(r)

	events := lib.GetEvents(vars["key"], c, m)
	cal := ics.NewCalendar()

	for i := range events {
		event, err := faceloader.InterfaceToIcal(events[i])
		if err != nil {
			log.Println(err)
		}
		cal.Components = append(cal.Components, &event)
	}
	w.Header().Set("Content-Type", "calendar/text")
	fmt.Fprint(w, cal.Serialize())
}

func PageDataHandler(w http.ResponseWriter, r *http.Request) {
	vars := mux.Vars(r)

	events := lib.GetEvents(vars["key"], c, m)
	w.Header().Set("Content-Type", "application/json")
	json.NewEncoder(w).Encode(events)
}