RunServer.kt

/* Copyright 2016-present The KotlinNLP Authors. All Rights Reserved.
 *
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, you can obtain one at http://mozilla.org/MPL/2.0/.
 * ------------------------------------------------------------------*/

package com.kotlinnlp.nlpserver

import com.kotlinnlp.nlpserver.routes.*
import com.xenomachina.argparser.mainBody
import org.apache.log4j.*

/**
 * Run the NLP Server.
 *
 * Launch with the '-h' option for help about the command line arguments.
 */
fun main(args: Array<String>): Unit = mainBody {

  val parsedArgs = CommandLineArguments(args)
  val logger = Logger.getLogger("Main").setAppender()

  setupLogging(debugMode = parsedArgs.debug)

  NLPServer(port = parsedArgs.port, enableCORS = parsedArgs.enableCORS, routes = buildRoutes(parsedArgs)).start()

  if (parsedArgs.debug) logger.info("Debug mode enabled")
  logger.info("Parallel threads: ${parsedArgs.threads}")
}

/**
 * @param parsedArgs the parsed command line arguments
 *
 * @return the list of routes to make available in the server, depending on the given arguments
 */
private fun buildRoutes(parsedArgs: CommandLineArguments): List<Route> {

  val builder = NLPBuilder(parsedArgs)
  val routes: MutableList<Route> = mutableListOf()

  if (builder.languageDetector != null)
    routes.add(DetectLanguage(builder.languageDetector))

  if (builder.tokenizers != null)
    routes.add(Tokenize(tokenizers = builder.tokenizers, languageDetector = builder.languageDetector))

  if (builder.parsers != null && builder.tokenizers != null)
    routes.add(
      Parse(
        languageDetector = builder.languageDetector,
        tokenizers = builder.tokenizers,
        parsers = builder.parsers,
        morphoPreprocessors = builder.morphoPreprocessors))

  if (builder.locationsDictionary != null && builder.tokenizers != null)
    routes.add(
      Locations(
        languageDetector = builder.languageDetector,
        tokenizers = builder.tokenizers,
        dictionary = builder.locationsDictionary,
        nerLabelers = builder.nerLabelers))

  if (builder.tokenizers != null && builder.frameExtractors != null)
    routes.add(
      Frames(
        languageDetector = builder.languageDetector,
        tokenizers = builder.tokenizers,
        frameExtractors = builder.frameExtractors))

  if (builder.tokenizers != null && builder.classifiers != null)
    routes.add(
      Categorize(
        languageDetector = builder.languageDetector,
        tokenizers = builder.tokenizers,
        classifiers = builder.classifiers))

  if (builder.tokenizers != null && builder.labelers != null)
    routes.add(
      Label(
        languageDetector = builder.languageDetector,
        tokenizers = builder.tokenizers,
        labelers = builder.labelers))

  if (builder.comparators != null)
    routes.add(
      Compare(
        languageDetector = builder.languageDetector,
        comparators = builder.comparators,
        parallelization = parsedArgs.threads))

  if (builder.tokenizers != null && builder.morphoAnalyzers.isNotEmpty())
    routes.add(
      Morpho(
        languageDetector = builder.languageDetector,
        tokenizers = builder.tokenizers,
        analyzers = builder.morphoAnalyzers))

  if (builder.parsers != null && builder.tokenizers != null)
    routes.add(
      Summarize(
        languageDetector = builder.languageDetector,
        tokenizers = builder.tokenizers,
        parsers = builder.parsers,
        morphoPreprocessors = builder.morphoPreprocessors,
        lemmasBlacklist = builder.summaryBlacklists))

  return routes
}