rssagg/main.go at main · divakaivan/rssagg · 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
package main

import (
	"database/sql"
	"log"
	"net/http"
	"os"
	"time"

	"github.com/divakaivan/rssagg/internal/database"
	"github.com/divakaivan/rssagg/internal/handlers"
	"github.com/divakaivan/rssagg/internal/rss"
	"github.com/go-chi/chi/v5"
	"github.com/go-chi/cors"
	httpSwagger "github.com/swaggo/http-swagger/v2"

	_ "github.com/divakaivan/rssagg/docs"
	_ "github.com/lib/pq"
)

//	@title			RSS Aggregator API
//	@version		1.0
//	@description This is a sample server for RSS Aggregator.

//	@contact.name	Ivan
//	@contact.url	https://github.com/divakaivan

//	@license.name	Apache 2.0
//	@license.url	http://www.apache.org/licenses/LICENSE-2.0.html

// @host		localhost
// @BasePath	:8080/v1
func main() {

	portString := os.Getenv("PORT")
	if portString == "" {
		log.Fatal("[Error] PORT not found in env")
	}
	dbUrl := os.Getenv("DB_URL")
	if dbUrl == "" {
		log.Fatal("[Error] dbUrl not found in env")
	}

	conn, err := sql.Open("postgres", dbUrl)
	if err != nil {
		log.Fatal("[Error] Cannot connect to database: ", err)
	}

	db := database.New(conn)
	queries := database.New(conn)
	api := handlers.NewAPI(queries)
	// need to start it on its own goroutine
	// so it doesn't block the main thread
	go rss.StartScraping(db, 5, time.Minute)

	router := chi.NewRouter()

	router.Use(cors.Handler(cors.Options{
		AllowedOrigins:   []string{"https://*", "http://*"},
		AllowedMethods:   []string{"GET", "POST", "PUT", "DELETE", "OPTIONS"},
		AllowedHeaders:   []string{"*"},
		ExposedHeaders:   []string{"Link"},
		AllowCredentials: false,
		MaxAge:           300,
	}))

	router.Use(api.LogToDbMiddleware)

	v1Router := chi.NewRouter()
	v1Router.Get("/healthz", handlers.HandlerReadiness)
	v1Router.Get("/err", handlers.HandlerErr)
	v1Router.Post("/users", api.HandlerCreateUser)
	v1Router.Get("/users", api.AuthMiddleware(api.HandlerGetUser))

	v1Router.Post("/feeds", api.AuthMiddleware(api.HandlerCreateFeed))
	v1Router.Get("/feeds", api.HandlerGetFeeds)

	v1Router.Get("/posts", api.AuthMiddleware(api.HandlerGetPostsForUser))

	v1Router.Post("/feed_follows", api.AuthMiddleware(api.HandlerCreateFeedFollow))
	v1Router.Get("/feed_follows", api.AuthMiddleware(api.HandlerGetFeedFollows))
	v1Router.Delete("/feed_follows/{feedFollowID}", api.AuthMiddleware(api.HandlerDeleteFeedFollow))

	router.Mount("/v1", v1Router)
	router.Get("/swagger/*", httpSwagger.Handler(
		httpSwagger.URL("http://localhost:8080/swagger/doc.json"), //The url pointing to API definition
	))
	srv := &http.Server{
		Handler: router,
		Addr:    ":" + portString,
	}

	log.Printf("[Info] Server listening on port %v", portString)
	err = srv.ListenAndServe()
	if err != nil {
		log.Fatal("[Error] Server failed to start: ", err)
	}
}