-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.go
83 lines (67 loc) · 2.43 KB
/
server.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
package main
import (
"context"
"log"
"time"
"github.com/awildboop/gosharex/common"
"github.com/awildboop/gosharex/handlers/api"
"github.com/awildboop/gosharex/handlers/features"
"github.com/gin-gonic/gin"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
)
func main() {
conf, err := common.LoadConfiguration("./config.yaml")
if err != nil {
log.Fatalf("Encountered error while loading configuration file\n%v\n", err)
}
cfFeatures := conf.Features
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
todo := context.TODO()
defer cancel()
client, err := mongo.Connect(ctx, options.Client().ApplyURI(conf.GetURI()))
if err != nil {
log.Fatalf("Encountered error while connecting to MongoDB database\n%v\n", err)
}
defer client.Disconnect(ctx)
err = client.Ping(context.TODO(), nil)
if err != nil {
log.Fatalf("Encountered error while testing MongoDB database\n%v\n", err)
}
redirects := client.Database(conf.MongoDB.DB).Collection("redirects")
texts := client.Database(conf.MongoDB.DB).Collection("texts")
images := client.Database(conf.MongoDB.DB).Collection("images")
r := gin.Default()
r.LoadHTMLGlob("templates/*")
// r = redirect (shortener), i = image, t = text, f = file,
// potentially merge image/text/file into a single one since they really are all just files in the end
if cfFeatures.EnableRedirector {
r.GET("/r/*identifier", features.HandleRedirect(redirects, conf, todo))
r.POST("/r", api.CreateRedirect(redirects, conf, todo))
r.PUT("/r", api.CreateRedirect(redirects, conf, todo))
}
if cfFeatures.EnableText {
r.GET("/t/*identifier", features.HandleText(texts, conf, todo))
r.POST("/t", api.CreateText(texts, conf, todo))
r.PUT("/t", api.CreateText(texts, conf, todo))
}
if cfFeatures.EnableImages {
r.GET("/i/*identifier", features.HandleImage(images, conf, todo))
r.GET("/ri/*identifier", features.HandleRawImage(images, conf, todo))
r.POST("/i", api.CreateImage(images, conf, todo))
r.PUT("/i", api.CreateImage(images, conf, todo))
}
if cfFeatures.API.EnableAPI {
apiFeatures := cfFeatures.API
v1 := r.Group("v1")
if apiFeatures.ManageRedirects {
v1.GET("/r", api.GetRedirect(redirects, todo)) // Returns information & stats about the redirect
v1.DELETE("/r", api.DeleteRedirect(redirects, todo))
}
// if apiFeatures.ManageText {
// }
// if apiFeatures.ManageImages {
// }
}
r.Run(conf.GetWebserverAddress())
}