markdown图片地址批量下载到本地-526互联

`
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.nio.charset.Charset;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.AbstractMap;
import java.util.ArrayList;
import java.util.List;
import java.util.ListIterator;
import java.util.Map;
import java.util.Objects;
import java.util.SortedMap;
import java.util.TreeMap;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Collectors;
import java.util.stream.Stream;

/**

software：IntelliJ IDEA 2023.1.1
class name: MarkdownImageDownloader
class description: markdown 图片下载程序
@author MoBaiJun 2023/4/21 17:23
/
public class MarkdownImageDownloader {
/*
- 存储图片的文件夹名称
  */
  private static final String IMG_FOLDER = "img";
/**
- 正则表达式，用于匹配markdown中的图片链接
  /
  private static final String IMG_REGEX = "!\[.?\]\((http.*?\.(png|jpe?g|gif|svg))\)";
/**
- 超时时间，单位为毫秒
  */
  private static final int TIMEOUT = 5000;
/**
- 线程数
  */
  private static final int THREAD_NUM = 10;
/**
- 线程池
  */
  static ExecutorService executorService = Executors.newFixedThreadPool(THREAD_NUM);
public static void main(String[] args) {
// Markdown文件所在目录
String dirPath = "D:\ideaProject\kjs-projects\.docs\Java";
// 图片保存目录
String img = "D:\ideaProject\kjs-projects\.docs";
File dir = new File(dirPath);
List markdownFiles = new ArrayList<>();
scanMarkdownFiles(dir, markdownFiles, "");
for (File file : markdownFiles) {
processFile(file, img);
}
executorService.shutdown();
}

/**
- 递归扫描Markdown文件
  */
  public static void scanMarkdownFiles(File dir, List markdownFiles, String... ignoreDirs) {
  if (dir.isDirectory()) {
  File[] files = dir.listFiles();
  for (File file : Objects.requireNonNull(files)) {
  boolean ignore = false;
  for (String ignoreDir : ignoreDirs) {
  if (file.getName().equals(ignoreDir) || file.getAbsolutePath().endsWith(File.separator + ignoreDir)) {
  ignore = true;
  break;
  }
  }
  if (!ignore) {
  if (file.isDirectory()) {
  scanMarkdownFiles(file, markdownFiles, ignoreDirs);
  } else if (file.getName().toLowerCase().endsWith(".md")) {
  markdownFiles.add(file);
  }
  }
  }
  }
  }
/**
- 处理单个Markdown文件
  */
  private static void processFile(File file, String imgDirPath) {
  SortedMap<Integer, String> imageUrls = getImageUrls(file);
  if (imageUrls.size() == 0) {
  System.out.println("未在 " + file.toPath() + " 中找到图片链接");
  return;
  }
  
  // 创建 img/java/{filename} 目录
  String folderName = file.getName().replace(".md", "");
  String newImgDirPath = imgDirPath + File.separator + folderName;
  File imgDir = new File(newImgDirPath);
  if (!imgDir.exists()) {
  imgDir.mkdirs();
  }
  
  for (Map.Entry<Integer, String> entry : imageUrls.entrySet()) {
  Integer index = entry.getKey();
  String imageUrl = entry.getValue();
  // 获取图片扩展名
  String extension = imageUrl.substring(imageUrl.lastIndexOf("."));
  String imageName = index + extension;
  String imgPath = imgDir.getAbsolutePath() + File.separator + imageName;
```
 try {
     executorService.submit(() -> downloadImage(imageUrl, imgPath));
     String newImageUrl = "./../" + IMG_FOLDER + "/" + folderName + "/" + imageName;
     replaceStringInFile(file, imageUrl, newImageUrl);
 } catch (Exception e) {
     System.out.println("在 " + file.getName() + " 中下载图片 " + imageUrl + " 失败: " + e.getMessage());
 }
```
  }
  
  System.out.println("在 " + file.getName() + " 中处理了 " + imageUrls.size() + " 张图片");
  }
/**
- 下载图片
  */
  private static void downloadImage(String imageUrl, String imagePath) {
  try {
  URL url = new URL(imageUrl);
  HttpURLConnection connection = (HttpURLConnection) url.openConnection();
  connection.setConnectTimeout(TIMEOUT);
  connection.setReadTimeout(TIMEOUT);
  connection.setRequestMethod("GET");
  try (InputStream inputStream = connection.getInputStream();
  OutputStream outputStream = new FileOutputStream(createPath(imagePath).toFile())) {
  byte[] buffer = new byte[4096];
  int bytesRead;
  while ((bytesRead = inputStream.read(buffer)) != -1) {
  outputStream.write(buffer, 0, bytesRead);
  }
  }
  } catch (Exception e) {
  e.printStackTrace();
  }
  }
/**
- 创建文件路径
  */
  private static Path createPath(String imagePath) throws Exception {
  Path path = Path.of(imagePath);
  if (!Files.exists(path.getParent())) {
  Files.createDirectories(path.getParent());
  }
  return path;
  }
/**
- 获取Markdown文件中的图片链接
  */
  private static SortedMap<Integer, String> getImageUrls(File file) {
  SortedMap<Integer, String> imageUrls = new TreeMap<>();
  try {
  Stream lines = Files.lines(file.toPath());
  AtomicInteger index = new AtomicInteger(0);
  imageUrls = lines.flatMap(line -> {
  Matcher matcher = Pattern.compile(IMG_REGEX).matcher(line);
  List urls = new ArrayList<>();
  while (matcher.find()) {
  urls.add(matcher.group(1));
  }
  return urls.stream().map(url -> new AbstractMap.SimpleEntry<>(index.getAndIncrement(), url));
  }).collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue, (u, v) -> u, TreeMap::new));
  } catch (IOException e) {
  System.out.println("读取 " + file.getName() + " 失败: " + e.getMessage());
  }
  return imageUrls;
  }
/**
- 替换文件中的字符串
  */
  private static void replaceStringInFile(File file, String oldStr, String newStr) {
  try {
  // 指定文件编码
  String encoding = "UTF-8";
  Path path = file.toPath();
  // 读取文件内容
  List lines = Files.readAllLines(path, Charset.forName(encoding));
  ListIterator iterator = lines.listIterator();
  while (iterator.hasNext()) {
  String line = iterator.next();
  if (line.contains(oldStr)) {
  // 替换指定字符串
  iterator.set(line.replace(oldStr, newStr));
  }
  }
  // 将新内容写入文件
  Files.write(path, lines, Charset.forName(encoding));
  } catch (IOException e) {
  e.printStackTrace();
  }
  }
  }`

markdown