git.r.bdr.sh - rbdr/lyricli/blob - Sources/main.swift

   1 import CommandLineKit
   2 import Foundation
   3
   4 func main() {
   5     let (flags, parser) = createParser()
   6
   7     do {
   8         try parser.parse()
   9     } catch {
  10         parser.printUsage(error)
  11         exit(EX_USAGE)
  12     }
  13
  14     // Boolean Options
  15
  16     checkHelpFlag(flags["help"], withParser: parser)
  17     checkVersionFlag(flags["version"], withParser: parser)
  18     checkListSourcesFlag(flags["listSources"], withParser: parser)
  19     checkTitleFlag(flags["title"], withParser: parser)
  20
  21     // String Options
  22
  23     checkEnableSourceFlag(flags["enableSource"], withParser: parser)
  24     checkDisableSourceFlag(flags["disableSource"], withParser: parser)
  25     checkResetSourceFlag(flags["resetSource"], withParser: parser)
  26
  27     // Remove any flags so anyone after this gets the unprocessed values
  28
  29     let programName: [String] = [CommandLine.arguments[0]]
  30     CommandLine.arguments = programName + parser.unparsedArguments
  31
  32     // Run Lyricli
  33
  34     Lyricli.printLyrics()
  35 }
  36
  37 /// Sets up and returns a new options parser
  38 ///
  39 /// - Returns: A Dictionary of Options, and a new CommandLineKit instance
  40 private func createParser() -> ([String:Option], CommandLineKit) {
  41     let parser = CommandLineKit()
  42     var flags: [String:Option] = [:]
  43
  44     flags["help"] = BoolOption(shortFlag: "h", longFlag: "help", helpMessage: "Prints a help message.")
  45     flags["version"] = BoolOption(shortFlag: "v", longFlag: "version", helpMessage: "Prints the version.")
  46
  47     flags["enableSource"] = StringOption(shortFlag: "e", longFlag: "enable-source", helpMessage: "Enables a source")
  48     flags["disableSource"] = StringOption(shortFlag: "d", longFlag: "disable-source", helpMessage: "Disables a source")
  49     flags["resetSource"] = StringOption(shortFlag: "r", longFlag: "reset-source", helpMessage: "Resets a source")
  50     flags["listSources"] = BoolOption(shortFlag: "l", longFlag: "list-sources", helpMessage: "Lists all sources")
  51
  52     flags["title"] = BoolOption(shortFlag: "t", longFlag: "title", helpMessage: "Shows title of song if true")
  53
  54     parser.addOptions(Array(flags.values))
  55
  56     parser.formatOutput = {parseString, type in
  57
  58         var formattedString: String
  59
  60         switch type {
  61         case .About:
  62             formattedString = "\(parseString) [<artist_name> <song_name>]"
  63             break
  64         default:
  65             formattedString = parseString
  66         }
  67
  68         return parser.defaultFormat(formattedString, type: type)
  69     }
  70
  71     return (flags, parser)
  72 }
  73
  74 // Handle the Help flag
  75
  76 private func checkHelpFlag(_ flag: Option?, withParser parser: CommandLineKit) {
  77     if let helpFlag = flag as? BoolOption {
  78         if helpFlag.value {
  79             parser.printUsage()
  80             exit(0)
  81         }
  82     }
  83 }
  84
  85 // Handle the version flag
  86
  87 private func checkVersionFlag(_ flag: Option?, withParser parser: CommandLineKit) {
  88     if let versionFlag = flag as? BoolOption {
  89         if versionFlag.value {
  90             print(Lyricli.version)
  91             exit(0)
  92         }
  93     }
  94 }
  95
  96 // Handle the list sources flag
  97
  98 private func checkListSourcesFlag(_ flag: Option?, withParser parser: CommandLineKit) {
  99     if let listSourcesFlag = flag as? BoolOption {
 100         if listSourcesFlag.value {
 101             Lyricli.printSources()
 102             exit(0)
 103         }
 104     }
 105 }
 106
 107 // Handle the title flag
 108
 109 private func checkTitleFlag(_ flag: Option?, withParser parser: CommandLineKit) {
 110     if let titleFlag = flag as? BoolOption {
 111         if titleFlag.value {
 112             Lyricli.showTitle = true
 113         }
 114     }
 115 }
 116
 117 // Handle the enable source flag
 118
 119 private func checkEnableSourceFlag(_ flag: Option?, withParser parser: CommandLineKit) {
 120     if let enableSourceFlag = flag as? StringOption {
 121         if let source = enableSourceFlag.value {
 122             Lyricli.enableSource(source)
 123             exit(0)
 124         }
 125     }
 126 }
 127
 128 // Handle the disable source flag
 129
 130 private func checkDisableSourceFlag(_ flag: Option?, withParser parser: CommandLineKit) {
 131     if let disableSourceFlag = flag as? StringOption {
 132         if let source = disableSourceFlag.value {
 133             Lyricli.disableSource(source)
 134             exit(0)
 135         }
 136     }
 137 }
 138
 139 // Handle the reset source flag
 140
 141 private func checkResetSourceFlag(_ flag: Option?, withParser parser: CommandLineKit) {
 142     if let resetSourceFlag = flag as? StringOption {
 143         if let source = resetSourceFlag.value {
 144             Lyricli.resetSource(source)
 145             exit(0)
 146         }
 147     }
 148 }
 149
 150 main()