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