1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
package ftbsc.lll.mapper.writer;
import ftbsc.lll.mapper.MapperProvider;
import ftbsc.lll.mapper.utils.Mapper;
import org.apache.commons.cli.*;
import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.*;
/**
* Writes a mapping to a certain format.
*/
public class MappingWriter {
/**
* The main function.
* @param args the command line arguments
* @throws IOException if something goes wrong while writing the file
* @throws ParseException if something goes wrong while parsin arguments
*/
public static void main(String[] args) throws IOException, ParseException {
Options options = new Options()
.addOption("r", "reverse", false, "Writes down inverted mappings")
.addOption("o", "overwrite", false, "Overwrites the file even if it exists");
DefaultParser parser = new DefaultParser();
CommandLine cmdLine = parser.parse(options, args);
args = cmdLine.getArgs();
//separate normal args from custom args
if(args.length < 3) {
System.err.println("Bad argument count!");
System.err.println("java -jar mapping-writer.jar [-r] [-o] <location> <format> <output> [[custom args]]");
return;
}
//load the mapper
List<String> lines = MapperProvider.fetchFromLocalOrRemote(args[0]);
Mapper mapper = !cmdLine.hasOption("reverse")
? MapperProvider.getMapper(lines).getMapper(lines, false)
: MapperProvider.getMapper(lines).getInvertedMapper(lines, false);
//load the writers
Map<String, IWriter> writerMap = new HashMap<>();
for(IWriter w : ServiceLoader.load(IWriter.class))
writerMap.put(w.uniqueId(), w);
//get the one we need
IWriter writer = writerMap.get(args[1]);
if(writer == null) {
System.err.printf("%s was not recognised as a valid format!", args[1]);
return;
}
//now for the file
File targetFile = new File(args[2]);
if(!cmdLine.hasOption("overwrite") || !targetFile.exists()) {
if(!targetFile.createNewFile()) {
System.err.println("File already exists!");
return;
}
}
if(!targetFile.canWrite()) {
System.err.println("Failed to write to file: access denied.");
return;
}
//call the writer
PrintWriter printWriter = new PrintWriter(new FileWriter(targetFile));
writer.write(mapper, printWriter, Arrays.copyOfRange(args, 3, args.length));
printWriter.close();
}
}
|
zaaarf