fix(export-pdf): fix invalid links of tag and category (#73)

laggage · web-flow · commit 838779bfa88d · 2022-10-15T20:31:19.000+08:00
diff --git a/.vscode/launch.json b/.vscode/launch.json
@@ -13,7 +13,8 @@
             "outFiles": ["${workspaceFolder}/dist/**/*.js"],
             "preLaunchTask": "${defaultBuildTask}",
             "env": {
-                "NODE_ENV": "Development"
+                "NODE_ENV": "Development",
+                "NODE_TLS_REJECT_UNAUTHORIZED": "0"
             }
         },
         {
diff --git a/src/commands/pdf/export-pdf.command.ts b/src/commands/pdf/export-pdf.command.ts
@@ -10,6 +10,8 @@ import { extensionViews } from '../../tree-view-providers/tree-view-registration
 import { postPdfTemplateBuilder } from './post-pdf-template-builder';
 import { chromiumPathProvider } from '../../utils/chromium-path-provider';
 import { Settings } from '../../services/settings.service';
+import { accountService } from '../../services/account.service';
+import { AlertService } from '../../services/alert.service';
 
 const launchBrowser = async (
     chromiumPath: string
@@ -42,7 +44,8 @@ const exportOne = async (
     post: Post,
     page: puppeteer.Page,
     targetFileUri: Uri,
-    progress: Progress<{ message: string; increment: number }>
+    progress: Progress<{ message: string; increment: number }>,
+    blogApp: string
 ) => {
     let message = `[${idx + 1}/${total}]正在导出 - ${post.title}`;
     const report = (increment: number) => {
@@ -52,7 +55,7 @@ const exportOne = async (
         });
     };
     report(10);
-    let html = await postPdfTemplateBuilder.build(post);
+    let html = await postPdfTemplateBuilder.build(post, blogApp);
     report(15);
     // Wait for code block highlight finished
     await Promise.all([
@@ -151,23 +154,20 @@ const handlePostInput = (post: Post): Promise<Post[]> => {
 
 const handleUriInput = async (uri: Uri): Promise<Post[]> => {
     const posts: Post[] = [];
-    const postId = PostFileMapManager.getPostId(uri.fsPath);
-    let inputPost: Post | undefined;
-    if (postId && postId > 0) {
-        inputPost = (await postService.fetchPostEditDto(postId))?.post;
-    } else {
-        const { fsPath } = uri;
-        inputPost = Object.assign((await postService.fetchPostEditDto(-1))?.post, {
+    const { fsPath } = uri;
+    const postId = PostFileMapManager.getPostId(fsPath);
+    const { post: inputPost } = (await postService.fetchPostEditDto(postId && postId > 0 ? postId : -1)) ?? {};
+
+    if (!inputPost) {
+        return [];
+    } else if (inputPost.id <= 0) {
+        Object.assign(inputPost, {
             id: -1,
             title: path.basename(fsPath, path.extname(fsPath)),
             postBody: new TextDecoder().decode(await workspace.fs.readFile(uri)),
         } as Post);
     }
 
-    if (!inputPost) {
-        return [];
-    }
-
     posts.push(inputPost);
 
     return posts;
@@ -187,6 +187,13 @@ const exportPostToPdf = async (input: Post | Uri): Promise<void> => {
     if (!chromiumPath) {
         return;
     }
+    const {
+        curUser: { blogApp },
+    } = accountService;
+
+    if (!blogApp) {
+        return AlertService.warning('无法获取到博客地址, 请检查登录状态');
+    }
 
     reportErrors(
         await window.withProgress<string[] | undefined>(
@@ -216,7 +223,7 @@ const exportPostToPdf = async (input: Post | Uri): Promise<void> => {
                 const { length: total } = selectedPosts;
                 for (const post of selectedPosts) {
                     try {
-                        await exportOne(idx++, total, post, page, dir!, progress);
+                        await exportOne(idx++, total, post, page, dir!, progress, blogApp);
                     } catch (err) {
                         errors.push(`导出"${post.title}失败", ${err}`);
                     }
diff --git a/src/commands/pdf/post-pdf-template-builder.ts b/src/commands/pdf/post-pdf-template-builder.ts
@@ -5,12 +5,14 @@ import { markdownItFactory } from '@cnblogs-gitlab/markdown-it-presets';
 import { blogSettingsService } from '../../services/blog-settings.service';
 import { accountService } from '../../services/account.service';
 import { postCategoryService } from '../../services/post-category.service';
+import { PostCategory } from '../../models/post-category';
 
 export namespace postPdfTemplateBuilder {
     export const highlightedMessage = 'markdown-highlight-finished';
 
-    export const build = async (post: Post): Promise<string> => {
+    export const build = async (post: Post, blogApp: string): Promise<string> => {
         let { postBody, isMarkdown, id: postId } = post;
+
         const localFilePath = PostFileMapManager.getFilePath(postId);
         postBody = localFilePath ? fs.readFileSync(localFilePath).toString('utf-8') : postBody;
 
@@ -26,7 +28,7 @@ export namespace postPdfTemplateBuilder {
         const buildTagHtml = (): Promise<string> => {
             let html =
                 post.tags && post.tags.length > 0
-                    ? post.tags.map(t => `<a href="https://www.cnblogs.com/laggage/tag/linux/">${t}</a>`).join(', ')
+                    ? post.tags.map(t => `<a href="https://www.cnblogs.com/${blogApp}/tag/${t}/">${t}</a>`).join(', ')
                     : '';
             html = html ? `<div id="EntryTag">标签: ${html}</div>` : '';
             return Promise.resolve(html);
@@ -35,13 +37,15 @@ export namespace postPdfTemplateBuilder {
         const buildCategoryHtml = async (): Promise<string> => {
             let categories = await postCategoryService.fetchCategories();
             const postCategories =
-                post.categoryIds?.map(categoryId => categories.find(x => x.categoryId === categoryId)) ?? [];
+                post.categoryIds
+                    ?.map(categoryId => categories.find(x => x.categoryId === categoryId))
+                    .filter((x): x is PostCategory => x != null) ?? [];
             let html =
                 postCategories.length > 0
                     ? postCategories
                           .map(
                               c =>
-                                  `<a href="https://www.cnblogs.com/laggage/category/1565066.html" target="_blank">${c?.title}</a>`
+                                  `<a href="https://www.cnblogs.com/${blogApp}/category/${c.categoryId}.html" target="_blank">${c?.title}</a>`
                           )
                           .join(', ')
                     : '';
@@ -51,7 +55,7 @@ export namespace postPdfTemplateBuilder {
 
         const tagHtml = await buildTagHtml();
         const categoryHtml = await buildCategoryHtml();
-        const { codeHighlightEngine, codeHighlightTheme, enableCodeLineNumber, blogId, application } =
+        const { codeHighlightEngine, codeHighlightTheme, enableCodeLineNumber, blogId } =
             await blogSettingsService.getBlogSettings();
         const { userId } = accountService.curUser;
         return `<html>
@@ -75,7 +79,7 @@ export namespace postPdfTemplateBuilder {
             <link rel="stylesheet" href="https://www.cnblogs.com/css/blog-common.min.css">
             <script>
             var currentBlogId = ${blogId};
-            var currentBlogApp = '${application}';
+            var currentBlogApp = '${blogApp}';
             var cb_enable_mathjax = true;
             var isLogined = true;
             var isBlogOwner = true;
diff --git a/src/models/user-settings.ts b/src/models/user-settings.ts
@@ -1,12 +1,24 @@
+import { trim } from 'lodash';
+
 export class UserAuthorizationInfo {
     constructor(public idToken: string, public accessToken: string, expiresIn: number, public tokenType: string) {}
 }
 
 export class UserInfo {
+    private _blogApp?: string | null;
+
     get userId() {
         return this.sub;
     }
 
+    get blogApp(): string | null {
+        if (this._blogApp == null) {
+            this._blogApp = this.parseBlogApp();
+        }
+
+        return this._blogApp;
+    }
+
     /**
      * Creates an instance of UserInfo.
      * @param {UserAuthorizationInfo} [authorizationInfo]
@@ -26,4 +38,12 @@ export class UserInfo {
         public sub: string = '',
         public accountId: number = -1
     ) {}
+
+    private parseBlogApp(): string | null {
+        return (
+            trim(this.website ?? '', '/')
+                .split('/')
+                .pop() ?? null
+        );
+    }
 }
diff --git a/src/services/account.service.ts b/src/services/account.service.ts
@@ -38,7 +38,10 @@ export class AccountService extends vscode.Disposable {
     }
 
     get curUser(): UserInfo {
-        return this._curUser ? this._curUser : globalState.storage.get('user') ?? new UserInfo();
+        return (
+            this._curUser ||
+            (this._curUser = Object.assign(new UserInfo(), globalState.storage.get('user') ?? new UserInfo()))
+        );
     }
 
     protected constructor() {
@@ -80,6 +83,7 @@ export class AccountService extends vscode.Disposable {
         const token = this.curUser?.authorizationInfo?.accessToken;
 
         await globalState.storage.update('user', {});
+        this._curUser = undefined;
         await this.setIsAuthorized(false);
 
         if (token) {

Original file line number	Diff line number	Diff line change
`@@ -13,7 +13,8 @@`
`13`	`13`	`"outFiles": ["${workspaceFolder}/dist/*/.js"],`
`14`	`14`	`"preLaunchTask": "${defaultBuildTask}",`
`15`	`15`	`"env": {`
`16`		`- "NODE_ENV": "Development"`
	`16`	`+ "NODE_ENV": "Development",`
	`17`	`+ "NODE_TLS_REJECT_UNAUTHORIZED": "0"`
`17`	`18`	`}`
`18`	`19`	`},`
`19`	`20`	`{`