Added register functionality and credentials creation to the REST API.

Author: sebinside

src/main/scala/org/codeoverflow/chatoverflow/configuration/CredentialsService.scala

@@ -48,6 +48,13 @@ class CredentialsService(val credentialsFilePath: String) extends WithLogger {
     }
   }
 
+  /**
+    * Returns if the credentials file exists. Can be used to determine if login or register is the right method.
+    *
+    * @return true, if the file was previously created
+    */
+  def credentialsFileExists(): Boolean = new File(credentialsFilePath).exists()
+
   /**
     * Loads the credentials form the credentials file and decrypts them.
     *

src/main/scala/org/codeoverflow/chatoverflow/connector/Connector.scala

@@ -34,6 +34,11 @@ abstract class Connector(val sourceIdentifier: String) extends WithLogger {
     */
   def setCredentials(credentials: Credentials): Unit = this.credentials = Some(credentials)
 
+  /**
+    * Removes the entire credentials object from the connector.
+    */
+  def removeCredentials(): Unit = this.credentials = None
+
   def setCredentialsValue(key: String, value: String): Boolean = {
     if (credentials.isEmpty) false else {
       if (credentials.get.exists(key)) {

src/main/scala/org/codeoverflow/chatoverflow/ui/web/rest/config/ConfigController.scala

@@ -51,5 +51,31 @@ class ConfigController(implicit val swagger: Swagger) extends JsonServlet with C
     }
   }
 
-  // TODO: Handle first time login / Creation of the config files (same as postLogin above...?)
+  get("/register", operation(getRegister)) {
+    chatOverflow.credentialsService.credentialsFileExists()
+  }
+
+  post("/register", operation(postRegister)) {
+    parsedAs[Password] {
+      case Password(password) =>
+
+        if (chatOverflow.credentialsService.credentialsFileExists()) {
+          ResultMessage(success = false, "Already registered.")
+
+          // Extreme exotic case. Wtf?
+        } else if (chatOverflow.isLoaded) {
+          ResultMessage(success = false, "Framework already loaded.")
+
+        } else {
+          chatOverflow.credentialsService.setPassword(password.toCharArray)
+
+          if (!chatOverflow.load()) {
+            ResultMessage(success = false, "Unable to load.")
+          } else {
+
+            ResultMessage(success = true, CryptoUtil.generateAuthKey(password))
+          }
+        }
+    }
+  }
 }

src/main/scala/org/codeoverflow/chatoverflow/ui/web/rest/config/ConfigControllerDefinition.scala

@@ -27,5 +27,17 @@ trait ConfigControllerDefinition extends SwaggerSupport {
       description "Tries to decrypt the credentials using the provided password. If already loaded, does only return the communication auth key."
       parameter bodyParam[Password]("body").description("Requires the user framework password. The auth key is based on this input."))
 
+  val getRegister: OperationBuilder =
+    (apiOperation[Boolean]("getRegister")
+      summary "Returns if a credentials file has been created and thus a specific password been registered."
+      description "Returns true, if the credentials file (with set password) already exists.")
+
+  val postRegister: OperationBuilder =
+    (apiOperation[ResultMessage]("postRegister")
+      summary "Registers the user with the given password. Can only be called if there is no credentials file."
+      description "Creates a credentials file with the given password. Acts like post(login) after this (returning an auth key)."
+      parameter bodyParam[Password]("body").description("Requires the user framework password. The auth key is based on this input."))
+
+
   override protected def applicationDescription: String = "Handles configuration and information retrieval."
 }

src/main/scala/org/codeoverflow/chatoverflow/ui/web/rest/connector/ConnectorController.scala

@@ -1,5 +1,6 @@
 package org.codeoverflow.chatoverflow.ui.web.rest.connector
 
+import org.codeoverflow.chatoverflow.configuration.Credentials
 import org.codeoverflow.chatoverflow.connector.ConnectorRegistry
 import org.codeoverflow.chatoverflow.ui.web.JsonServlet
 import org.codeoverflow.chatoverflow.ui.web.rest.DTOs.{ConnectorDetails, ConnectorRef, ResultMessage}
@@ -23,7 +24,6 @@ class ConnectorController(implicit val swagger: Swagger) extends JsonServlet wit
   }
 
   post("/", operation(postConnector)) {
-    // TODO: Should implicitly create the credentials object
     parsedAs[ConnectorRef] {
       case ConnectorRef(sourceIdentifier, uniqueTypeString) =>
         val connector = ConnectorRegistry.getConnector(sourceIdentifier, uniqueTypeString)
@@ -40,15 +40,21 @@ class ConnectorController(implicit val swagger: Swagger) extends JsonServlet wit
             ResultMessage(success = false, "Unable to add connector.")
 
           } else {
-            chatOverflow.save()
-            ResultMessage(success = true)
+
+            val credentials = new Credentials(sourceIdentifier)
+            if (!ConnectorRegistry.setConnectorCredentials(sourceIdentifier, uniqueTypeString, credentials)) {
+              ResultMessage(success = false, "Unable to create credentials.")
+
+            } else {
+              chatOverflow.save()
+              ResultMessage(success = true)
+            }
           }
         }
     }
   }
 
   delete("/:sourceIdentifier/:qualifiedConnectorType", operation(deleteConnector)) {
-    // TODO: Should implicitly delete the credentials object
     val sourceIdentifier = params("sourceIdentifier")
     val qualifiedConnectorType = params("qualifiedConnectorType")
 
@@ -64,6 +70,7 @@ class ConnectorController(implicit val swagger: Swagger) extends JsonServlet wit
       ResultMessage(success = false, "Unable to remove connector.")
 
     } else {
+      connector.get.removeCredentials()
       chatOverflow.save()
       ResultMessage(success = true)
     }