1

blog-springboot/.gitignore

@@ -0,0 +1 @@
+target

blog-springboot/src/main/java/com/minzheng/blog/constant/HexoConst.java

@@ -0,0 +1,48 @@
+package com.minzheng.blog.constant;
+
+/**
+ * Hexo文章常量
+ * <p>
+ * Hexo结构
+ * ---
+ * title: xxx
+ * date: yyyy-MM-dd HH:mm:ss
+ * categories: [a, b, c]
+ * tags: [d, e, f]
+ * ---
+ * 正文
+ *
+ * @author linweiyuan
+ * @date 2022/07/28
+ */
+public class HexoConst {
+    /**
+     * 标题
+     */
+    public static final String TITLE_PREFIX = "title:";
+
+    /**
+     * 日期
+     */
+    public static final String DATE_PREFIX = "date:";
+
+    /**
+     * 分类
+     */
+    public static final String CATEGORIES_PREFIX = "categories:";
+
+    /**
+     * 标签
+     */
+    public static final String TAGS_PREFIX = "tags:";
+
+    /**
+     * 分隔符
+     */
+    public static final String DELIMITER = "---";
+
+    /**
+     * 换行符
+     */
+    public static final String NEW_LINE = "\n";
+}

blog-springboot/src/main/java/com/minzheng/blog/controller/ArticleController.java

@@ -5,18 +5,18 @@
 import com.minzheng.blog.dto.*;
 import com.minzheng.blog.enums.FilePathEnum;
 import com.minzheng.blog.service.ArticleService;
+import com.minzheng.blog.strategy.context.ArticleImportStrategyContext;
 import com.minzheng.blog.strategy.context.UploadStrategyContext;
 import com.minzheng.blog.vo.*;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiImplicitParam;
 import io.swagger.annotations.ApiOperation;
 import org.springframework.beans.factory.annotation.Autowired;
-
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
 import javax.validation.Valid;
-import java.util.*;
+import java.util.List;
 
 import static com.minzheng.blog.constant.OptTypeConst.*;
 
@@ -33,6 +33,8 @@ public class ArticleController {
     private ArticleService articleService;
     @Autowired
     private UploadStrategyContext uploadStrategyContext;
+    @Autowired
+    private ArticleImportStrategyContext articleImportStrategyContext;
 
     /**
      * 查看文章归档
@@ -218,14 +220,13 @@ public Result<List<String>> exportArticles(@RequestBody List<Integer> articleIdL
      * 导入文章
      *
      * @param file 文件
+     * @param type 文章类型
      * @return {@link Result<>}
      */
     @ApiOperation(value = "导入文章")
     @PostMapping("/admin/articles/import")
-    public Result<?> importArticles(MultipartFile file) {
-        articleService.importArticles(file);
+    public Result<?> importArticles(MultipartFile file, @RequestParam(required = false) String type) {
+        articleImportStrategyContext.importArticles(file, type);
         return Result.ok();
     }
-
-}
-
+}

blog-springboot/src/main/java/com/minzheng/blog/enums/MarkdownTypeEnum.java

@@ -0,0 +1,45 @@
+package com.minzheng.blog.enums;
+
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+
+/**
+ * Markdown文章类型枚举
+ *
+ * @author linweiyuan
+ * @date 2022/07/28
+ */
+@Getter
+@AllArgsConstructor
+public enum MarkdownTypeEnum {
+    /**
+     * 普通文章
+     */
+    NORMAL("", "normalArticleImportStrategyImpl"),
+    /**
+     * Hexo文章
+     */
+    HEXO("hexo", "hexoArticleImportStrategyImpl");
+
+    /**
+     * 类型
+     */
+    private final String type;
+
+    /**
+     * 策略
+     */
+    private final String strategy;
+
+    public static String getMarkdownType(String name) {
+        if (name == null) {
+            return NORMAL.getStrategy();
+        }
+        for (MarkdownTypeEnum value : MarkdownTypeEnum.values()) {
+            if (value.getType().equalsIgnoreCase(name)) {
+                return value.getStrategy();
+            }
+        }
+        return null;
+    }
+}

blog-springboot/src/main/java/com/minzheng/blog/service/ArticleService.java

@@ -1,10 +1,9 @@
 package com.minzheng.blog.service;
 
+import com.baomidou.mybatisplus.extension.service.IService;
 import com.minzheng.blog.dto.*;
 import com.minzheng.blog.entity.Article;
-import com.baomidou.mybatisplus.extension.service.IService;
 import com.minzheng.blog.vo.*;
-import org.springframework.web.multipart.MultipartFile;
 
 import java.util.List;
 
@@ -112,12 +111,4 @@ public interface ArticleService extends IService<Article> {
      * @return {@link List}<{@link String}> 文件地址
      */
     List<String> exportArticles(List<Integer> articleIdList);
-
-    /**
-     * 导入文章
-     *
-     * @param file 文件
-     */
-    void importArticles(MultipartFile file);
-
 }

blog-springboot/src/main/java/com/minzheng/blog/service/impl/ArticleServiceImpl.java

@@ -5,7 +5,11 @@
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
-import com.minzheng.blog.dao.*;
+import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
+import com.minzheng.blog.dao.ArticleDao;
+import com.minzheng.blog.dao.ArticleTagDao;
+import com.minzheng.blog.dao.CategoryDao;
+import com.minzheng.blog.dao.TagDao;
 import com.minzheng.blog.dto.*;
 import com.minzheng.blog.entity.Article;
 import com.minzheng.blog.entity.ArticleTag;
@@ -15,7 +19,6 @@
 import com.minzheng.blog.enums.FilePathEnum;
 import com.minzheng.blog.exception.BizException;
 import com.minzheng.blog.service.ArticleService;
-import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.minzheng.blog.service.ArticleTagService;
 import com.minzheng.blog.service.RedisService;
 import com.minzheng.blog.service.TagService;
@@ -30,10 +33,9 @@
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
-import org.springframework.web.multipart.MultipartFile;
 
 import javax.servlet.http.HttpSession;
-import java.io.*;
+import java.io.ByteArrayInputStream;
 import java.util.*;
 import java.util.concurrent.CompletableFuture;
 import java.util.stream.Collectors;
@@ -276,28 +278,6 @@ public List<String> exportArticles(List<Integer> articleIdList) {
         return urlList;
     }
 
-    @Override
-    public void importArticles(MultipartFile file) {
-        // 获取文件名作为文章标题
-        String articleTitle = Objects.requireNonNull(file.getOriginalFilename()).split("\\.")[0];
-        // 获取文章内容
-        StringBuilder articleContent = new StringBuilder();
-        try (BufferedReader reader = new BufferedReader(new InputStreamReader(file.getInputStream()))) {
-            while (reader.ready()) {
-                articleContent.append((char) reader.read());
-            }
-        } catch (IOException e) {
-            log.error(StrUtil.format("导入文章失败,堆栈:{}", ExceptionUtil.stacktraceToString(e)));
-            throw new BizException("导入文章失败");
-        }
-        articleDao.insert(Article.builder()
-                .userId(UserUtils.getLoginUser().getUserInfoId())
-                .articleTitle(articleTitle)
-                .articleContent(articleContent.toString())
-                .status(DRAFT.getStatus())
-                .build());
-    }
-
     @Override
     public List<ArticleSearchDTO> listArticlesBySearch(ConditionVO condition) {
         return searchStrategyContext.executeSearchStrategy(condition.getKeywords());

blog-springboot/src/main/java/com/minzheng/blog/strategy/ArticleImportStrategy.java

@@ -0,0 +1,19 @@
+package com.minzheng.blog.strategy;
+
+import org.springframework.web.multipart.MultipartFile;
+
+/**
+ * 文章导入策略
+ *
+ * @author linweiyuan
+ * @date 2022/07/28
+ */
+public interface ArticleImportStrategy {
+
+    /**
+     * 导入文章
+     *
+     * @param file 文件
+     */
+    void importArticles(MultipartFile file);
+}

blog-springboot/src/main/java/com/minzheng/blog/strategy/context/ArticleImportStrategyContext.java

@@ -0,0 +1,25 @@
+package com.minzheng.blog.strategy.context;
+
+import com.minzheng.blog.enums.MarkdownTypeEnum;
+import com.minzheng.blog.strategy.ArticleImportStrategy;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.util.Map;
+
+/**
+ * 文章导入策略上下文
+ *
+ * @author linweiyuan
+ * @date 2022/07/28
+ */
+@Service
+public class ArticleImportStrategyContext {
+    @Autowired
+    private Map<String, ArticleImportStrategy> articleImportStrategyMap;
+
+    public void importArticles(MultipartFile file, String type) {
+        articleImportStrategyMap.get(MarkdownTypeEnum.getMarkdownType(type)).importArticles(file);
+    }
+}

blog-springboot/src/main/java/com/minzheng/blog/strategy/impl/HexoArticleImportStrategyImpl.java

@@ -0,0 +1,95 @@
+package com.minzheng.blog.strategy.impl;
+
+import cn.hutool.core.exceptions.ExceptionUtil;
+import cn.hutool.core.util.StrUtil;
+import cn.hutool.json.JSONArray;
+import com.minzheng.blog.exception.BizException;
+import com.minzheng.blog.service.ArticleService;
+import com.minzheng.blog.strategy.ArticleImportStrategy;
+import com.minzheng.blog.vo.ArticleVO;
+import lombok.Data;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+import org.springframework.web.multipart.MultipartFile;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.nio.charset.StandardCharsets;
+import java.time.LocalDateTime;
+import java.time.format.DateTimeFormatter;
+import java.util.List;
+import java.util.concurrent.atomic.AtomicInteger;
+
+import static com.minzheng.blog.constant.HexoConst.*;
+import static com.minzheng.blog.enums.ArticleStatusEnum.PUBLIC;
+
+/**
+ * Hexo文章导入策略
+ *
+ * @author linweiyuan
+ * @date 2022/07/28
+ */
+@Slf4j
+@Service("hexoArticleImportStrategyImpl")
+public class HexoArticleImportStrategyImpl implements ArticleImportStrategy {
+    @Autowired
+    private ArticleService articleService;
+
+    private final DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
+
+    @Override
+    public void importArticles(MultipartFile file) {
+        try {
+            HexoArticleVO hexoArticleVO = new HexoArticleVO();
+            hexoArticleVO.setType(1); // 原创
+            hexoArticleVO.setStatus(PUBLIC.getStatus()); // 公开 （DRAFT不保存分类）
+
+            AtomicInteger hexoDelimiterCount = new AtomicInteger();
+            StringBuilder articleContent = new StringBuilder();
+
+            BufferedReader reader = new BufferedReader(new InputStreamReader(file.getInputStream(), StandardCharsets.UTF_8));
+            reader.lines().forEach(line -> {
+                if (hexoDelimiterCount.get() == 2) {
+                    // 分隔符结束就是正文
+                    articleContent.append(line).append(NEW_LINE);
+                } else {
+                    if (line.equals(DELIMITER)) {
+                        hexoDelimiterCount.getAndIncrement();
+                    }
+                    if (hexoDelimiterCount.get() == 1) {
+                        if (line.startsWith(TITLE_PREFIX)) {
+                            hexoArticleVO.setArticleTitle(line.replace(TITLE_PREFIX, "").trim());
+                        } else if (line.startsWith(DATE_PREFIX)) {
+                            hexoArticleVO.setCreateTime(LocalDateTime.parse(line.replace(DATE_PREFIX, "").trim(), formatter));
+                        } else if (line.startsWith(CATEGORIES_PREFIX)) {
+                            // 不支持多级分类
+                            hexoArticleVO.setCategoryName(new JSONArray(line.replace(CATEGORIES_PREFIX, "").trim()).getStr(0));
+                        } else if (line.startsWith(TAGS_PREFIX)) {
+                            hexoArticleVO.setTagNameList(new JSONArray(line.replace(TAGS_PREFIX, "").trim()).toList(String.class));
+                        }
+                    }
+                }
+            });
+            hexoArticleVO.setArticleContent(articleContent.toString());
+
+            articleService.saveOrUpdateArticle(hexoArticleVO);
+        } catch (IOException e) {
+            log.error(StrUtil.format("导入Hexo文章失败, 堆栈:{}", ExceptionUtil.stacktraceToString(e)));
+            throw new BizException("导入Hexo文章失败");
+        }
+    }
+
+    @SuppressWarnings("Lombok")
+    @Data
+    static class HexoArticleVO extends ArticleVO {
+        private String articleTitle;
+        private String categoryName;
+        private List<String> tagNameList;
+        private String articleContent;
+        private Integer type;
+        private Integer status;
+        private LocalDateTime createTime;
+    }
+}