88 lines
2.7 KiB
TypeScript
88 lines
2.7 KiB
TypeScript
import fs from "fs"
|
|
import path from "path"
|
|
import { Repository } from "@napi-rs/simple-git"
|
|
import { QuartzTransformerPlugin } from "../types"
|
|
import chalk from "chalk"
|
|
|
|
export interface Options {
|
|
priority: ("frontmatter" | "git" | "filesystem")[]
|
|
}
|
|
|
|
const defaultOptions: Options = {
|
|
priority: ["frontmatter", "git", "filesystem"],
|
|
}
|
|
|
|
export const CreatedModifiedDate: QuartzTransformerPlugin<Partial<Options> | undefined> = (
|
|
userOpts,
|
|
) => {
|
|
const opts = { ...defaultOptions, ...userOpts }
|
|
return {
|
|
name: "CreatedModifiedDate",
|
|
markdownPlugins() {
|
|
return [
|
|
() => {
|
|
let repo: Repository | undefined = undefined
|
|
return async (_tree, file) => {
|
|
let created: Date | undefined = undefined
|
|
let modified: Date | undefined = undefined
|
|
let published: Date | undefined = undefined
|
|
|
|
const fp = file.data.filePath!
|
|
const fullFp = path.posix.join(file.cwd, fp)
|
|
for (const source of opts.priority) {
|
|
if (source === "filesystem") {
|
|
const st = await fs.promises.stat(fullFp)
|
|
created ||= new Date(st.birthtimeMs)
|
|
modified ||= new Date(st.mtimeMs)
|
|
} else if (source === "frontmatter" && file.data.frontmatter) {
|
|
created ||= file.data.frontmatter.created
|
|
modified ||= file.data.frontmatter.modified
|
|
published ||= file.data.frontmatter.published
|
|
} else if (source === "git") {
|
|
if (!repo) {
|
|
// Get a reference to the main git repo.
|
|
// It's either the same as the workdir,
|
|
// or 1+ level higher in case of a submodule/subtree setup
|
|
repo = Repository.discover(file.cwd)
|
|
}
|
|
|
|
try {
|
|
modified ||= new Date(
|
|
await repo.getFileLatestModifiedDateAsync(file.data.filePath!),
|
|
)
|
|
} catch {
|
|
console.log(
|
|
chalk.yellow(
|
|
`\nWarning: ${file.data
|
|
.filePath!} isn't yet tracked by git, last modification date is not available for this file`,
|
|
),
|
|
)
|
|
}
|
|
}
|
|
}
|
|
|
|
created ||= new Date()
|
|
modified ||= new Date()
|
|
published ||= new Date()
|
|
file.data.dates = {
|
|
created,
|
|
modified,
|
|
published,
|
|
}
|
|
}
|
|
},
|
|
]
|
|
},
|
|
}
|
|
}
|
|
|
|
declare module "vfile" {
|
|
interface DataMap {
|
|
dates: {
|
|
created: Date
|
|
modified: Date
|
|
published: Date
|
|
}
|
|
}
|
|
}
|