1 /*
2  * sys-emu - A system emulator for tutorials
3  * Copyright (C) 2018 - 2019 osdevelopment-info
4  *
5  * This program is free software: you can redistribute it and/or modify
6  * it under the terms of the GNU Affero General Public License as published by
7  * the Free Software Foundation, either version 3 of the License, or
8  * (at your option) any later version.
9  *
10  * This program is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13  * GNU Affero General Public License for more details.
14  *
15  * You should have received a copy of the GNU Affero General Public License
16  * along with this program.  If not, see <https://www.gnu.org/licenses/>.
17  */
18 package info.osdevelopment.sysemu.rest
19 
20 import akka.actor.{Actor, ActorLogging, ActorSystem, Props}
21 import akka.event.slf4j.SLF4JLogging
22 import akka.http.scaladsl.Http
23 import akka.http.scaladsl.Http.ServerBinding
24 import akka.http.scaladsl.model.StatusCodes
25 import akka.http.scaladsl.server.Directives._
26 import akka.http.scaladsl.server.Route
27 import info.osdevelopment.sysemu.config.Configuration
28 import info.osdevelopment.sysemu.system.SystemRepository
29 import io.swagger.v3.oas.annotations.OpenAPIDefinition
30 import io.swagger.v3.oas.annotations.info.{Contact, Info, License}
31 
32 import scala.concurrent.{ExecutionContextExecutor, Future}
33 
34 @OpenAPIDefinition(
35   info =  new Info(
36     title = "sys-emu",
37     version = "0.1",
38     description = "Remote interface for sys-emu",
39     license = new License(name = "GNU AGPLv3", url = "https://www.gnu.org/licenses/agpl.html"),
40     contact = new Contact(url = "https://sys-emu.osdevelopment.info")
41   )
42 )
43 class RestServer(implicit val actorSystem: ActorSystem, val config: Configuration) extends SLF4JLogging {
44 
45   object ShutdownActor {
46 
47     def props(): Props = Props(new ShutdownActor())
48 
49   }
50 
51   class ShutdownActor extends Actor with ActorLogging {
52 
53     override def receive: Actor.Receive = {
54       case "shutdown" => shutdown()
55     }
56 
57   }
58 
59   var bindingFuture: Option[Future[ServerBinding]] = None
60 
61   implicit val executionContext: ExecutionContextExecutor = actorSystem.dispatcher
62   val shutdownActor = actorSystem.actorOf(ShutdownActor.props())
63   val systemRepository = SystemRepository()
64 
65   def start(): Unit = {
66     bindingFuture match {
67       case None => bindingFuture = Some(Http().bindAndHandle(this.route, config.serviceHost, config.servicePort))
68       case _ => log.warn("Tried to start an already running server")
69     }
70   }
71 
72   def shutdown(): Unit = {
73     bindingFuture match {
74       case Some(future) =>
75         future.flatMap(_.unbind())
76         actorSystem.terminate()
77       case None => log.warn("Tried to shutdown a not running server")
78     }
79   }
80 
81   def route: Route = {
82     new RestProcessorService().route ~
83     new RestSystemService(systemRepository).route ~
84     SwaggerService.routes ~
85     path("swagger-ui") {
86       getFromResource("swagger-ui/index.html")
87     } ~
88     getFromResourceDirectory("swagger-ui") ~
89     path("shutdown") {
90       post {
91         shutdownActor ! "shutdown"
92         complete(StatusCodes.OK)
93       }
94     }
95   }
96 
97 }