/
main.go
121 lines (99 loc) · 3.95 KB
/
main.go
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
112
113
114
115
116
117
118
119
120
121
package main
import (
"log"
"net/http"
"github.com/gorilla/mux"
"github.com/gorilla/sessions"
"github.com/mrjones/oauth"
"database/sql"
_ "github.com/mattn/go-sqlite3"
"github.com/hoisie/mustache"
)
func indexHandler(s Services, w http.ResponseWriter, r *http.Request) {
args := map[string]string{
"homepage": "yes",
}
user,_ := getUser(s, w, r)
if(user != nil) {
args["username"] = user.screenName
}
data := mustache.RenderFileInLayout(
"/usr/share/tweetautofeeder/templates/main_page.must",
"/usr/share/tweetautofeeder/templates/layout.must",
args)
w.Write([]byte(data))
return
}
func renderError(w http.ResponseWriter, errorMessage string) {
data := mustache.RenderFileInLayout(
"/usr/share/tweetautofeeder/templates/error_page.must",
"/usr/share/tweetautofeeder/templates/layout.must",
map[string]string{
"errorMessage": errorMessage,
})
w.Write([]byte(data))
w.WriteHeader(http.StatusBadRequest)
return
}
type Services struct {
customer *oauth.Consumer
sessions *sessions.CookieStore
storage *sql.DB
router *mux.Router
}
type ServicesHandler func(s Services, w http.ResponseWriter, r *http.Request)
func makeServicesHandler( s Services, sh ServicesHandler ) http.HandlerFunc {
return func (w http.ResponseWriter, r *http.Request) {
sh(s, w, r)
}
}
type User struct {
sessionCookie string
userId string
screenName string
atoken oauth.AccessToken
}
func main() {
var err error
err = nil
services := Services{}
services.customer = oauth.NewConsumer(
API_CONSUMER_KEY,
API_CONSUMER_SECRET,
oauth.ServiceProvider{
RequestTokenUrl: "https://api.twitter.com/oauth/request_token",
AuthorizeTokenUrl: "https://api.twitter.com/oauth/authorize",
AccessTokenUrl: "https://api.twitter.com/oauth/access_token",
})
services.customer.Debug(true)
// TODO parameterize
services.sessions = sessions.NewCookieStore([]byte(COOKIE_KEY))
// TODO paramterize
services.storage, err = sql.Open("sqlite3", "/home/vagrant/storage.sqlite")
if(err != nil) {
log.Fatal(err)
}
defer services.storage.Close()
r := mux.NewRouter()
services.router = r
r.HandleFunc("/", makeServicesHandler(services, indexHandler) ).Name("root")
r.HandleFunc("/login", makeServicesHandler(services, loginHandler) ).Name("login")
r.HandleFunc("/login/begin", makeServicesHandler(services, beginLoginHandler) ).Name("login_begin")
r.HandleFunc("/login/oauth_callback", makeServicesHandler(services, oauthCallbackHandler)).Name("oauth_callback")
r.HandleFunc("/pending", makeServicesHandler(services, pendingHandler) ).Name("pending").Methods("GET")
r.HandleFunc("/pending", makeServicesHandler(services, postNewHandler) ).Name("pending_post").Methods("POST")
r.HandleFunc("/pending/delete/{tweetid}",makeServicesHandler(services, deleteHandler) ).Name("delete")
r.HandleFunc("/pending/post/{tweetid}",makeServicesHandler(services, postNowHandler) ).Name("post")
r.HandleFunc("/posted", makeServicesHandler(services, postedHandler) ).Name("posted")
// TODO:
//r.HandleFunc("/schedule", makeServicesHandler(services, scheduleHandler) ).Name("schedule")
//r.HandleFunc("/logout", makeServicesHandler(services, logoutHandler) ).Name("logout")
// TODO: robots.txt, humans.txt
r.PathPrefix("/static/").Handler(http.StripPrefix("/static/", http.FileServer(http.Dir("/usr/share/tweetautofeeder/www"))))
r.PathPrefix("/css" ).Handler(http.StripPrefix("/css/", http.FileServer(http.Dir("/usr/share/tweetautofeeder/css"))))
r.PathPrefix("/js" ).Handler(http.StripPrefix("/js/", http.FileServer(http.Dir("/usr/share/tweetautofeeder/js"))))
err = http.ListenAndServe(":8080", r)
if err != nil {
log.Fatal(err)
}
}