acrisci / commander-scala Goto Github PK

import com.github.acrisci.commander.Program
import java.io.File

abstract sealed case class Options(
  consumerKey: String,
  consumerSecret: String,
  accessToken: String,
  accessTokenSecret: String,
  overWrite: Boolean = false,
  tweetDirectory: File = new File("~/twitterClassifier/tweets/"),
  numTweetsToCollect: Int = 100,
  intervalSecs: Int = 1,
  partitionsEachInterval: Int = 1
)

object Options {
  def apply(args: Array[String]): Options = {
    val program = new Program()
      .version("2.0.0")
      .option(flags="-c, --consumerKey [type]",       description="Twitter OAuth Consumer Key",        required=true)
      .option(flags="-s, --consumerSecret [type]",    description="Twitter OAuth Consumer Secret",     required=true)
      .option(flags="-t, --accessToken [type]",       description="Twitter OAuth Access Token",        required=true)
      .option(flags="-v, --accessTokenSecret [type]", description="Twitter OAuth Access Token Secret", required=true)
      .option(flags="-x, --overWrite",                description="Overwrite data files on each run [false]", default=false)
      .parse(args)

    if (args.isEmpty) program.help

    new Options(
      consumerKey = program.consumerKey,
      consumerSecret = program.consumerSecret,
      accessToken = program.accessToken,
      accessTokenSecret = program.accessTokenSecret,
      overWrite = program.overWrite
     // mandatory unnamed parameters would go here
    ){}
  }
}