123 lines
3.3 KiB
Go
123 lines
3.3 KiB
Go
package routes
|
|
|
|
import (
|
|
"context"
|
|
"database/sql"
|
|
|
|
//"net/http"
|
|
|
|
"github.com/go-chi/chi/v5"
|
|
"github.com/go-chi/chi/v5/middleware"
|
|
_ "github.com/lib/pq"
|
|
httpSwagger "github.com/swaggo/http-swagger"
|
|
|
|
"github.com/jtom38/newsbot/collector/database"
|
|
"github.com/jtom38/newsbot/collector/services/config"
|
|
)
|
|
|
|
type Server struct {
|
|
Router *chi.Mux
|
|
Db *database.Queries
|
|
ctx *context.Context
|
|
}
|
|
|
|
var (
|
|
ErrIdValueMissing string = "id value is missing"
|
|
ErrValueNotUuid string = "a value given was expected to be a uuid but was not correct."
|
|
ErrNoRecordFound string = "no record was found."
|
|
ErrUnableToConvertToJson string = "Unable to convert to json"
|
|
)
|
|
|
|
func NewServer(ctx context.Context) *Server {
|
|
s := &Server{
|
|
ctx: &ctx,
|
|
}
|
|
|
|
db, err := openDatabase(ctx)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
s.Db = db
|
|
|
|
s.Router = chi.NewRouter()
|
|
s.MountMiddleware()
|
|
s.MountRoutes()
|
|
return s
|
|
}
|
|
|
|
func openDatabase(ctx context.Context) (*database.Queries, error) {
|
|
_env := config.New()
|
|
connString := _env.GetConfig(config.Sql_Connection_String)
|
|
db, err := sql.Open("postgres", connString)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
queries := database.New(db)
|
|
return queries, err
|
|
}
|
|
|
|
func (s *Server) MountMiddleware() {
|
|
s.Router.Use(middleware.Logger)
|
|
s.Router.Use(middleware.Recoverer)
|
|
//s.Router.Use(middleware.Heartbeat())
|
|
}
|
|
|
|
func (s *Server) MountRoutes() {
|
|
s.Router.Get("/swagger/*", httpSwagger.Handler(
|
|
httpSwagger.URL("doc.json"), //The url pointing to API definition
|
|
))
|
|
|
|
/* Root Routes */
|
|
s.Router.Get("/api/helloworld", helloWorld)
|
|
s.Router.Get("/api/hello/{who}", helloWho)
|
|
s.Router.Get("/api/ping", ping)
|
|
|
|
/* Article Routes */
|
|
s.Router.Get("/api/articles", s.listArticles)
|
|
s.Router.Route("/api/articles/{ID}", func(r chi.Router) {
|
|
r.Get("/", s.getArticleById)
|
|
})
|
|
s.Router.Get("/api/articles/by/sourceid", s.GetArticlesBySourceId)
|
|
|
|
/* Discord Queue */
|
|
s.Router.Get("/api/discord/queue", s.GetDiscordQueue)
|
|
|
|
/* Discord WebHooks */
|
|
s.Router.Post("/api/discord/webhooks/new", s.NewDiscordWebHook)
|
|
s.Router.Get("/api/discord/webhooks", s.GetDiscordWebHooks)
|
|
//s.Router.Get("/api/discord/webhooks/byId", s.GetDiscordWebHooksById)
|
|
s.Router.Route("/api/discord/webhooks/{ID}", func(r chi.Router) {
|
|
r.Get("/", s.GetDiscordWebHooksById)
|
|
r.Delete("/", s.deleteDiscordWebHook)
|
|
r.Post("/disable", s.disableDiscordWebHook)
|
|
r.Post("/enable", s.enableDiscordWebHook)
|
|
})
|
|
|
|
/* Settings */
|
|
s.Router.Get("/api/settings", s.getSettings)
|
|
|
|
/* Source Routes */
|
|
s.Router.Get("/api/config/sources", s.listSources)
|
|
s.Router.Get("/api/config/sources/by/source", s.listSourcesBySource)
|
|
|
|
/* Reddit Source Routes */
|
|
|
|
s.Router.Post("/api/config/sources/new/reddit", s.newRedditSource)
|
|
|
|
s.Router.Post("/api/config/sources/new/youtube", s.newYoutubeSource)
|
|
s.Router.Post("/api/config/sources/new/twitch", s.newTwitchSource)
|
|
s.Router.Route("/api/config/sources/{ID}", func(r chi.Router) {
|
|
r.Get("/", s.getSources)
|
|
r.Delete("/", s.deleteSources)
|
|
r.Post("/disable", s.disableSource)
|
|
r.Post("/enable", s.enableSource)
|
|
})
|
|
|
|
/* Subscriptions */
|
|
s.Router.Get("/api/subscriptions", s.ListSubscriptions)
|
|
s.Router.Get("/api/subscriptions/byDiscordId", s.GetSubscriptionsByDiscordId)
|
|
s.Router.Get("/api/subscriptions/bySourceId", s.GetSubscriptionsBySourceId)
|
|
s.Router.Post("/api/subscriptions/new/discordwebhook", s.newDiscordWebHookSubscription)
|
|
}
|