1
/* This is a script to create a new post markdown file with front-matter */
2

3
import fs from "fs"
4
import path from "path"
5
import { Command } from "commander"
6
import inquirer from "inquirer"
7

8
function getDate() {
9
  const today = new Date()
10
  const year = today.getFullYear()
11
  const month = String(today.getMonth() + 1).padStart(2, "0")
12
  const day = String(today.getDate()).padStart(2, "0")
13

14
  return `${year}-${month}-${day}`
15
}
16

17
function createPostContent(title, description, tags, category, lang, fileName) {
18
  const publishedDate = getDate()
19

20
  // Parse tags from string like "#tag1 #tag2" to array
21
  let tagsArray = []
22
  if (tags) {
23
    tagsArray = tags.split(/\s+/).filter(tag => tag.trim() !== '').map(tag => {
24
      // Remove # prefix if present
25
      return tag.replace(/^#/, '').trim()
26
    })
27
  }
28

29
  return `---
30
title: ${title || fileName}
31
published: ${publishedDate}
32
description: '${description || ""}'
33
image: ''
34
tags: [${tagsArray.map(tag => `"${tag}"`).join(", ")}]
35
category: '${category || ""}'
36
draft: false
37
lang: '${lang || ""}'
38
---
39
`
40
}
41

42
async function createPostInteractive() {
43
  console.log("Create New Post")
44

45
  const questions = [
46
    {
47
      type: "input",
48
      name: "fileName",
49
      message: "File name:",
50
      default: "Leave it blank to cancel",
51
      validate: (input) => {
52
        if (input == "Leave it blank to cancel") {
53
          process.exit(0)
54
        }
55
        return true
56
      }
57
    },
58
    {
59
      type: "input",
60
      name: "title",
61
      message: "Post title:",
62
      default: (answers) => answers.fileName
63
    },
64
    {
65
      type: "input",
66
      name: "description",
67
      message: "Post description (optional):",
68
      default: ""
69
    },
70
    {
71
      type: "input",
72
      name: "tags",
73
      message: "Tags (Format: #tag_1 #tag_2) (optional):",
74
      default: ""
75
    },
76
    {
77
      type: "input",
78
      name: "category",
79
      message: "Category (optional):",
80
      default: ""
81
    },
82
    {
83
      type: "list",
84
      name: "lang",
85
      message: "Language:",
86
      choices: ["zh_CN", "en", ""],
87
      default: "zh_CN"
88
    }
89
  ]
90

91
  const answers = await inquirer.prompt(questions)
92
  return createAndSavePost(answers.fileName, answers.title, answers.description, answers.tags, answers.category, answers.lang)
93
}
94

95
function createAndSavePost(fileName, title, description, tags, category, lang) {
96
  // Add .md extension if not present
97
  const fileExtensionRegex = /\.(md|mdx)$/i
98
  if (!fileExtensionRegex.test(fileName)) {
99
    fileName += ".md"
100
  }
101

102
  const targetDir = "./src/content/posts/"
103
  const fullPath = path.join(targetDir, fileName)
104

105
  if (fs.existsSync(fullPath)) {
106
    console.error(`Error: file ${fullPath} already exists!`)
107
    process.exit(1)
108
  }
109

110
  // recursive mode creates multi-level directories
111
  const dirPath = path.dirname(fullPath)
112
  if (!fs.existsSync(dirPath)) {
113
    fs.mkdirSync(dirPath, { recursive: true })
114
  }
115

116
  const content = createPostContent(title, description, tags, category, lang, fileName)
117
  fs.writeFileSync(path.join(targetDir, fileName), content)
118

119
  console.log(`The file ${fullPath} is successful created!`)
120
}
121

122
async function main() {
123
  const program = new Command()
124

125
  program
126
    .name("new-post")
127
    .description("Create new post from interactive TUI or arguments")
128
    .version("1.0.0")
129

130
  program
131
    .argument("[fileName]", "filename (without the extension)")
132
    .option("-t, --title <title>", "Post title")
133
    .option("-d, --description <description>", "Post description")
134
    .option("--tags <tags>", "Tags (format: '#tag_1 #tag_2')")
135
    .option("-c, --category <category>", "Category")
136
    .option("--lang <lang>", "Language (zh_CN or en)", "zh_CN")
137
    .action(async (fileName, options) => {
138
      if (fileName) {
139
        // Use command line arguments
140
        createAndSavePost(fileName, options.title, options.description, options.tags, options.category, options.lang)
141
      } else {
142
        // No arguments provided, use interactive mode
143
        await createPostInteractive()
144
      }
145
    })
146

147
  try {
148
    await program.parseAsync(process.argv)
149
  } catch (error) {
150
    console.error("ERROR:", error.message)
151
    process.exit(1)
152
  }
153
}
154

155
main()