Merge branch 'master' of http://khuhub.khu.ac.kr/2020-2-capstone-design1/JJS_project1
Showing
11 changed files
with
463 additions
and
288 deletions
1 | - | 1 | + |
2 | -# Created by https://www.toptal.com/developers/gitignore/api/go,vscode | 2 | +# Created by https://www.toptal.com/developers/gitignore/api/go,vscode |
3 | -# Edit at https://www.toptal.com/developers/gitignore?templates=go,vscode | 3 | +# Edit at https://www.toptal.com/developers/gitignore?templates=go,vscode |
4 | - | 4 | + |
5 | -### Go ### | 5 | +### Go ### |
6 | -# Binaries for programs and plugins | 6 | +# Binaries for programs and plugins |
7 | -*.exe | 7 | +*.exe |
8 | -*.exe~ | 8 | +*.exe~ |
9 | -*.dll | 9 | +*.dll |
10 | -*.so | 10 | +*.so |
11 | -*.dylib | 11 | +*.dylib |
12 | - | 12 | + |
13 | -# Test binary, built with `go test -c` | 13 | +# Test binary, built with `go test -c` |
14 | -*.test | 14 | +*.test |
15 | - | 15 | + |
16 | -# Output of the go coverage tool, specifically when used with LiteIDE | 16 | +# Output of the go coverage tool, specifically when used with LiteIDE |
17 | -*.out | 17 | +*.out |
18 | - | 18 | + |
19 | -# Dependency directories (remove the comment below to include it) | 19 | +# Dependency directories (remove the comment below to include it) |
20 | -# vendor/ | 20 | +# vendor/ |
21 | - | 21 | + |
22 | -### Go Patch ### | 22 | +### Go Patch ### |
23 | -/vendor/ | 23 | +/vendor/ |
24 | -/Godeps/ | 24 | +/Godeps/ |
25 | - | 25 | + |
26 | -### vscode ### | 26 | +### vscode ### |
27 | -.vscode/* | 27 | +.vscode/* |
28 | -!.vscode/settings.json | 28 | +!.vscode/settings.json |
29 | -!.vscode/tasks.json | 29 | +!.vscode/tasks.json |
30 | -!.vscode/launch.json | 30 | +!.vscode/launch.json |
31 | -!.vscode/extensions.json | 31 | +!.vscode/extensions.json |
32 | -*.code-workspace | 32 | +*.code-workspace |
33 | - | 33 | + |
34 | -# End of https://www.toptal.com/developers/gitignore/api/go,vscode | 34 | +# End of https://www.toptal.com/developers/gitignore/api/go,vscode |
35 | - | 35 | + |
36 | -__debug_bin | 36 | +__debug_bin |
37 | -config.json | 37 | +config.json |
38 | data | 38 | data |
... | \ No newline at end of file | ... | \ No newline at end of file | ... | ... |
1 | -package main | 1 | +package main |
2 | - | 2 | + |
3 | -import ( | 3 | +import ( |
4 | - "fmt" | 4 | + "fmt" |
5 | - "net/http" | 5 | + "net/http" |
6 | - | 6 | + |
7 | - "github.com/gorilla/mux" | 7 | + "github.com/gorilla/mux" |
8 | - | 8 | + |
9 | - _ "github.com/go-sql-driver/mysql" | 9 | + _ "github.com/go-sql-driver/mysql" |
10 | - "github.com/jmoiron/sqlx" | 10 | + "github.com/jmoiron/sqlx" |
11 | -) | 11 | +) |
12 | - | 12 | + |
13 | -type Prop int | 13 | +type Prop int |
14 | - | 14 | + |
15 | -const ( | 15 | +const ( |
16 | - PropUserNo Prop = iota | 16 | + PropUserNo Prop = iota |
17 | -) | 17 | +) |
18 | - | 18 | + |
19 | -type App struct { | 19 | +type App struct { |
20 | - Config Config | 20 | + Config Config |
21 | - db *sqlx.DB | 21 | + db *sqlx.DB |
22 | - router *mux.Router | 22 | + router *mux.Router |
23 | -} | 23 | +} |
24 | - | 24 | + |
25 | -func NewApp(config Config) *App { | 25 | +func NewApp(config Config) *App { |
26 | - app := new(App) | 26 | + app := new(App) |
27 | - app.Config = config | 27 | + app.Config = config |
28 | - | 28 | + |
29 | - dsn := fmt.Sprintf("%s:%s@tcp(%s)/%s", config.Database.User, config.Database.Password, config.Database.Host, config.Database.Name) | 29 | + dsn := fmt.Sprintf("%s:%s@tcp(%s)/%s", config.Database.User, config.Database.Password, config.Database.Host, config.Database.Name) |
30 | - app.db = sqlx.MustOpen("mysql", dsn) | 30 | + app.db = sqlx.MustOpen("mysql", dsn) |
31 | - | 31 | + |
32 | - app.router = mux.NewRouter() | 32 | + app.router = mux.NewRouter() |
33 | - app.router.HandleFunc("/users", app.PostUsers).Methods("POST") | 33 | + app.router.HandleFunc("/users", app.PostUsers).Methods("POST") |
34 | - app.router.HandleFunc("/users/tokens", app.PostTokens).Methods("POST") | 34 | + app.router.HandleFunc("/users/tokens", app.PostTokens).Methods("POST") |
35 | - app.router.Handle("/extractions", app.WithAuth(app.PostExtractions)).Methods("Post") | 35 | + app.router.Handle("/extractions", app.WithAuth(app.PostExtractions)).Methods("Post") |
36 | - | 36 | + app.router.HandleFunc("/extractions/{file}/calls", app.GetCalls).Methods("GET") |
37 | - return app | 37 | + app.router.HandleFunc("/extractions/{file}/messages", app.GetMessages).Methods("GET") |
38 | -} | 38 | + app.router.HandleFunc("/extractions/{file}/calls/analyses", app.GetCallsAnalyses).Methods("GET") |
39 | - | 39 | + app.router.HandleFunc("/extractions/{file}/apps/analyses", app.GetAppsAnalyses).Methods("GET") |
40 | -func (app *App) Serve() { | 40 | + app.router.HandleFunc("/extractions/{file}/messages/analyses", app.GetMessagesAnalyses).Methods("GET") |
41 | - http.ListenAndServe(fmt.Sprintf(":%d", app.Config.Port), app.router) | 41 | + |
42 | -} | 42 | + return app |
43 | +} | ||
44 | + | ||
45 | +func (app *App) Serve() { | ||
46 | + http.ListenAndServe(fmt.Sprintf(":%d", app.Config.Port), app.router) | ||
47 | +} | ... | ... |
1 | -package main | 1 | +package main |
2 | - | 2 | + |
3 | -import ( | 3 | +import ( |
4 | - "encoding/json" | 4 | + "encoding/json" |
5 | - "io/ioutil" | 5 | + "io/ioutil" |
6 | -) | 6 | +) |
7 | - | 7 | + |
8 | -type Config struct { | 8 | +type Config struct { |
9 | - Port int `json:"port"` | 9 | + Port int `json:"port"` |
10 | - Database struct { | 10 | + Database struct { |
11 | - Host string `json:"host"` | 11 | + Host string `json:"host"` |
12 | - Name string `json:"name"` | 12 | + Name string `json:"name"` |
13 | - User string `json:"user"` | 13 | + User string `json:"user"` |
14 | - Password string `json:"password"` | 14 | + Password string `json:"password"` |
15 | - } `json:"database"` | 15 | + } `json:"database"` |
16 | - TokenSecret string `json:"token_secret"` | 16 | + TokenSecret string `json:"token_secret"` |
17 | -} | 17 | +} |
18 | - | 18 | + |
19 | -func LoadConfig(path string) (Config, error) { | 19 | +func LoadConfig(path string) (Config, error) { |
20 | - config := Config{} | 20 | + config := Config{} |
21 | - | 21 | + |
22 | - data, err := ioutil.ReadFile(path) | 22 | + data, err := ioutil.ReadFile(path) |
23 | - if err == nil { | 23 | + if err == nil { |
24 | - err = json.Unmarshal(data, &config) | 24 | + err = json.Unmarshal(data, &config) |
25 | - } | 25 | + } |
26 | - | 26 | + |
27 | - return config, err | 27 | + return config, err |
28 | -} | 28 | +} | ... | ... |
source/server/data.go
0 → 100644
1 | +package main | ||
2 | + | ||
3 | +import ( | ||
4 | + "fmt" | ||
5 | + "net/http" | ||
6 | + "time" | ||
7 | + | ||
8 | + "github.com/gorilla/mux" | ||
9 | + "github.com/jmoiron/sqlx" | ||
10 | + | ||
11 | + _ "github.com/mattn/go-sqlite3" | ||
12 | +) | ||
13 | + | ||
14 | +type Call struct { | ||
15 | + ID int `json:"id" db:"id"` | ||
16 | + Type int `json:"type" db:"type"` | ||
17 | + Name *string `json:"name" db:"name"` | ||
18 | + Number int `json:"number" db:"number"` | ||
19 | + Duration int `json:"duration" db:"duration"` | ||
20 | + Date Time `json:"date" db:"date"` | ||
21 | +} | ||
22 | + | ||
23 | +func (app *App) GetCalls(w http.ResponseWriter, r *http.Request) { | ||
24 | + vars := mux.Vars(r) | ||
25 | + | ||
26 | + calls := []Call{} | ||
27 | + | ||
28 | + db, err := sqlx.Connect("sqlite3", fmt.Sprintf("data/1/%s", vars["file"])) | ||
29 | + if err != nil { | ||
30 | + WriteError(w, http.StatusInternalServerError, "Could not open db file") | ||
31 | + return | ||
32 | + } | ||
33 | + defer db.Close() | ||
34 | + | ||
35 | + query := `SELECT * FROM calllog` | ||
36 | + fmt.Println(db.Select(&calls, query)) | ||
37 | + | ||
38 | + WriteJson(w, calls) | ||
39 | +} | ||
40 | + | ||
41 | +type CallStats struct { | ||
42 | + Number string `json:"number" db:"number"` | ||
43 | + Name *string `json:"name" db:"name"` | ||
44 | + Incoming int `json:"incoming" db:"incoming"` | ||
45 | + Outgoing int `json:"outgoing" db:"outgoing"` | ||
46 | + Duration int `json:"duration" db:"duration"` | ||
47 | +} | ||
48 | + | ||
49 | +func (app *App) GetCallsAnalyses(w http.ResponseWriter, r *http.Request) { | ||
50 | + vars := mux.Vars(r) | ||
51 | + | ||
52 | + calls := []CallStats{} | ||
53 | + | ||
54 | + db, err := sqlx.Connect("sqlite3", fmt.Sprintf("data/1/%s", vars["file"])) | ||
55 | + if err != nil { | ||
56 | + WriteError(w, http.StatusInternalServerError, "Could not open db file") | ||
57 | + return | ||
58 | + } | ||
59 | + defer db.Close() | ||
60 | + | ||
61 | + query := `SELECT number, name, | ||
62 | + (SELECT COUNT(1) FROM calllog s WHERE s.number=c.number AND s.type=1) incoming, | ||
63 | + (SELECT COUNT(1) FROM calllog s WHERE s.number=c.number AND s.type=2) outgoing, | ||
64 | + SUM(duration) duration | ||
65 | + FROM calllog c GROUP BY number ORDER BY duration DESC` | ||
66 | + db.Select(&calls, query) | ||
67 | + | ||
68 | + WriteJson(w, calls) | ||
69 | +} | ||
70 | + | ||
71 | +type AppInfo struct { | ||
72 | + PackageName string `json:"package_name" db:"packagename"` | ||
73 | + Name string `json:"name" db:"name"` | ||
74 | + Version string `json:"version" db:"version"` | ||
75 | + WifiUsage int `json:"wifi_usage" db:"wifiusage"` | ||
76 | + CellularUsage int `json:"cellular_usage" db:"cellularusage"` | ||
77 | + LastUsed time.Time `json:"last_used" db:"lasttimeused"` | ||
78 | + ForegroundTime int `json:"foreground_time" db:"totaltimeforeground"` | ||
79 | +} | ||
80 | + | ||
81 | +func (app *App) GetAppsAnalyses(w http.ResponseWriter, r *http.Request) { | ||
82 | + vars := mux.Vars(r) | ||
83 | + | ||
84 | + apps := []AppInfo{} | ||
85 | + db, err := sqlx.Connect("sqlite3", fmt.Sprintf("data/1/%s", vars["file"])) | ||
86 | + if err != nil { | ||
87 | + WriteError(w, http.StatusInternalServerError, "Could not open db file") | ||
88 | + return | ||
89 | + } | ||
90 | + defer db.Close() | ||
91 | + | ||
92 | + query := `SELECT | ||
93 | + a.packagename, a.name, a.version, a.wifiusage, a.cellularusage, | ||
94 | + u.lasttimeused, u.totaltimeforeground | ||
95 | + FROM AppInfo a JOIN AppUsageYear u | ||
96 | + ORDER BY totaltimeforeground DESC` | ||
97 | + db.Select(&apps, query) | ||
98 | + | ||
99 | + WriteJson(w, apps) | ||
100 | +} | ||
101 | + | ||
102 | +type Message struct { | ||
103 | + ID int `json:"id" db:"mid"` | ||
104 | + Type int `json:"type" db:"type"` | ||
105 | + Address string `json:"address"` | ||
106 | + Body string `json:"body"` | ||
107 | + Date Time `json:"date" db:"date"` | ||
108 | +} | ||
109 | + | ||
110 | +func (app *App) GetMessages(w http.ResponseWriter, r *http.Request) { | ||
111 | + vars := mux.Vars(r) | ||
112 | + | ||
113 | + messages := []Message{} | ||
114 | + db, err := sqlx.Connect("sqlite3", fmt.Sprintf("data/1/%s", vars["file"])) | ||
115 | + if err != nil { | ||
116 | + WriteError(w, http.StatusInternalServerError, "Could not open db file") | ||
117 | + return | ||
118 | + } | ||
119 | + defer db.Close() | ||
120 | + | ||
121 | + query := `SELECT mid, type, address, body, date FROM sms` | ||
122 | + db.Select(&messages, query) | ||
123 | + | ||
124 | + WriteJson(w, messages) | ||
125 | +} | ||
126 | + | ||
127 | +type MessageStats struct { | ||
128 | + Address string `json:"number" db:"number"` | ||
129 | + Receive int `json:"incoming" db:"incoming"` | ||
130 | + Send int `json:"outgoing" db:"outgoing"` | ||
131 | +} | ||
132 | + | ||
133 | +func (app *App) GetMessagesAnalyses(w http.ResponseWriter, r *http.Request) { | ||
134 | + vars := mux.Vars(r) | ||
135 | + | ||
136 | + messages := []MessageStats{} | ||
137 | + db, err := sqlx.Connect("sqlite3", fmt.Sprintf("data/1/%s", vars["file"])) | ||
138 | + if err != nil { | ||
139 | + WriteError(w, http.StatusInternalServerError, "Could not open db file") | ||
140 | + return | ||
141 | + } | ||
142 | + defer db.Close() | ||
143 | + | ||
144 | + query := `SELECT address, | ||
145 | + (SELECT COUNT(1) FROM sms m WHERE m.address=s.address AND m.type=1) receive, | ||
146 | + (SELECT COUNT(1) FROM sms m WHERE m.address=s.address AND m.type=2) send | ||
147 | + FROM sms s GROUP BY address ORDER BY receive + send DESC` | ||
148 | + db.Select(&messages, query) | ||
149 | + | ||
150 | + WriteJson(w, messages) | ||
151 | +} |
1 | -package main | 1 | +package main |
2 | - | 2 | + |
3 | -import ( | 3 | +import ( |
4 | - "fmt" | 4 | + "fmt" |
5 | - "io" | 5 | + "io" |
6 | - "net/http" | 6 | + "net/http" |
7 | - "os" | 7 | + "os" |
8 | - "strings" | 8 | + "strings" |
9 | - | 9 | + |
10 | - "github.com/google/uuid" | 10 | + "github.com/google/uuid" |
11 | -) | 11 | +) |
12 | - | 12 | + |
13 | -func (app *App) PostExtractions(w http.ResponseWriter, r *http.Request) { | 13 | +func (app *App) PostExtractions(w http.ResponseWriter, r *http.Request) { |
14 | - userNo := r.Context().Value(PropUserNo).(uint64) | 14 | + userNo := r.Context().Value(PropUserNo).(uint64) |
15 | - r.ParseMultipartForm(32 << 20) | 15 | + r.ParseMultipartForm(32 << 20) |
16 | - | 16 | + |
17 | - form, _, err := r.FormFile("file") | 17 | + form, _, err := r.FormFile("file") |
18 | - if err != nil { | 18 | + if err != nil { |
19 | - WriteError(w, http.StatusInternalServerError, "Unknown error") | 19 | + WriteError(w, http.StatusInternalServerError, "Unknown error") |
20 | - return | 20 | + return |
21 | - } | 21 | + } |
22 | - | 22 | + |
23 | - defer form.Close() | 23 | + defer form.Close() |
24 | - | 24 | + |
25 | - dir := fmt.Sprintf("data/%d", userNo) | 25 | + dir := fmt.Sprintf("data/%d", userNo) |
26 | - os.MkdirAll(dir, 0644) | 26 | + os.MkdirAll(dir, 0644) |
27 | - | 27 | + |
28 | - name := strings.Replace(uuid.New().String(), "-", "", -1) | 28 | + name := strings.Replace(uuid.New().String(), "-", "", -1) |
29 | - file, err := os.Create(fmt.Sprintf("%s/%s", dir, name)) | 29 | + file, err := os.Create(fmt.Sprintf("%s/%s", dir, name)) |
30 | - if err != nil { | 30 | + if err != nil { |
31 | - WriteError(w, http.StatusInternalServerError, "Unknown error") | 31 | + WriteError(w, http.StatusInternalServerError, "Unknown error") |
32 | - return | 32 | + return |
33 | - } | 33 | + } |
34 | - defer file.Close() | 34 | + defer file.Close() |
35 | - | 35 | + |
36 | - _, err = io.Copy(file, form) | 36 | + _, err = io.Copy(file, form) |
37 | - if err != nil { | 37 | + if err != nil { |
38 | - WriteError(w, http.StatusInternalServerError, "Unknown error") | 38 | + WriteError(w, http.StatusInternalServerError, "Unknown error") |
39 | - return | 39 | + return |
40 | - } | 40 | + } |
41 | - | 41 | + |
42 | - w.Write([]byte("success")) | 42 | + w.Write([]byte("success")) |
43 | -} | 43 | +} | ... | ... |
... | @@ -8,5 +8,6 @@ require ( | ... | @@ -8,5 +8,6 @@ require ( |
8 | github.com/google/uuid v1.1.2 | 8 | github.com/google/uuid v1.1.2 |
9 | github.com/gorilla/mux v1.8.0 | 9 | github.com/gorilla/mux v1.8.0 |
10 | github.com/jmoiron/sqlx v1.2.0 | 10 | github.com/jmoiron/sqlx v1.2.0 |
11 | + github.com/mattn/go-sqlite3 v1.9.0 | ||
11 | golang.org/x/crypto v0.0.0-20201016220609-9e8e0b390897 | 12 | golang.org/x/crypto v0.0.0-20201016220609-9e8e0b390897 |
12 | ) | 13 | ) | ... | ... |
... | @@ -11,6 +11,7 @@ github.com/gorilla/mux v1.8.0/go.mod h1:DVbg23sWSpFRCP0SfiEN6jmj59UnW/n46BH5rLB7 | ... | @@ -11,6 +11,7 @@ github.com/gorilla/mux v1.8.0/go.mod h1:DVbg23sWSpFRCP0SfiEN6jmj59UnW/n46BH5rLB7 |
11 | github.com/jmoiron/sqlx v1.2.0 h1:41Ip0zITnmWNR/vHV+S4m+VoUivnWY5E4OJfLZjCJMA= | 11 | github.com/jmoiron/sqlx v1.2.0 h1:41Ip0zITnmWNR/vHV+S4m+VoUivnWY5E4OJfLZjCJMA= |
12 | github.com/jmoiron/sqlx v1.2.0/go.mod h1:1FEQNm3xlJgrMD+FBdI9+xvCksHtbpVBBw5dYhBSsks= | 12 | github.com/jmoiron/sqlx v1.2.0/go.mod h1:1FEQNm3xlJgrMD+FBdI9+xvCksHtbpVBBw5dYhBSsks= |
13 | github.com/lib/pq v1.0.0/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo= | 13 | github.com/lib/pq v1.0.0/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo= |
14 | +github.com/mattn/go-sqlite3 v1.9.0 h1:pDRiWfl+++eC2FEFRy6jXmQlvp4Yh3z1MJKg4UeYM/4= | ||
14 | github.com/mattn/go-sqlite3 v1.9.0/go.mod h1:FPy6KqzDD04eiIsT53CuJW3U88zkxoIYsOqkbpncsNc= | 15 | github.com/mattn/go-sqlite3 v1.9.0/go.mod h1:FPy6KqzDD04eiIsT53CuJW3U88zkxoIYsOqkbpncsNc= |
15 | golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= | 16 | golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= |
16 | golang.org/x/crypto v0.0.0-20201016220609-9e8e0b390897 h1:pLI5jrR7OSLijeIDcmRxNmw2api+jEfxLoykJVice/E= | 17 | golang.org/x/crypto v0.0.0-20201016220609-9e8e0b390897 h1:pLI5jrR7OSLijeIDcmRxNmw2api+jEfxLoykJVice/E= | ... | ... |
1 | -package main | 1 | +package main |
2 | - | 2 | + |
3 | -import ( | 3 | +import ( |
4 | - "log" | 4 | + "log" |
5 | -) | 5 | +) |
6 | - | 6 | + |
7 | -func main() { | 7 | +func main() { |
8 | - config, err := LoadConfig("config.json") | 8 | + config, err := LoadConfig("config.json") |
9 | - if err != nil { | 9 | + if err != nil { |
10 | - log.Fatal(err) | 10 | + log.Fatal(err) |
11 | - } | 11 | + } |
12 | - | 12 | + |
13 | - app := NewApp(config) | 13 | + app := NewApp(config) |
14 | - app.Serve() | 14 | + app.Serve() |
15 | -} | 15 | +} | ... | ... |
source/server/types.go
0 → 100644
1 | +package main | ||
2 | + | ||
3 | +import ( | ||
4 | + "time" | ||
5 | +) | ||
6 | + | ||
7 | +type Time time.Time | ||
8 | + | ||
9 | +func (t *Time) Scan(v interface{}) error { | ||
10 | + p, err := time.Parse("2006-01-02 15:04:05", string(v.([]byte))) | ||
11 | + *t = Time(p) | ||
12 | + return err | ||
13 | +} | ||
14 | + | ||
15 | +func (t *Time) MarshalJSON() ([]byte, error) { | ||
16 | + return time.Time(*t).MarshalJSON() | ||
17 | +} |
1 | -package main | 1 | +package main |
2 | - | 2 | + |
3 | -import ( | 3 | +import ( |
4 | - "context" | 4 | + "context" |
5 | - "encoding/json" | 5 | + "encoding/json" |
6 | - "net/http" | 6 | + "net/http" |
7 | - "strings" | 7 | + "strings" |
8 | - "time" | 8 | + "time" |
9 | - | 9 | + |
10 | - "github.com/dgrijalva/jwt-go" | 10 | + "github.com/dgrijalva/jwt-go" |
11 | - "github.com/go-sql-driver/mysql" | 11 | + "github.com/go-sql-driver/mysql" |
12 | - "golang.org/x/crypto/sha3" | 12 | + "golang.org/x/crypto/sha3" |
13 | -) | 13 | +) |
14 | - | 14 | + |
15 | -type User struct { | 15 | +type User struct { |
16 | - No uint64 `json:"no"` | 16 | + No uint64 `json:"no"` |
17 | - ID string `json:"id"` | 17 | + ID string `json:"id"` |
18 | - Name string `json:"name"` | 18 | + Name string `json:"name"` |
19 | - CreatedAt time.Time `json:"created_at"` | 19 | + CreatedAt time.Time `json:"created_at"` |
20 | - ExpiredAt time.Time `json:"expired_at"` | 20 | + ExpiredAt time.Time `json:"expired_at"` |
21 | -} | 21 | +} |
22 | - | 22 | + |
23 | -func (app *App) PostUsers(w http.ResponseWriter, r *http.Request) { | 23 | +func (app *App) PostUsers(w http.ResponseWriter, r *http.Request) { |
24 | - body := make(map[string]interface{}) | 24 | + body := make(map[string]interface{}) |
25 | - err := json.NewDecoder(r.Body).Decode(&body) | 25 | + err := json.NewDecoder(r.Body).Decode(&body) |
26 | - if err != nil { | 26 | + if err != nil { |
27 | - WriteError(w, http.StatusBadRequest, "Failed to parse request json") | 27 | + WriteError(w, http.StatusBadRequest, "Failed to parse request json") |
28 | - return | 28 | + return |
29 | - } | 29 | + } |
30 | - | 30 | + |
31 | - hash := sha3.Sum256([]byte(body["password"].(string))) | 31 | + hash := sha3.Sum256([]byte(body["password"].(string))) |
32 | - | 32 | + |
33 | - res, err := app.db.Exec("INSERT INTO users (`id`, `password`, `name`) VALUES (?, ?, ?)", body["id"], hash[:], body["name"]) | 33 | + res, err := app.db.Exec("INSERT INTO users (`id`, `password`, `name`) VALUES (?, ?, ?)", body["id"], hash[:], body["name"]) |
34 | - if err != nil { | 34 | + if err != nil { |
35 | - if merr, ok := err.(*mysql.MySQLError); ok { | 35 | + if merr, ok := err.(*mysql.MySQLError); ok { |
36 | - if merr.Number == 1062 { | 36 | + if merr.Number == 1062 { |
37 | - WriteError(w, http.StatusConflict, "Already registered") | 37 | + WriteError(w, http.StatusConflict, "Already registered") |
38 | - return | 38 | + return |
39 | - } | 39 | + } |
40 | - } | 40 | + } |
41 | - | 41 | + |
42 | - WriteError(w, http.StatusInternalServerError, "Failed to register") | 42 | + WriteError(w, http.StatusInternalServerError, "Failed to register") |
43 | - return | 43 | + return |
44 | - } | 44 | + } |
45 | - | 45 | + |
46 | - no, _ := res.LastInsertId() | 46 | + no, _ := res.LastInsertId() |
47 | - WriteJson(w, map[string]interface{}{"user_no": no}) | 47 | + WriteJson(w, map[string]interface{}{"user_no": no}) |
48 | -} | 48 | +} |
49 | - | 49 | + |
50 | -type AuthClaims struct { | 50 | +type AuthClaims struct { |
51 | - UserNo uint64 `json:"user_no"` | 51 | + UserNo uint64 `json:"user_no"` |
52 | - jwt.StandardClaims | 52 | + jwt.StandardClaims |
53 | -} | 53 | +} |
54 | - | 54 | + |
55 | -func (app *App) PostTokens(w http.ResponseWriter, r *http.Request) { | 55 | +func (app *App) PostTokens(w http.ResponseWriter, r *http.Request) { |
56 | - body := make(map[string]interface{}) | 56 | + body := make(map[string]interface{}) |
57 | - err := json.NewDecoder(r.Body).Decode(&body) | 57 | + err := json.NewDecoder(r.Body).Decode(&body) |
58 | - if err != nil { | 58 | + if err != nil { |
59 | - WriteError(w, http.StatusBadRequest, "Failed to parse request json") | 59 | + WriteError(w, http.StatusBadRequest, "Failed to parse request json") |
60 | - return | 60 | + return |
61 | - } | 61 | + } |
62 | - | 62 | + |
63 | - hash := sha3.Sum256([]byte(body["password"].(string))) | 63 | + hash := sha3.Sum256([]byte(body["password"].(string))) |
64 | - rows, err := app.db.Query("SELECT `no` FROM users WHERE `id`=? AND `password`=?", body["id"], hash[:]) | 64 | + rows, err := app.db.Query("SELECT `no` FROM users WHERE `id`=? AND `password`=?", body["id"], hash[:]) |
65 | - if err != nil { | 65 | + if err != nil { |
66 | - WriteError(w, http.StatusInternalServerError, "Failed to register") | 66 | + WriteError(w, http.StatusInternalServerError, "Failed to register") |
67 | - return | 67 | + return |
68 | - } | 68 | + } |
69 | - | 69 | + |
70 | - if !rows.Next() { | 70 | + if !rows.Next() { |
71 | - WriteError(w, http.StatusUnauthorized, "Login failed") | 71 | + WriteError(w, http.StatusUnauthorized, "Login failed") |
72 | - return | 72 | + return |
73 | - } | 73 | + } |
74 | - | 74 | + |
75 | - no := uint64(0) | 75 | + no := uint64(0) |
76 | - rows.Scan(&no) | 76 | + rows.Scan(&no) |
77 | - | 77 | + |
78 | - token := jwt.NewWithClaims(jwt.SigningMethodHS256, AuthClaims{UserNo: no}) | 78 | + token := jwt.NewWithClaims(jwt.SigningMethodHS256, AuthClaims{UserNo: no}) |
79 | - auth, err := token.SignedString([]byte(app.Config.TokenSecret)) | 79 | + auth, err := token.SignedString([]byte(app.Config.TokenSecret)) |
80 | - if err != nil { | 80 | + if err != nil { |
81 | - WriteError(w, http.StatusInternalServerError, "Login failed") | 81 | + WriteError(w, http.StatusInternalServerError, "Login failed") |
82 | - return | 82 | + return |
83 | - } | 83 | + } |
84 | - | 84 | + |
85 | - WriteJson(w, map[string]interface{}{"token": auth}) | 85 | + WriteJson(w, map[string]interface{}{"token": auth}) |
86 | -} | 86 | +} |
87 | - | 87 | + |
88 | -func (app *App) WithAuth(next func(http.ResponseWriter, *http.Request)) http.Handler { | 88 | +func (app *App) WithAuth(next func(http.ResponseWriter, *http.Request)) http.Handler { |
89 | - return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { | 89 | + return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { |
90 | - auth := r.Header.Get("Authorization") | 90 | + auth := r.Header.Get("Authorization") |
91 | - if len(auth) > 6 && strings.Index(auth, "Bearer ") == 0 { | 91 | + if len(auth) > 6 && strings.Index(auth, "Bearer ") == 0 { |
92 | - token, err := jwt.ParseWithClaims(auth[7:], &AuthClaims{}, func(token *jwt.Token) (interface{}, error) { | 92 | + token, err := jwt.ParseWithClaims(auth[7:], &AuthClaims{}, func(token *jwt.Token) (interface{}, error) { |
93 | - return []byte(app.Config.TokenSecret), nil | 93 | + return []byte(app.Config.TokenSecret), nil |
94 | - }) | 94 | + }) |
95 | - | 95 | + |
96 | - if err == nil { | 96 | + if err == nil { |
97 | - claims := token.Claims.(*AuthClaims) | 97 | + claims := token.Claims.(*AuthClaims) |
98 | - ctx := context.WithValue(r.Context(), PropUserNo, claims.UserNo) | 98 | + ctx := context.WithValue(r.Context(), PropUserNo, claims.UserNo) |
99 | - next(w, r.WithContext(ctx)) | 99 | + next(w, r.WithContext(ctx)) |
100 | - return | 100 | + return |
101 | - } | 101 | + } |
102 | - } | 102 | + } |
103 | - | 103 | + |
104 | - WriteError(w, http.StatusUnauthorized, "Authorization failed") | 104 | + WriteError(w, http.StatusUnauthorized, "Authorization failed") |
105 | - }) | 105 | + }) |
106 | -} | 106 | +} | ... | ... |
1 | -package main | 1 | +package main |
2 | - | 2 | + |
3 | -import ( | 3 | +import ( |
4 | - "encoding/json" | 4 | + "encoding/json" |
5 | - "net/http" | 5 | + "net/http" |
6 | -) | 6 | +) |
7 | - | 7 | + |
8 | -func WriteJson(w http.ResponseWriter, data interface{}) { | 8 | +func WriteJson(w http.ResponseWriter, data interface{}) { |
9 | - w.Header().Set("Content-Type", "application/json") | 9 | + w.Header().Set("Content-Type", "application/json") |
10 | - json.NewEncoder(w).Encode(data) | 10 | + json.NewEncoder(w).Encode(data) |
11 | -} | 11 | +} |
12 | - | 12 | + |
13 | -func WriteError(w http.ResponseWriter, status int, message string) { | 13 | +func WriteError(w http.ResponseWriter, status int, message string) { |
14 | - w.Header().Set("Content-Type", "application/json") | 14 | + w.Header().Set("Content-Type", "application/json") |
15 | - w.WriteHeader(status) | 15 | + w.WriteHeader(status) |
16 | - json.NewEncoder(w).Encode(map[string]interface{}{"msg": message}) | 16 | + json.NewEncoder(w).Encode(map[string]interface{}{"msg": message}) |
17 | -} | 17 | +} | ... | ... |
-
Please register or login to post a comment