2020-01-30 14:28:14 +01:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
|
|
|
"log"
|
|
|
|
"net/http"
|
2020-02-19 21:33:28 +01:00
|
|
|
"path"
|
2021-02-20 22:43:07 +01:00
|
|
|
"time"
|
2020-01-30 14:28:14 +01:00
|
|
|
|
2020-10-04 16:34:47 +02:00
|
|
|
"github.com/gorilla/handlers"
|
2020-02-21 16:50:50 +01:00
|
|
|
"github.com/gorilla/mux"
|
|
|
|
|
2020-02-05 14:51:46 +01:00
|
|
|
"github.com/joho/godotenv"
|
2020-01-30 14:49:39 +01:00
|
|
|
|
2020-12-17 22:51:43 +01:00
|
|
|
"github.com/photoview/photoview/api/database"
|
|
|
|
"github.com/photoview/photoview/api/graphql/auth"
|
2021-02-07 22:17:33 +01:00
|
|
|
"github.com/photoview/photoview/api/graphql/dataloader"
|
2020-12-17 22:51:43 +01:00
|
|
|
"github.com/photoview/photoview/api/routes"
|
|
|
|
"github.com/photoview/photoview/api/scanner"
|
2021-02-15 17:35:28 +01:00
|
|
|
"github.com/photoview/photoview/api/scanner/face_detection"
|
2020-12-17 22:51:43 +01:00
|
|
|
"github.com/photoview/photoview/api/server"
|
|
|
|
"github.com/photoview/photoview/api/utils"
|
2020-01-30 14:49:39 +01:00
|
|
|
|
2020-01-30 14:28:14 +01:00
|
|
|
"github.com/99designs/gqlgen/handler"
|
2020-12-17 22:51:43 +01:00
|
|
|
photoview_graphql "github.com/photoview/photoview/api/graphql"
|
|
|
|
"github.com/photoview/photoview/api/graphql/resolvers"
|
2020-01-30 14:28:14 +01:00
|
|
|
)
|
|
|
|
|
|
|
|
func main() {
|
2020-01-30 14:49:39 +01:00
|
|
|
|
2021-03-03 13:32:44 +01:00
|
|
|
log.Println("Starting Photoview...")
|
|
|
|
|
2020-01-30 14:49:39 +01:00
|
|
|
if err := godotenv.Load(); err != nil {
|
|
|
|
log.Println("No .env file found")
|
|
|
|
}
|
|
|
|
|
2021-01-17 16:50:48 +01:00
|
|
|
devMode := utils.DevelopmentMode()
|
2020-02-19 22:28:21 +01:00
|
|
|
|
2020-04-15 16:18:51 +02:00
|
|
|
db, err := database.SetupDatabase()
|
|
|
|
if err != nil {
|
|
|
|
log.Panicf("Could not connect to database: %s\n", err)
|
|
|
|
}
|
2020-01-31 15:22:58 +01:00
|
|
|
|
|
|
|
// Migrate database
|
|
|
|
if err := database.MigrateDatabase(db); err != nil {
|
2020-04-15 16:20:09 +02:00
|
|
|
log.Panicf("Could not migrate database: %s\n", err)
|
2020-01-31 15:22:58 +01:00
|
|
|
}
|
|
|
|
|
2020-09-21 21:32:30 +02:00
|
|
|
if err := scanner.InitializeScannerQueue(db); err != nil {
|
|
|
|
log.Panicf("Could not initialize scanner queue: %s\n", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := scanner.InitializePeriodicScanner(db); err != nil {
|
|
|
|
log.Panicf("Could not initialize periodic scanner: %s", err)
|
|
|
|
}
|
2020-06-23 00:40:47 +02:00
|
|
|
|
2020-09-23 14:34:26 +02:00
|
|
|
scanner.InitializeExecutableWorkers()
|
|
|
|
|
2021-02-15 17:35:28 +01:00
|
|
|
if err := face_detection.InitializeFaceDetector(db); err != nil {
|
|
|
|
log.Panicf("Could not initialize face detector: %s\n", err)
|
|
|
|
}
|
|
|
|
|
2020-02-21 16:50:50 +01:00
|
|
|
rootRouter := mux.NewRouter()
|
2020-01-31 23:30:34 +01:00
|
|
|
|
2021-02-07 22:17:33 +01:00
|
|
|
rootRouter.Use(dataloader.Middleware(db))
|
2020-02-21 20:51:50 +01:00
|
|
|
rootRouter.Use(auth.Middleware(db))
|
|
|
|
rootRouter.Use(server.LoggingMiddleware)
|
2020-02-21 16:50:50 +01:00
|
|
|
rootRouter.Use(server.CORSMiddleware(devMode))
|
2020-02-19 21:33:28 +01:00
|
|
|
|
2020-02-05 14:51:46 +01:00
|
|
|
graphqlResolver := resolvers.Resolver{Database: db}
|
2020-01-31 23:30:34 +01:00
|
|
|
graphqlDirective := photoview_graphql.DirectiveRoot{}
|
|
|
|
graphqlDirective.IsAdmin = photoview_graphql.IsAdmin(db)
|
|
|
|
|
|
|
|
graphqlConfig := photoview_graphql.Config{
|
|
|
|
Resolvers: &graphqlResolver,
|
|
|
|
Directives: graphqlDirective,
|
|
|
|
}
|
2020-01-30 14:49:39 +01:00
|
|
|
|
2021-03-01 22:03:57 +01:00
|
|
|
apiListenURL := utils.ApiListenUrl()
|
2020-02-19 21:33:28 +01:00
|
|
|
|
2021-03-01 22:03:57 +01:00
|
|
|
endpointRouter := rootRouter.PathPrefix(apiListenURL.Path).Subrouter()
|
2020-02-21 16:50:50 +01:00
|
|
|
|
|
|
|
if devMode {
|
2021-03-01 22:03:57 +01:00
|
|
|
endpointRouter.Handle("/", handler.Playground("GraphQL playground", path.Join(apiListenURL.Path, "/graphql")))
|
2020-02-21 16:50:50 +01:00
|
|
|
} else {
|
2020-04-06 22:16:25 +02:00
|
|
|
endpointRouter.HandleFunc("/", func(w http.ResponseWriter, req *http.Request) {
|
2020-02-21 16:50:50 +01:00
|
|
|
w.Write([]byte("photoview api endpoint"))
|
|
|
|
})
|
|
|
|
}
|
2020-02-19 22:28:21 +01:00
|
|
|
|
2020-04-06 22:16:25 +02:00
|
|
|
endpointRouter.Handle("/graphql",
|
2020-02-21 17:53:04 +01:00
|
|
|
handler.GraphQL(photoview_graphql.NewExecutableSchema(graphqlConfig),
|
|
|
|
handler.IntrospectionEnabled(devMode),
|
|
|
|
handler.WebsocketUpgrader(server.WebsocketUpgrader(devMode)),
|
2021-02-20 22:43:07 +01:00
|
|
|
handler.WebsocketKeepAliveDuration(time.Second*10),
|
2020-02-21 17:53:04 +01:00
|
|
|
handler.WebsocketInitFunc(auth.AuthWebsocketInit(db)),
|
|
|
|
),
|
|
|
|
)
|
2020-01-30 14:28:14 +01:00
|
|
|
|
2020-04-06 22:16:25 +02:00
|
|
|
photoRouter := endpointRouter.PathPrefix("/photo").Subrouter()
|
2020-02-21 16:50:50 +01:00
|
|
|
routes.RegisterPhotoRoutes(db, photoRouter)
|
2020-02-09 12:53:21 +01:00
|
|
|
|
2020-07-11 15:57:58 +02:00
|
|
|
videoRouter := endpointRouter.PathPrefix("/video").Subrouter()
|
|
|
|
routes.RegisterVideoRoutes(db, videoRouter)
|
|
|
|
|
2021-01-17 16:50:48 +01:00
|
|
|
shouldServeUI := utils.ShouldServeUI()
|
2020-04-06 22:16:25 +02:00
|
|
|
|
|
|
|
if shouldServeUI {
|
2021-02-26 15:10:28 +01:00
|
|
|
spa := routes.NewSpaHandler(utils.UIPath(), "index.html")
|
2020-04-06 22:16:25 +02:00
|
|
|
rootRouter.PathPrefix("/").Handler(spa)
|
|
|
|
}
|
2020-03-01 02:06:18 +01:00
|
|
|
|
2020-02-19 22:28:21 +01:00
|
|
|
if devMode {
|
2021-03-01 22:03:57 +01:00
|
|
|
log.Printf("🚀 Graphql playground ready at %s\n", apiListenURL.String())
|
2020-02-19 22:28:21 +01:00
|
|
|
} else {
|
2021-03-01 22:03:57 +01:00
|
|
|
log.Printf("Photoview API endpoint listening at %s\n", apiListenURL.String())
|
2020-04-06 22:16:25 +02:00
|
|
|
|
|
|
|
apiEndpoint := utils.ApiEndpointUrl()
|
|
|
|
log.Printf("Photoview API public endpoint ready at %s\n", apiEndpoint.String())
|
2021-03-01 22:03:57 +01:00
|
|
|
|
|
|
|
if uiEndpoint := utils.UiEndpointUrl(); uiEndpoint != nil {
|
|
|
|
log.Printf("Photoview UI public endpoint ready at %s\n", uiEndpoint.String())
|
2021-03-01 22:10:20 +01:00
|
|
|
} else {
|
|
|
|
log.Println("Photoview UI public endpoint ready at /")
|
2021-03-01 22:03:57 +01:00
|
|
|
}
|
2020-04-05 18:54:13 +02:00
|
|
|
|
2020-04-30 14:59:29 +02:00
|
|
|
if !shouldServeUI {
|
2021-01-17 16:50:48 +01:00
|
|
|
log.Printf("Notice: UI is not served by the the api (%s=0)", utils.EnvServeUI.GetName())
|
2020-04-30 14:59:29 +02:00
|
|
|
}
|
|
|
|
|
2020-02-19 22:28:21 +01:00
|
|
|
}
|
|
|
|
|
2021-03-01 22:03:57 +01:00
|
|
|
log.Panic(http.ListenAndServe(":"+apiListenURL.Port(), handlers.CompressHandler(rootRouter)))
|
2020-01-30 14:28:14 +01:00
|
|
|
}
|