Moved to Java 8. Code cleanup.
[meviz.git] / src / main / java / eu / svjatoslav / meviz / texttruncate / Main.java
1 /*
2  * Meviz - Various tools collection to work with multimedia.
3  * Copyright (C) 2012, Svjatoslav Agejenko, svjatoslav@svjatoslav.eu
4  *
5  * This program is free software; you can redistribute it and/or
6  * modify it under the terms of version 2 of the GNU General Public License
7  * as published by the Free Software Foundation.
8  */
9
10 package eu.svjatoslav.meviz.texttruncate;
11
12 import eu.svjatoslav.commons.file.FilePathParser;
13 import eu.svjatoslav.commons.string.String2;
14 import eu.svjatoslav.commons.string.WildCardMatcher;
15 import eu.svjatoslav.meviz.Module;
16
17 import java.io.*;
18
19 public class Main implements Module {
20
21     private static final int targetLineLength = 200;
22
23     private static final int maxSplittedLines = 3;
24
25     private static final String splitLinePrefix = "    ";
26
27     private final CommandlineHandler commandlineHandler = new CommandlineHandler();
28
29     private TextTruncatingOptions options;
30
31     private boolean fileMatchesInputPattern(final File file) {
32         final String fileName = file.getName().toLowerCase();
33
34         for (final String inputPattern : options.inputPatterns)
35             if (WildCardMatcher.match(fileName, inputPattern.toLowerCase()))
36                 return true;
37
38         return false;
39     }
40
41     @Override
42     public String getDescription() {
43         return "Truncate text with long lines into multiple shorter lines.";
44     }
45
46     @Override
47     public String getModuleCommand() {
48         return "trunctext";
49     }
50
51     private File getTargetFile(final File sourceFile) {
52         final StringBuilder targetFilePath = new StringBuilder();
53         targetFilePath.append(sourceFile.getParent());
54         targetFilePath.append("/");
55         targetFilePath.append(FilePathParser
56                 .getFileNameWithoutExtension(sourceFile));
57
58         targetFilePath.append(".truncated");
59
60         // add file extension
61         {
62             final String fileExtension = FilePathParser
63                     .getFileExtension(sourceFile);
64
65             if (fileExtension.length() > 0)
66                 targetFilePath.append("." + fileExtension);
67         }
68         return new File(targetFilePath.toString());
69     }
70
71     private void processDirectory(final File directory) {
72
73         for (final File subFile : directory.listFiles())
74             if (subFile.isDirectory()) {
75                 if (options.recursive)
76                     processDirectory(subFile);
77             } else if (fileMatchesInputPattern(subFile))
78                 try {
79                     processFile(subFile);
80                 } catch (final IOException exception) {
81                     System.out.println("Error processing file: "
82                             + subFile.getAbsolutePath());
83                     System.out.println("   exception: "
84                             + exception.getMessage());
85                 }
86
87     }
88
89     private void processFile(final File file) throws IOException {
90         final File targetFile = getTargetFile(file);
91
92         final BufferedReader bufferedReader = new BufferedReader(
93                 new FileReader(file));
94
95         final BufferedWriter bufferedWriter = new BufferedWriter(
96                 new FileWriter(targetFile));
97
98         while (true) {
99             final String readLine = bufferedReader.readLine();
100             if (readLine == null)
101                 break;
102
103             if (readLine.length() <= targetLineLength)
104                 bufferedWriter.write(readLine + "\n");
105             else {
106
107                 final String2 cuttableString = new String2(readLine);
108
109                 bufferedWriter.write(cuttableString.trimLeft(targetLineLength)
110                         + "\n");
111
112                 int splittedLinesCount = 0;
113
114                 while (!cuttableString.isEmpty()) {
115                     splittedLinesCount++;
116                     if (splittedLinesCount >= maxSplittedLines)
117                         break;
118
119                     bufferedWriter.write(splitLinePrefix
120                             + cuttableString.trimLeft(targetLineLength
121                             - splitLinePrefix.length()) + "\n");
122                 }
123             }
124
125         }
126
127         bufferedReader.close();
128         bufferedWriter.close();
129     }
130
131     @Override
132     public void run(final String[] args) throws IOException {
133
134         options = commandlineHandler.parseCommandlineArguments(args);
135
136         if (options == null) {
137             showCommandlineHelp();
138             return;
139         }
140
141         processDirectory(options.targetDirectory);
142
143     }
144
145     @Override
146     public void showCommandlineHelp() {
147         commandlineHandler.parser.showHelp();
148     }
149
150 }