forked from WnP/vimwiki_markdown
-
Notifications
You must be signed in to change notification settings - Fork 0
/
vimwiki_markdown.py
173 lines (144 loc) · 5.29 KB
/
vimwiki_markdown.py
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import datetime
import json
import os
import shutil
import subprocess
import sys
import markdown
default_template = """<!DOCTYPE html>
<html>
<head>
<meta charset="UTF-8" />
<meta name="date" content="%date%" scheme="YYYY-MM-DD">
<meta name="viewport" content="width=device-width" />
<title>%title%</title>
<link rel="stylesheet" href="%root_path%style.css" type="text/css"
media="screen" title="no title" charset="utf-8">
<link rel="stylesheet" href="%root_path%pygmentize.css" type="text/css"
media="screen" title="no title" charset="utf-8">
</head>
<body>
%content%
</body>
</html>
"""
default_extension = ["fenced_code", "tables", "codehilite"]
vim = shutil.which("vim") and "vim" or (shutil.which("nvim") and "nvim")
if vim:
# Retrieve auto index vimwiki option
with subprocess.Popen(
[vim, "-c", "echo g:vimwiki_dir_link", "-c", ":q", "--headless"],
stdout=subprocess.PIPE,
stderr=subprocess.PIPE,
) as proc:
auto_index = proc.stderr.read() == b"index"
else:
auto_index = False
class LinkInlineProcessor(markdown.inlinepatterns.LinkInlineProcessor):
"""Fix wiki links"""
def getLink(self, *args, **kwargs):
href, title, index, handled = super().getLink(*args, **kwargs)
if not href.startswith("http") and not href.endswith(".html"):
if auto_index and href.endswith("/"):
href += "index.html"
elif not href.endswith("/"):
href += ".html"
return href, title, index, handled
def get(l, index, default):
return l[index] if index < len(l) else default
def main():
FORCE = sys.argv[1] # noqa - not supported
SYNTAX = sys.argv[2]
EXTENSION = sys.argv[3] # noqa - not supported
OUTPUT_DIR = sys.argv[4]
INPUT_FILE = sys.argv[5]
CSS_FILE = sys.argv[6] # noqa - not supported
TEMPLATE_PATH = get(sys.argv, 7, os.getenv("VIMWIKI_TEMPLATE_PATH", ""))
TEMPLATE_DEFAULT = get(
sys.argv, 8, os.getenv("VIMWIKI_TEMPLATE_DEFAULT", "")
)
TEMPLATE_EXT = get(sys.argv, 9, os.getenv("VIMWIKI_TEMPLATE_EXT", ""))
ROOT_PATH = get(sys.argv, 10, os.getenv("VIMWIKI_ROOT_PATH", os.getcwd()))
# Only markdown is supported
if SYNTAX != "markdown":
sys.stderr.write("Unsupported syntax: " + SYNTAX)
sys.exit(1)
# Asign template
template = default_template
template_file = (
os.path.join(TEMPLATE_PATH, TEMPLATE_DEFAULT) + TEMPLATE_EXT
)
if os.path.isfile(template_file):
with open(template_file, "rb") as f:
template = f.read().decode()
# Get output filename
filename, _ = os.path.splitext(os.path.basename(INPUT_FILE))
output_file = os.path.join(OUTPUT_DIR, filename + ".html")
# parse extension into dict
extensions_env = os.getenv(
"VIMWIKI_MARKDOWN_EXTENSIONS",
"{}",
)
try:
extensions = json.loads(extensions_env)
except json.decoder.JSONDecodeError:
# backward compatible
extensions = extensions_env.split(",")
if isinstance(extensions, list):
extensions = {e: {} for e in extensions}
extension_names = default_extension + list(extensions.keys())
# Setup markdown parser
md = markdown.Markdown(
extensions=extension_names,
extension_configs=extensions,
)
md.inlinePatterns.deregister("link")
md.inlinePatterns.register(
LinkInlineProcessor(markdown.inlinepatterns.LINK_RE, md), "link", 160
)
with open(INPUT_FILE, "rb") as f:
content = ""
placeholders = {}
# Retrieve vimwiki placeholders
for line in f:
line = line.decode()[:-1]
if line.startswith("%nohtml"):
sys.exit(0)
elif line.startswith("%title"):
placeholders["%title%"] = line[7:]
elif line.startswith("%date"):
placeholders["%date%"] = line[6:]
elif line.startswith("%template"):
placeholders["template"] = line[10:]
else:
content += line + "\n"
# Set default values
if "%title%" not in placeholders:
placeholders["%title%"] = filename
if "%date%" not in placeholders:
placeholders["%date%"] = datetime.datetime.today().strftime(
"%Y-%m-%d"
)
if "template" in placeholders:
t = placeholders.pop("template")
template_file = os.path.join(TEMPLATE_PATH, t) + TEMPLATE_EXT
if os.path.isfile(template_file):
with open(template_file, "rb") as f:
template = f.read().decode()
# Parse template
for placeholder, value in placeholders.items():
template = template.replace(placeholder, value)
# use blank insted of os.getcwd() because - mean in root directory that contain css
template = template.replace(
"%root_path%", ROOT_PATH if ROOT_PATH != "-" else ""
)
# Parse content
content = md.convert(content)
# Merge template
template = template.replace("%content%", content)
with open(output_file, "wb") as o:
o.write(template.encode())
if __name__ == "__main__":
main()