Jacky Zhao
2023-05-31 21c007e2fcf73fe4ef04dd07db7116afed46047a
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
import { PluggableList } from "unified"
import fs from "fs"
import path from 'path'
import { Repository } from "@napi-rs/simple-git"
import { QuartzTransformerPlugin } from "../types"
 
export interface Options {
  priority: ('frontmatter' | 'git' | 'filesystem')[],
}
 
const defaultOptions: Options = {
  priority: ['frontmatter', 'git', 'filesystem']
}
 
export class CreatedModifiedDate extends QuartzTransformerPlugin {
  name = "CreatedModifiedDate"
  opts: Options
 
  constructor(opts?: Options) {
    super()
    this.opts = {
      ...defaultOptions,
      ...opts,
    }
  }
 
  markdownPlugins(): PluggableList {
    return [
      () => {
        let repo: Repository | undefined = undefined
        return async (_tree, file) => {
          let created: undefined | Date = undefined
          let modified: undefined | Date = undefined
          let published: undefined | Date = undefined
 
          const fp = path.join(file.cwd, file.data.filePath as string)
          for (const source of this.opts.priority) {
            if (source === "filesystem") {
              const st = await fs.promises.stat(fp)
              created ||= new Date(st.birthtimeMs)
              modified ||= new Date(st.mtimeMs)
            } else if (source === "frontmatter" && file.data.frontmatter) {
              created ||= file.data.frontmatter.date
              modified ||= file.data.frontmatter.lastmod
              modified ||= file.data.frontmatter["last-modified"]
              published ||= file.data.frontmatter.publishDate
            } else if (source === "git") {
              if (!repo) {
                repo = new Repository(file.cwd)
              }
 
              modified ||= new Date(await repo.getFileLatestModifiedDateAsync(file.data.filePath!))
            }
          }
 
          file.data.dates = {
            created: created ?? new Date(),
            modified: modified ?? new Date(),
            published: published ?? new Date()
          }
        }
      }
    ]
  }
 
  htmlPlugins(): PluggableList {
    return []
  }
}
 
declare module 'vfile' {
  interface DataMap {
    dates: {
      created: Date
      modified: Date
      published: Date
    }
  }
}