mirror of
https://github.com/langgenius/dify-docs.git
synced 2026-03-27 13:28:32 +07:00
171 lines
5.6 KiB
Python
171 lines
5.6 KiB
Python
#!/usr/bin/env python3
|
||
"""
|
||
GitBook to Mintlify格式转换脚本
|
||
|
||
将GitBook风格的Markdown组件转换为Mintlify MDX格式。
|
||
例如将 {% hint style="warning" %} xxx {% endhint %}
|
||
转换为 <Warning>xxx</Warning>
|
||
"""
|
||
|
||
import re
|
||
import os
|
||
|
||
# 定义转换映射表
|
||
STYLE_MAPPING = {
|
||
"info": "Info",
|
||
"warning": "Warning",
|
||
"danger": "Danger",
|
||
"success": "Success",
|
||
"note": "Note",
|
||
"tip": "Tip",
|
||
"caution": "Caution"
|
||
}
|
||
|
||
def convert_hints(content):
|
||
"""转换GitBook的hint组件为Mintlify组件"""
|
||
# 使用正则表达式匹配所有hint块
|
||
hint_pattern = r'{%\s*hint\s+style="([^"]*)"\s*%}(.*?){%\s*endhint\s*%}'
|
||
|
||
def replace_hint(match):
|
||
style = match.group(1)
|
||
content = match.group(2).strip()
|
||
|
||
# 获取对应的Mintlify组件名
|
||
component = STYLE_MAPPING.get(style.lower(), "Info")
|
||
|
||
# 处理多行内容
|
||
if '\n' in content:
|
||
# 保持缩进
|
||
lines = content.split('\n')
|
||
formatted_content = '\n'.join(lines)
|
||
return f"<{component}>\n{formatted_content}\n</{component}>"
|
||
else:
|
||
# 单行内容
|
||
return f"<{component}>{content}</{component}>"
|
||
|
||
# 使用re.DOTALL使.能匹配所有字符包括换行符
|
||
return re.sub(hint_pattern, replace_hint, content, flags=re.DOTALL)
|
||
|
||
def convert_tabs(content):
|
||
"""转换GitBook的tabs组件为Mintlify Tabs组件"""
|
||
# 找到所有tabs块
|
||
tabs_start_pattern = r'{%\s*tabs\s*%}'
|
||
tabs_end_pattern = r'{%\s*endtabs\s*%}'
|
||
tab_pattern = r'{%\s*tab title="([^"]*)"\s*%}(.*?){%\s*endtab\s*%}'
|
||
|
||
# 先替换tab项
|
||
def replace_tab(match):
|
||
title = match.group(1)
|
||
tab_content = match.group(2).strip()
|
||
return f'<Tab title="{title}">\n{tab_content}\n</Tab>'
|
||
|
||
content = re.sub(tab_pattern, replace_tab, content, flags=re.DOTALL)
|
||
|
||
# 再替换整个tabs块
|
||
content = re.sub(tabs_start_pattern, '<Tabs>', content)
|
||
content = re.sub(tabs_end_pattern, '</Tabs>', content)
|
||
|
||
return content
|
||
|
||
def convert_videos(content):
|
||
"""转换GitBook的视频embed为Mintlify video格式"""
|
||
# 匹配多种可能的embed视频格式
|
||
video_pattern1 = r'{%\s*embed\s+url="([^"]+)"\s*%}'
|
||
video_pattern2 = r'{%\s*embed\s+url=[\"]([^\"]+)[\"]\s*%}'
|
||
# 更宽松的模式，匹配各种可能的url格式
|
||
video_pattern3 = r'{%\s*embed\s+url\s*=\s*["\']?([^"\'>}\s]+)["\']?\s*%}'
|
||
|
||
def replace_video(match):
|
||
url = match.group(1).strip()
|
||
|
||
# 在转换后打印匹配到的视频链接
|
||
print(f" - 找到视频链接: {url}")
|
||
|
||
# 生成简单的video标签格式
|
||
return f'<video controls src="{url}" width="100%"></video>'
|
||
|
||
# 先应用第一种模式
|
||
content = re.sub(video_pattern1, replace_video, content)
|
||
# 再应用第二种模式
|
||
content = re.sub(video_pattern2, replace_video, content)
|
||
# 最后应用更宽松的模式
|
||
content = re.sub(video_pattern3, replace_video, content)
|
||
|
||
return content
|
||
|
||
def convert_file(file_path):
|
||
"""转换指定文件中的GitBook组件为Mintlify格式"""
|
||
try:
|
||
# 读取文件内容
|
||
with open(file_path, 'r', encoding='utf-8') as file:
|
||
content = file.read()
|
||
|
||
# 执行转换
|
||
original_content = content # 保存原始内容用于比较
|
||
|
||
print(f"\n开始转换文件: {file_path}")
|
||
|
||
# 检查文件中的特定模式
|
||
hint_count = len(re.findall(r'{%\s*hint', content))
|
||
tabs_count = len(re.findall(r'{%\s*tabs', content))
|
||
embed_count = len(re.findall(r'{%\s*embed', content))
|
||
|
||
print(f" 检测到: {hint_count} 个 hint 标签, {tabs_count} 个 tabs 标签, {embed_count} 个 embed 标签")
|
||
|
||
# 应用转换
|
||
content = convert_hints(content)
|
||
content = convert_tabs(content)
|
||
content = convert_videos(content)
|
||
|
||
# 只有在内容有变化时才写入文件
|
||
if content != original_content:
|
||
# 备份原文件
|
||
backup_path = f"{file_path}.bak"
|
||
with open(backup_path, 'w', encoding='utf-8') as backup:
|
||
backup.write(original_content)
|
||
|
||
# 写入转换后的内容
|
||
with open(file_path, 'w', encoding='utf-8') as file:
|
||
file.write(content)
|
||
|
||
print(f"✅ 文件已转换并备份: {file_path}")
|
||
else:
|
||
print(f"ℹ️ 文件未发生变化: {file_path}")
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f"❌ 转换失败: {str(e)}")
|
||
return False
|
||
|
||
def main():
|
||
"""主函数，处理用户输入的文件路径"""
|
||
print("GitBook 到 Mintlify 格式转换工具")
|
||
print("=" * 40)
|
||
|
||
while True:
|
||
file_path = input("\n请输入要转换的MD/MDX文件路径 (输入q退出): ").strip()
|
||
|
||
if file_path.lower() == 'q':
|
||
break
|
||
|
||
if not os.path.exists(file_path):
|
||
print(f"❌ 错误: 文件 '{file_path}' 不存在!")
|
||
continue
|
||
|
||
if os.path.isdir(file_path):
|
||
print(f"❌ 错误: '{file_path}' 是目录，请输入文件路径!")
|
||
continue
|
||
|
||
if not file_path.endswith(('.md', '.mdx')):
|
||
print(f"⚠️ 警告: 文件 '{file_path}' 不是Markdown或MDX文件。确定要继续吗? (y/n)")
|
||
if input().lower() != 'y':
|
||
continue
|
||
|
||
convert_file(file_path)
|
||
|
||
print("\n程序已退出。")
|
||
|
||
if __name__ == "__main__":
|
||
main()
|