update
This commit is contained in:
罗祥
@ -1,167 +0,0 @@
|
||||
import javafx.util.Pair;
|
||||
|
||||
import java.io.File;
|
||||
import java.io.FileReader;
|
||||
import java.io.FileWriter;
|
||||
import java.io.IOException;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
import java.util.regex.Matcher;
|
||||
import java.util.regex.Pattern;
|
||||
|
||||
import static java.util.regex.Pattern.*;
|
||||
|
||||
/**
|
||||
* @author : heibaiying
|
||||
* @description : 生成导航和图片格式转换
|
||||
*/
|
||||
public class Reformatting {
|
||||
|
||||
/**
|
||||
* GITHUB 用户名
|
||||
**/
|
||||
private static final String GITHUB_USERNAME = "heibaiying";
|
||||
/**
|
||||
* 项目地址
|
||||
**/
|
||||
private static final String PROJECT_NAME = "BigData-Notes";
|
||||
|
||||
public static void main(String[] args) {
|
||||
|
||||
if (args.length < 1) {
|
||||
System.out.println("请输入文件路径");
|
||||
return;
|
||||
}
|
||||
|
||||
String dir = "D:\\BigData-Notes\\notes\\Hbase协处理器.md";
|
||||
|
||||
String preUrl = "https://github.com/" + GITHUB_USERNAME + "/" + PROJECT_NAME + "/blob/master/pictures/";
|
||||
String regex = "(!\\[(\\S*)]\\(\\S:\\\\" + PROJECT_NAME + "\\\\pictures\\\\(\\S*)\\)[^(</br>)]*?)";
|
||||
|
||||
List<String> filesList = getAllFile(dir, new ArrayList<>());
|
||||
for (String filePath : filesList) {
|
||||
// 获取文件内容
|
||||
String content = getContent(filePath);
|
||||
// 修改图片
|
||||
String newContent = changeImageUrl(content, preUrl, regex);
|
||||
// 获取全部标题
|
||||
List<Pair<String, String>> allTitle = getAllTitle(newContent);
|
||||
// 生成导航
|
||||
String nav = genNav(allTitle);
|
||||
// 写出并覆盖原文件
|
||||
write(filePath, newContent, nav);
|
||||
}
|
||||
System.out.println("格式转换成功！");
|
||||
}
|
||||
|
||||
|
||||
private static String changeImageUrl(String content, String preUrl, String oldImageUrlRegex) {
|
||||
|
||||
//github 支持的居中方式 <div align="center"> <img src=""/> </div>
|
||||
return content.replaceAll(oldImageUrlRegex,
|
||||
String.format("<div align=\"center\"> <img src=\"%s$3\"/> </div>", preUrl));
|
||||
|
||||
}
|
||||
|
||||
private static List<String> getAllFile(String dir, List<String> filesList) {
|
||||
File file = new File(dir);
|
||||
//如果是文件 则不遍历
|
||||
if (file.isFile() && file.getName().endsWith(".md")) {
|
||||
filesList.add(file.getAbsolutePath());
|
||||
}
|
||||
//如果是文件夹 则遍历下面的所有文件
|
||||
File[] files = file.listFiles();
|
||||
if (files != null) {
|
||||
for (File f : files) {
|
||||
if (f.isDirectory() && !f.getName().startsWith(".")) {
|
||||
getAllFile(f.getAbsolutePath(), filesList);
|
||||
} else if (f.getName().endsWith(".md")) {
|
||||
filesList.add(f.getAbsolutePath());
|
||||
}
|
||||
}
|
||||
}
|
||||
return filesList;
|
||||
}
|
||||
|
||||
|
||||
private static void write(String filePath, String content, String nav) {
|
||||
try {
|
||||
String newContent = "";
|
||||
if (content.contains("<nav>") && content.contains("</nav>")) {
|
||||
// 如果原来有目录则替换
|
||||
newContent = content.replaceAll("(?m)(<nav>[\\s\\S]*</nav>)", nav);
|
||||
} else {
|
||||
StringBuilder stringBuilder = new StringBuilder(content);
|
||||
// 如果原来没有目录，则title和正文一个标题间写入
|
||||
int index = content.indexOf("## ");
|
||||
stringBuilder.insert(index - 1, nav);
|
||||
newContent = stringBuilder.toString();
|
||||
}
|
||||
// 写出覆盖文件
|
||||
FileWriter fileWriter = new FileWriter(new File(filePath));
|
||||
fileWriter.write(newContent);
|
||||
fileWriter.flush();
|
||||
} catch (IOException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
private static String genNav(List<Pair<String, String>> flagAndTitles) {
|
||||
StringBuilder builder = new StringBuilder();
|
||||
// 目录头
|
||||
builder.append("<nav>\n");
|
||||
for (Pair<String, String> ft : flagAndTitles) {
|
||||
String flag = ft.getKey();
|
||||
String title = ft.getValue();
|
||||
builder.append(genBlank(flag.length() - 2, 4));
|
||||
// Github有效目录格式: <a href="#21-预备">页面锚点</a> url中不能出现特殊符号
|
||||
String formatTitle = title.trim().replaceAll("[.():：（）|、,，@。&/\\\\]", "").replace(" ", "-");
|
||||
builder.append(String.format("<a href=\"%s\">%s</a><br/>\n", "#" + formatTitle, title));
|
||||
}
|
||||
// 目录尾
|
||||
builder.append("</nav>\n");
|
||||
return builder.toString();
|
||||
}
|
||||
|
||||
private static String genBlank(int i, int scale) {
|
||||
StringBuilder builder = new StringBuilder();
|
||||
for (int j = 0; j < i; j++) {
|
||||
for (int k = 0; k < scale; k++) {
|
||||
builder.append(" ");
|
||||
}
|
||||
}
|
||||
return builder.toString();
|
||||
}
|
||||
|
||||
private static List<Pair<String, String>> getAllTitle(String content) {
|
||||
List<Pair<String, String>> list = new ArrayList<>();
|
||||
Pattern pattern = compile("(?m)^(#{2,10})\\s?(.*)");
|
||||
Matcher matcher = pattern.matcher(content);
|
||||
while (matcher.find()) {
|
||||
String group2 = matcher.group(2);
|
||||
if (!group2.contains("参考资料")) {
|
||||
list.add(new Pair<>(matcher.group(1), group2));
|
||||
}
|
||||
}
|
||||
return list;
|
||||
}
|
||||
|
||||
private static String getContent(String filePath) {
|
||||
StringBuilder builder = new StringBuilder();
|
||||
|
||||
try {
|
||||
FileReader reader = new FileReader(filePath);
|
||||
char[] chars = new char[1024 * 1024];
|
||||
|
||||
int read;
|
||||
while ((read = reader.read(chars)) != -1) {
|
||||
builder.append(new String(chars, 0, read));
|
||||
}
|
||||
} catch (IOException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
return builder.toString();
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user