forked from DevOps/deploy.stack
305 lines
9.8 KiB
Python
305 lines
9.8 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
Obsidian LiveSync CouchDB 配置生成器
|
|||
|
|
|
|||
|
|
此脚本用于生成运行 CouchDB 所需的配置文件，特别针对 Obsidian LiveSync 插件进行了优化配置。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
import re
|
|||
|
|
import getpass
|
|||
|
|
import platform
|
|||
|
|
from typing import Dict, Any, Optional, Tuple
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
def get_current_user_ids() -> Tuple[str, str]:
|
|||
|
|
"""
|
|||
|
|
获取当前用户的 UID 和 GID。
|
|||
|
|
在 Windows 上返回默认值，在 Linux/macOS 上返回实际值。
|
|||
|
|
"""
|
|||
|
|
if platform.system() in ['Linux', 'Darwin']: # Linux 或 macOS
|
|||
|
|
try:
|
|||
|
|
return str(os.getuid()), str(os.getgid())
|
|||
|
|
except AttributeError:
|
|||
|
|
pass
|
|||
|
|
# Windows 或其他系统返回默认值
|
|||
|
|
return "99", "100"
|
|||
|
|
|
|||
|
|
# 常量定义
|
|||
|
|
DEFAULT_CONFIG = {
|
|||
|
|
"couchdb_user": "obsidian_user",
|
|||
|
|
"host_port": "5984",
|
|||
|
|
"volume_name": "couchdb_data",
|
|||
|
|
"config_host_path": "./etc/couchdb/local.d",
|
|||
|
|
"single_node": "true",
|
|||
|
|
"cors_origins": "app://obsidian.md,capacitor://localhost,http://localhost",
|
|||
|
|
"puid": get_current_user_ids()[0], # 自动获取当前用户的 UID
|
|||
|
|
"pgid": get_current_user_ids()[1], # 自动获取当前用户的 GID
|
|||
|
|
"tz": "Asia/Shanghai"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
COMMON_TIMEZONES = {
|
|||
|
|
"1": "Asia/Shanghai",
|
|||
|
|
"2": "Asia/Singapore",
|
|||
|
|
"3": "America/New_York",
|
|||
|
|
"4": "Europe/London",
|
|||
|
|
"5": "Asia/Tokyo"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
def validate_port(port: str) -> bool:
|
|||
|
|
"""验证端口号是否有效。"""
|
|||
|
|
try:
|
|||
|
|
port_num = int(port)
|
|||
|
|
return 1 <= port_num <= 65535
|
|||
|
|
except ValueError:
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
def validate_username(username: str) -> bool:
|
|||
|
|
"""验证用户名是否有效。"""
|
|||
|
|
return bool(re.match(r'^[a-zA-Z0-9_-]{3,32}$', username))
|
|||
|
|
|
|||
|
|
def validate_password(password: str) -> bool:
|
|||
|
|
"""验证密码是否满足最低安全要求。"""
|
|||
|
|
if len(password) < 8:
|
|||
|
|
return False
|
|||
|
|
if not re.search(r'[A-Z]', password):
|
|||
|
|
return False
|
|||
|
|
if not re.search(r'[a-z]', password):
|
|||
|
|
return False
|
|||
|
|
if not re.search(r'[0-9]', password):
|
|||
|
|
return False
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
def get_input_with_default(prompt: str, default: str, validator: Optional[callable] = None) -> str:
|
|||
|
|
"""获取用户输入，支持默认值和验证。"""
|
|||
|
|
while True:
|
|||
|
|
value = input(f"{prompt} [{default}]：").strip()
|
|||
|
|
if not value:
|
|||
|
|
value = default
|
|||
|
|
if validator and not validator(value):
|
|||
|
|
print(f"输入无效，请重试。")
|
|||
|
|
continue
|
|||
|
|
return value
|
|||
|
|
|
|||
|
|
def get_password() -> str:
|
|||
|
|
"""安全地获取密码输入并进行确认。"""
|
|||
|
|
while True:
|
|||
|
|
password = getpass.getpass("请输入密码：").strip()
|
|||
|
|
if not validate_password(password):
|
|||
|
|
print("密码必须至少8个字符，并包含大写字母、小写字母和数字。")
|
|||
|
|
continue
|
|||
|
|
confirm = getpass.getpass("请确认密码：").strip()
|
|||
|
|
if password == confirm:
|
|||
|
|
return password
|
|||
|
|
print("两次输入的密码不一致，请重试。")
|
|||
|
|
|
|||
|
|
def get_timezone() -> str:
|
|||
|
|
"""获取用户选择的时区。"""
|
|||
|
|
print("\n可用的时区：")
|
|||
|
|
for key, value in COMMON_TIMEZONES.items():
|
|||
|
|
print(f" {key}. {value}")
|
|||
|
|
print(" （或输入自定义时区名称，例如：'Asia/Kuala_Lumpur'）")
|
|||
|
|
|
|||
|
|
while True:
|
|||
|
|
choice = input(f"选择时区 [{DEFAULT_CONFIG['tz']}]：").strip()
|
|||
|
|
if not choice:
|
|||
|
|
return DEFAULT_CONFIG['tz']
|
|||
|
|
if choice in COMMON_TIMEZONES:
|
|||
|
|
return COMMON_TIMEZONES[choice]
|
|||
|
|
# 基本验证自定义时区
|
|||
|
|
if '/' in choice and all(part.isalnum() or part in '_-' for part in choice.split('/')):
|
|||
|
|
return choice
|
|||
|
|
print("时区格式无效，请重试。")
|
|||
|
|
|
|||
|
|
def generate_docker_compose(config: Dict[str, Any]) -> str:
|
|||
|
|
"""生成 docker-compose.yml 内容。"""
|
|||
|
|
return f"""# 为 Obsidian LiveSync 生成的 docker-compose.yml
|
|||
|
|
# 使用最新的 Docker Compose 规范
|
|||
|
|
|
|||
|
|
services:
|
|||
|
|
couchdb-obsidian-livesync:
|
|||
|
|
container_name: couchdb-obsidian-livesync
|
|||
|
|
image: couchdb:3.3.3
|
|||
|
|
environment:
|
|||
|
|
PUID: {config['puid']}
|
|||
|
|
PGID: {config['pgid']}
|
|||
|
|
UMASK: 0022
|
|||
|
|
TZ: {config['tz']}
|
|||
|
|
COUCHDB_USER: {config['couchdb_user']}
|
|||
|
|
COUCHDB_PASSWORD: {config['couchdb_password']}
|
|||
|
|
volumes:
|
|||
|
|
- {config['volume_name']}:/opt/couchdb/data
|
|||
|
|
- {config['config_host_path']}:/opt/couchdb/etc/local.d
|
|||
|
|
ports:
|
|||
|
|
- "{config['host_port']}:5984"
|
|||
|
|
restart: unless-stopped
|
|||
|
|
networks:
|
|||
|
|
- couchdb_network
|
|||
|
|
|
|||
|
|
volumes:
|
|||
|
|
{config['volume_name']}:
|
|||
|
|
|
|||
|
|
networks:
|
|||
|
|
couchdb_network:
|
|||
|
|
driver: bridge
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
def generate_couchdb_config(config: Dict[str, Any]) -> str:
|
|||
|
|
"""生成 CouchDB local.ini 内容。"""
|
|||
|
|
return f"""[couchdb]
|
|||
|
|
single_node={config['single_node']}
|
|||
|
|
max_document_size = 50000000
|
|||
|
|
|
|||
|
|
[chttpd]
|
|||
|
|
require_valid_user = true
|
|||
|
|
max_http_request_size = 4294967296
|
|||
|
|
|
|||
|
|
[chttpd_auth]
|
|||
|
|
require_valid_user = true
|
|||
|
|
authentication_redirect = /_utils/session.html
|
|||
|
|
|
|||
|
|
[httpd]
|
|||
|
|
WWW-Authenticate = Basic realm="couchdb"
|
|||
|
|
enable_cors = true
|
|||
|
|
|
|||
|
|
[cors]
|
|||
|
|
origins = {config['cors_origins']}
|
|||
|
|
credentials = true
|
|||
|
|
headers = accept, authorization, content-type, origin, referer
|
|||
|
|
methods = GET, PUT, POST, HEAD, DELETE
|
|||
|
|
max_age = 3600
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
def save_config_files(config: Dict[str, Any]) -> None:
|
|||
|
|
"""保存所有配置文件。"""
|
|||
|
|
# 创建配置目录
|
|||
|
|
config_dir = Path(config['config_host_path'])
|
|||
|
|
config_dir.mkdir(parents=True, exist_ok=True)
|
|||
|
|
|
|||
|
|
# 保存 docker-compose.yml
|
|||
|
|
with open('docker-compose.yml', 'w') as f:
|
|||
|
|
f.write(generate_docker_compose(config))
|
|||
|
|
print(f"\n✓ 已生成 docker-compose.yml")
|
|||
|
|
|
|||
|
|
# 保存 CouchDB 配置
|
|||
|
|
config_path = config_dir / "my_obsidian_config.ini"
|
|||
|
|
with open(config_path, 'w') as f:
|
|||
|
|
f.write(generate_couchdb_config(config))
|
|||
|
|
print(f"✓ 已生成 {config_path}")
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
"""主函数，运行配置生成器。"""
|
|||
|
|
print("=== Obsidian LiveSync CouchDB 配置生成器 ===")
|
|||
|
|
print("此脚本将帮助您设置用于 Obsidian LiveSync 的 CouchDB。")
|
|||
|
|
print("随时可以按 Ctrl+C 退出。\n")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
config = DEFAULT_CONFIG.copy()
|
|||
|
|
|
|||
|
|
# 获取用户名
|
|||
|
|
config['couchdb_user'] = get_input_with_default(
|
|||
|
|
"请输入 CouchDB 管理员用户名",
|
|||
|
|
DEFAULT_CONFIG['couchdb_user'],
|
|||
|
|
validate_username
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 获取密码
|
|||
|
|
print("\n请输入 CouchDB 管理员密码（必须至少8个字符，包含大写字母、小写字母和数字）")
|
|||
|
|
config['couchdb_password'] = get_password()
|
|||
|
|
|
|||
|
|
# 获取时区
|
|||
|
|
config['tz'] = get_timezone()
|
|||
|
|
|
|||
|
|
# 获取端口
|
|||
|
|
config['host_port'] = get_input_with_default(
|
|||
|
|
"请输入 CouchDB 主机端口",
|
|||
|
|
DEFAULT_CONFIG['host_port'],
|
|||
|
|
validate_port
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 获取卷名称
|
|||
|
|
config['volume_name'] = get_input_with_default(
|
|||
|
|
"请输入 CouchDB 数据卷名称",
|
|||
|
|
DEFAULT_CONFIG['volume_name']
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 获取配置路径
|
|||
|
|
config['config_host_path'] = get_input_with_default(
|
|||
|
|
"请输入配置文件目录路径",
|
|||
|
|
DEFAULT_CONFIG['config_host_path']
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 获取单节点模式
|
|||
|
|
single_node = get_input_with_default(
|
|||
|
|
"是否使用单节点模式（true/false）",
|
|||
|
|
DEFAULT_CONFIG['single_node']
|
|||
|
|
).lower()
|
|||
|
|
config['single_node'] = "true" if single_node in ["true", "t", "yes", "y"] else "false"
|
|||
|
|
|
|||
|
|
# 获取 CORS 源
|
|||
|
|
config['cors_origins'] = get_input_with_default(
|
|||
|
|
"请输入允许的 CORS 源（用逗号分隔）",
|
|||
|
|
DEFAULT_CONFIG['cors_origins']
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 获取 PUID/PGID
|
|||
|
|
current_uid, current_gid = get_current_user_ids()
|
|||
|
|
if platform.system() in ['Linux', 'Darwin']:
|
|||
|
|
print(f"\n检测到当前用户 ID：UID={current_uid}, GID={current_gid}")
|
|||
|
|
print("这些值将用于设置容器内的文件权限。")
|
|||
|
|
use_current = get_input_with_default(
|
|||
|
|
"是否使用当前用户的 UID/GID（推荐）",
|
|||
|
|
"yes"
|
|||
|
|
).lower() in ["yes", "y", "true", "t"]
|
|||
|
|
|
|||
|
|
if use_current:
|
|||
|
|
config['puid'] = current_uid
|
|||
|
|
config['pgid'] = current_gid
|
|||
|
|
else:
|
|||
|
|
print("\n请输入自定义的 PUID/PGID")
|
|||
|
|
config['puid'] = get_input_with_default(
|
|||
|
|
"请输入 PUID",
|
|||
|
|
current_uid,
|
|||
|
|
lambda x: x.isdigit()
|
|||
|
|
)
|
|||
|
|
config['pgid'] = get_input_with_default(
|
|||
|
|
"请输入 PGID",
|
|||
|
|
current_gid,
|
|||
|
|
lambda x: x.isdigit()
|
|||
|
|
)
|
|||
|
|
else:
|
|||
|
|
print("\n当前系统不支持自动获取用户 ID，将使用默认值（99/100）")
|
|||
|
|
print("如果需要自定义，请手动输入。")
|
|||
|
|
config['puid'] = get_input_with_default(
|
|||
|
|
"请输入 PUID",
|
|||
|
|
DEFAULT_CONFIG['puid'],
|
|||
|
|
lambda x: x.isdigit()
|
|||
|
|
)
|
|||
|
|
config['pgid'] = get_input_with_default(
|
|||
|
|
"请输入 PGID",
|
|||
|
|
DEFAULT_CONFIG['pgid'],
|
|||
|
|
lambda x: x.isdigit()
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 保存所有配置文件
|
|||
|
|
save_config_files(config)
|
|||
|
|
|
|||
|
|
print("\n=== 配置完成 ===")
|
|||
|
|
print(f"✓ CouchDB 将在 http://localhost:{config['host_port']}/_utils/ 访问")
|
|||
|
|
print(f"✓ 用户名：{config['couchdb_user']}")
|
|||
|
|
print(f"✓ 文件权限：UID={config['puid']}, GID={config['pgid']}")
|
|||
|
|
print("\n要启动 CouchDB，请运行：")
|
|||
|
|
print(" docker-compose up -d")
|
|||
|
|
print("\n要查看日志，请运行：")
|
|||
|
|
print(" docker-compose logs -f")
|
|||
|
|
|
|||
|
|
except KeyboardInterrupt:
|
|||
|
|
print("\n\n配置已取消。")
|
|||
|
|
sys.exit(1)
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"\n错误：{str(e)}")
|
|||
|
|
sys.exit(1)
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|