import base64
import datetime as dt
import hashlib
import hmac
import html
import mimetypes
import os
import re
import secrets
import shlex
import shutil
import sqlite3
import subprocess
import time
from pathlib import Path, PurePosixPath
from urllib.parse import parse_qsl, quote, unquote, urlencode, urlparse
import bleach
import markdown
from bottle import (
Bottle,
HTTPResponse,
TEMPLATE_PATH,
abort,
redirect,
request,
response,
run,
static_file,
template,
)
from mercurial.hgweb import hgweb as make_hgweb
def env_bool(name, default=False):
value = os.environ.get(name)
if value is None:
return default
return value.lower() in {"1", "true", "yes", "on"}
def env_int(name, default, minimum=0):
try:
value = int(os.environ.get(name, str(default)))
except ValueError:
return default
return max(minimum, value)
BASE_DIR = Path(__file__).resolve().parent
DATA_DIR = BASE_DIR / "data"
DB_PATH = Path(os.environ.get("HG_HOST_DB", DATA_DIR / "hghost.sqlite3"))
REPO_ROOT = Path(os.environ.get("HG_HOST_REPO_ROOT", DATA_DIR / "repos"))
DEFAULT_SECRET_KEY = "dev-secret-change-me"
SECRET_KEY = os.environ.get("SECRET_KEY", DEFAULT_SECRET_KEY)
DEBUG = env_bool("HG_HOST_DEBUG")
PASSWORD_ITERATIONS = 260_000
SQLITE_BUSY_TIMEOUT_MS = 30_000
MAX_FORM_BYTES = env_int("HG_HOST_MAX_FORM_BYTES", 64 * 1024)
MAX_RENDER_BYTES = env_int("HG_HOST_MAX_RENDER_BYTES", 256 * 1024)
MAX_HG_RESPONSE_BYTES = env_int("HG_HOST_MAX_HG_RESPONSE_BYTES", 256 * 1024 * 1024)
RATE_LIMIT_ENABLED = env_bool("HG_HOST_RATE_LIMIT_ENABLED", True)
RATE_LIMIT_MAX_FAILURES = env_int("HG_HOST_RATE_LIMIT_MAX_FAILURES", 5, minimum=1)
RATE_LIMIT_WINDOW_SECONDS = env_int("HG_HOST_RATE_LIMIT_WINDOW_SECONDS", 300, minimum=1)
RATE_LIMIT_COOLDOWN_SECONDS = env_int("HG_HOST_RATE_LIMIT_COOLDOWN_SECONDS", 300, minimum=1)
SLUG_RE = re.compile(r"^[a-z0-9][a-z0-9._-]{1,62}$")
REV_RE = re.compile(r"^(null|[0-9a-fA-F]{1,40})$")
REF_TYPE_BRANCH = "branch"
REF_TYPE_BOOKMARK = "bookmark"
REF_TYPE_TAG = "tag"
REF_TYPE_TIP = "tip"
REF_TYPE_COMMIT = "commit"
REF_TYPES = {REF_TYPE_BRANCH, REF_TYPE_BOOKMARK, REF_TYPE_TAG, REF_TYPE_TIP, REF_TYPE_COMMIT}
PULL_REQUEST_REF_TYPES = {REF_TYPE_BRANCH, REF_TYPE_BOOKMARK, REF_TYPE_TIP}
TARGET_PULL_REQUEST_REF_TYPES = {REF_TYPE_BRANCH, REF_TYPE_BOOKMARK}
REF_PICKER_TABS = {"overview", "source", "commits", "tags", "branches", "bookmarks"}
REF_QUERY_KEYS = {"ref", "ref_type", "ref_value"}
REF_VALUE_SEPARATOR = "|"
SCRIPT_STYLE_RE = re.compile(r"(?is)<(script|style)\b[^>]*>.*?</\1>")
RESERVED_USERNAMES = {"dashboard", "favicon.ico", "hg", "login", "logout", "new", "settings", "signup", "static", "harrisonerd"}
WRITE_HG_COMMANDS = {"pushkey", "unbundle"}
CSRF_COOKIE_NAME = "csrf_token"
CSRF_FORM_FIELD = "_csrf_token"
NULL_REV = "null"
NULL_NODE = "0" * 40
README_CANDIDATES = ("README.md", "README.rst", "README.txt", "README")
MARKDOWN_EXTENSIONS = ("extra", "sane_lists")
MARKDOWN_TAGS = {
"a",
"abbr",
"acronym",
"blockquote",
"br",
"code",
"del",
"details",
"em",
"h1",
"h2",
"h3",
"h4",
"h5",
"h6",
"hr",
"img",
"li",
"ol",
"p",
"pre",
"strong",
"summary",
"table",
"tbody",
"td",
"th",
"thead",
"tr",
"ul",
}
MARKDOWN_ATTRIBUTES = {
"a": ["href", "title"],
"abbr": ["title"],
"acronym": ["title"],
"img": ["alt", "src", "title"],
"td": ["align"],
"th": ["align"],
}
MARKDOWN_LINK_TAGS = {"a"}
MARKDOWN_LINK_ATTRIBUTES = {"a": ["href", "title"]}
HIGHLIGHT_LANGUAGE_BY_EXTENSION = {
".c": "language-c",
".cc": "language-cpp",
".cpp": "language-cpp",
".cs": "language-csharp",
".css": "language-css",
".go": "language-go",
".h": "language-c",
".hpp": "language-cpp",
".html": "language-html",
".ini": "language-ini",
".java": "language-java",
".js": "language-javascript",
".json": "language-json",
".jsx": "language-javascript",
".lua": "language-lua",
".md": "language-markdown",
".php": "language-php",
".py": "language-python",
".rb": "language-ruby",
".rs": "language-rust",
".sh": "language-bash",
".sql": "language-sql",
".toml": "language-ini",
".ts": "language-typescript",
".tsx": "language-typescript",
".txt": "language-plaintext",
".xml": "language-xml",
".yaml": "language-yaml",
".yml": "language-yaml",
}
HIGHLIGHT_LANGUAGE_BY_NAME = {
"dockerfile": "language-dockerfile",
"makefile": "language-makefile",
}
SECURITY_HEADERS = {
"X-Content-Type-Options": "nosniff",
"Referrer-Policy": "same-origin",
"X-Frame-Options": "DENY",
}
CSP_HEADER = (
"default-src 'self'; "
"base-uri 'self'; "
"frame-ancestors 'none'; "
"form-action 'self'; "
"object-src 'none'; "
"img-src 'self' data: http: https:; "
"style-src 'self' 'unsafe-inline' https://cdnjs.cloudflare.com; "
"script-src 'self' 'unsafe-inline' https://cdnjs.cloudflare.com"
)
AUTH_FAILURES = {}
TEMPLATE_PATH.insert(0, str(BASE_DIR / "templates"))
app = Bottle()
class HgCommandError(RuntimeError):
def __init__(self, message, returncode=1):
super().__init__(message)
self.returncode = returncode
class HgResponseTooLarge(RuntimeError):
pass
def validate_startup_config():
if not DEBUG and SECRET_KEY == DEFAULT_SECRET_KEY:
raise RuntimeError("SECRET_KEY must be set to a non-default value when HG_HOST_DEBUG is disabled.")
def utcnow():
return dt.datetime.now(dt.UTC).replace(microsecond=0).isoformat().replace("+00:00", "Z")
def ensure_dirs():
DB_PATH.parent.mkdir(parents=True, exist_ok=True)
REPO_ROOT.mkdir(parents=True, exist_ok=True)
def configure_db_connection(conn):
conn.execute("PRAGMA foreign_keys = ON")
conn.execute(f"PRAGMA busy_timeout = {SQLITE_BUSY_TIMEOUT_MS}")
conn.execute("PRAGMA synchronous = NORMAL")
def db_connect():
conn = sqlite3.connect(DB_PATH, timeout=SQLITE_BUSY_TIMEOUT_MS / 1000)
conn.row_factory = sqlite3.Row
configure_db_connection(conn)
return conn
def init_db():
ensure_dirs()
with db_connect() as conn:
conn.execute("PRAGMA journal_mode = WAL")
with db_connect() as conn:
conn.executescript(
"""
CREATE TABLE IF NOT EXISTS users (
id INTEGER PRIMARY KEY AUTOINCREMENT,
username TEXT NOT NULL UNIQUE,
password_hash TEXT NOT NULL,
display_name TEXT NOT NULL DEFAULT '',
bio TEXT NOT NULL DEFAULT '',
website TEXT NOT NULL DEFAULT '',
created_at TEXT NOT NULL
);
CREATE TABLE IF NOT EXISTS repositories (
id INTEGER PRIMARY KEY AUTOINCREMENT,
owner_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
name TEXT NOT NULL,
description TEXT NOT NULL DEFAULT '',
forked_from_repo_id INTEGER REFERENCES repositories(id) ON DELETE SET NULL,
forked_at TEXT,
forked_from_node TEXT NOT NULL DEFAULT '',
created_at TEXT NOT NULL,
updated_at TEXT NOT NULL,
UNIQUE(owner_id, name)
);
CREATE TABLE IF NOT EXISTS issues (
id INTEGER PRIMARY KEY AUTOINCREMENT,
repo_id INTEGER NOT NULL REFERENCES repositories(id) ON DELETE CASCADE,
author_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
number INTEGER NOT NULL,
title TEXT NOT NULL,
body TEXT NOT NULL DEFAULT '',
status TEXT NOT NULL DEFAULT 'open',
created_at TEXT NOT NULL,
updated_at TEXT NOT NULL,
closed_at TEXT,
UNIQUE(repo_id, number)
);
CREATE TABLE IF NOT EXISTS issue_comments (
id INTEGER PRIMARY KEY AUTOINCREMENT,
issue_id INTEGER NOT NULL REFERENCES issues(id) ON DELETE CASCADE,
author_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
body TEXT NOT NULL,
created_at TEXT NOT NULL,
updated_at TEXT NOT NULL
);
CREATE TABLE IF NOT EXISTS repo_contributors (
repo_id INTEGER NOT NULL REFERENCES repositories(id) ON DELETE CASCADE,
user_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
added_by_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
created_at TEXT NOT NULL,
PRIMARY KEY (repo_id, user_id)
);
CREATE TABLE IF NOT EXISTS repo_stars (
repo_id INTEGER NOT NULL REFERENCES repositories(id) ON DELETE CASCADE,
user_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
created_at TEXT NOT NULL,
PRIMARY KEY (repo_id, user_id)
);
CREATE TABLE IF NOT EXISTS pull_requests (
id INTEGER PRIMARY KEY AUTOINCREMENT,
target_repo_id INTEGER NOT NULL REFERENCES repositories(id) ON DELETE CASCADE,
source_repo_id INTEGER NOT NULL REFERENCES repositories(id) ON DELETE CASCADE,
author_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
number INTEGER NOT NULL,
title TEXT NOT NULL,
body TEXT NOT NULL DEFAULT '',
status TEXT NOT NULL DEFAULT 'open',
base_node TEXT NOT NULL,
source_node TEXT NOT NULL,
target_ref_type TEXT NOT NULL DEFAULT '',
target_ref_name TEXT NOT NULL DEFAULT '',
source_ref_type TEXT NOT NULL DEFAULT '',
source_ref_name TEXT NOT NULL DEFAULT '',
created_at TEXT NOT NULL,
updated_at TEXT NOT NULL,
closed_at TEXT,
merged_at TEXT,
merged_by_id INTEGER REFERENCES users(id) ON DELETE SET NULL,
merge_node TEXT NOT NULL DEFAULT '',
UNIQUE(target_repo_id, number)
);
CREATE TABLE IF NOT EXISTS pull_request_comments (
id INTEGER PRIMARY KEY AUTOINCREMENT,
pull_request_id INTEGER NOT NULL REFERENCES pull_requests(id) ON DELETE CASCADE,
author_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
body TEXT NOT NULL,
created_at TEXT NOT NULL,
updated_at TEXT NOT NULL
);
CREATE TABLE IF NOT EXISTS commit_comments (
id INTEGER PRIMARY KEY AUTOINCREMENT,
repo_id INTEGER NOT NULL REFERENCES repositories(id) ON DELETE CASCADE,
commit_node TEXT NOT NULL,
author_id INTEGER NOT NULL REFERENCES users(id) ON DELETE CASCADE,
body TEXT NOT NULL,
created_at TEXT NOT NULL,
updated_at TEXT NOT NULL
);
CREATE INDEX IF NOT EXISTS idx_repositories_owner ON repositories(owner_id);
CREATE INDEX IF NOT EXISTS idx_issues_repo_number ON issues(repo_id, number);
CREATE INDEX IF NOT EXISTS idx_issues_repo_status ON issues(repo_id, status);
CREATE INDEX IF NOT EXISTS idx_issue_comments_issue ON issue_comments(issue_id);
CREATE INDEX IF NOT EXISTS idx_repo_contributors_user ON repo_contributors(user_id);
CREATE INDEX IF NOT EXISTS idx_repo_stars_user ON repo_stars(user_id);
CREATE INDEX IF NOT EXISTS idx_pull_requests_target_status ON pull_requests(target_repo_id, status);
CREATE INDEX IF NOT EXISTS idx_pull_requests_source ON pull_requests(source_repo_id);
CREATE INDEX IF NOT EXISTS idx_pull_request_comments_pull_request ON pull_request_comments(pull_request_id);
CREATE INDEX IF NOT EXISTS idx_commit_comments_commit ON commit_comments(repo_id, commit_node);
"""
)
ensure_user_profile_columns(conn)
ensure_repository_collaboration_columns(conn)
ensure_pull_request_ref_columns(conn)
def ensure_user_profile_columns(conn):
columns = {row["name"] for row in conn.execute("PRAGMA table_info(users)")}
profile_columns = {
"display_name": "ALTER TABLE users ADD COLUMN display_name TEXT NOT NULL DEFAULT ''",
"bio": "ALTER TABLE users ADD COLUMN bio TEXT NOT NULL DEFAULT ''",
"website": "ALTER TABLE users ADD COLUMN website TEXT NOT NULL DEFAULT ''",
}
for name, ddl in profile_columns.items():
if name not in columns:
conn.execute(ddl)
def ensure_repository_collaboration_columns(conn):
columns = {row["name"] for row in conn.execute("PRAGMA table_info(repositories)")}
collaboration_columns = {
"forked_from_repo_id": (
"ALTER TABLE repositories "
"ADD COLUMN forked_from_repo_id INTEGER REFERENCES repositories(id) ON DELETE SET NULL"
),
"forked_at": "ALTER TABLE repositories ADD COLUMN forked_at TEXT",
"forked_from_node": "ALTER TABLE repositories ADD COLUMN forked_from_node TEXT NOT NULL DEFAULT ''",
}
for name, ddl in collaboration_columns.items():
if name not in columns:
conn.execute(ddl)
conn.execute("CREATE INDEX IF NOT EXISTS idx_repositories_forked_from ON repositories(forked_from_repo_id)")
def ensure_pull_request_ref_columns(conn):
columns = {row["name"] for row in conn.execute("PRAGMA table_info(pull_requests)")}
ref_columns = {
"target_ref_type": "ALTER TABLE pull_requests ADD COLUMN target_ref_type TEXT NOT NULL DEFAULT ''",
"target_ref_name": "ALTER TABLE pull_requests ADD COLUMN target_ref_name TEXT NOT NULL DEFAULT ''",
"source_ref_type": "ALTER TABLE pull_requests ADD COLUMN source_ref_type TEXT NOT NULL DEFAULT ''",
"source_ref_name": "ALTER TABLE pull_requests ADD COLUMN source_ref_name TEXT NOT NULL DEFAULT ''",
}
for name, ddl in ref_columns.items():
if name not in columns:
conn.execute(ddl)
def normalize_slug(value, label):
slug = (value or "").strip().lower()
if not SLUG_RE.match(slug):
raise ValueError(f"{label} must be 2-63 characters: lowercase letters, numbers, dots, dashes, or underscores.")
if label.lower().startswith("username") and slug in RESERVED_USERNAMES:
raise ValueError("Username is reserved.")
if slug in {".", ".."} or slug.endswith(".hg"):
raise ValueError(f"{label} is reserved.")
return slug
def clean_repo_path(path):
raw = (path or "").strip("/")
if not raw:
return ""
raw_parts = raw.split("/")
if any(part in {"", ".", ".."} for part in raw_parts):
abort(400, "Invalid repository path.")
parts = PurePosixPath(raw).parts
if any(part in {"", ".", ".."} for part in parts):
abort(400, "Invalid repository path.")
return "/".join(parts)
def hash_password(password):
salt = secrets.token_hex(16)
digest = hashlib.pbkdf2_hmac(
"sha256",
password.encode("utf-8"),
salt.encode("ascii"),
PASSWORD_ITERATIONS,
).hex()
return f"pbkdf2_sha256${PASSWORD_ITERATIONS}${salt}${digest}"
def verify_password(password, stored_hash):
try:
algorithm, iterations, salt, expected = stored_hash.split("$", 3)
if algorithm != "pbkdf2_sha256":
return False
digest = hashlib.pbkdf2_hmac(
"sha256",
password.encode("utf-8"),
salt.encode("ascii"),
int(iterations),
).hex()
except (ValueError, TypeError):
return False
return hmac.compare_digest(digest, expected)
def get_user_by_id(user_id):
try:
user_id = int(user_id)
except (TypeError, ValueError):
return None
with db_connect() as conn:
return conn.execute("SELECT * FROM users WHERE id = ?", (user_id,)).fetchone()
def get_user_by_username(username):
with db_connect() as conn:
return conn.execute("SELECT * FROM users WHERE username = ?", (username,)).fetchone()
def current_user():
return request.environ.get("hghost.user")
def request_is_secure():
forwarded = request.get_header("X-Forwarded-Proto", "")
if forwarded:
return forwarded.split(",", 1)[0].strip().lower() == "https"
return request.urlparts.scheme == "https"
def is_hg_request_path():
path = request.environ.get("PATH_INFO", request.path) or ""
return path == "/hg" or path.startswith("/hg/")
def csrf_token():
token = request.environ.get("hghost.csrf_token")
if token:
return token
token = request.get_cookie(CSRF_COOKIE_NAME, secret=SECRET_KEY)
if not token:
token = secrets.token_urlsafe(32)
response.set_cookie(
CSRF_COOKIE_NAME,
token,
secret=SECRET_KEY,
httponly=True,
secure=request_is_secure(),
samesite="Lax",
path="/",
)
request.environ["hghost.csrf_token"] = token
return token
def csrf_field():
return (
f'<input type="hidden" name="{CSRF_FORM_FIELD}" '
f'value="{html.escape(csrf_token(), quote=True)}">'
)
def validate_csrf_token():
expected = request.get_cookie(CSRF_COOKIE_NAME, secret=SECRET_KEY)
submitted = request.forms.get(CSRF_FORM_FIELD, "")
if not expected or not submitted or not hmac.compare_digest(expected, submitted):
abort(403, "Invalid CSRF token.")
def request_content_length():
try:
return int(request.environ.get("CONTENT_LENGTH") or 0)
except ValueError:
return 0
def auth_rate_key(kind, identifier=""):
identifier = (identifier or "").strip().lower()[:100]
remote_addr = request.environ.get("REMOTE_ADDR", "")
return f"{kind}:{remote_addr}:{identifier}"
def rate_limit_blocked(kind, identifier=""):
if not RATE_LIMIT_ENABLED:
return False
now = time.time()
prune_auth_failures(now)
record = AUTH_FAILURES.get(auth_rate_key(kind, identifier))
return bool(record and record.get("blocked_until", 0) > now)
def prune_auth_failures(now=None):
now = now or time.time()
for key, record in list(AUTH_FAILURES.items()):
if record.get("reset_at", 0) <= now and record.get("blocked_until", 0) <= now:
AUTH_FAILURES.pop(key, None)
def record_auth_failure(kind, identifier=""):
if not RATE_LIMIT_ENABLED:
return
now = time.time()
prune_auth_failures(now)
key = auth_rate_key(kind, identifier)
record = AUTH_FAILURES.get(key)
if not record or record.get("reset_at", 0) <= now:
record = {"count": 0, "reset_at": now + RATE_LIMIT_WINDOW_SECONDS, "blocked_until": 0}
record["count"] += 1
if record["count"] >= RATE_LIMIT_MAX_FAILURES:
record["blocked_until"] = now + RATE_LIMIT_COOLDOWN_SECONDS
AUTH_FAILURES[key] = record
def clear_auth_failures(kind, identifier=""):
AUTH_FAILURES.pop(auth_rate_key(kind, identifier), None)
def too_many_requests_response(close_connection=False):
headers = {"Retry-After": str(RATE_LIMIT_COOLDOWN_SECONDS)}
if close_connection:
headers["Connection"] = "close"
return HTTPResponse(
"Too many failed attempts. Try again later.\n",
status=429,
headers=headers,
content_type="text/plain; charset=utf-8",
)
@app.hook("before_request")
def load_current_user():
user_id = request.get_cookie("session", secret=SECRET_KEY)
request.environ["hghost.user"] = get_user_by_id(user_id) if user_id else None
@app.hook("before_request")
def enforce_browser_post_security():
if request.method != "POST" or is_hg_request_path():
return
if MAX_FORM_BYTES and request_content_length() > MAX_FORM_BYTES:
abort(413, "Request body too large.")
validate_csrf_token()
@app.hook("after_request")
def add_security_headers():
for key, value in SECURITY_HEADERS.items():
response.headers.setdefault(key, value)
if not is_hg_request_path():
response.headers.setdefault("Content-Security-Policy", CSP_HEADER)
def render(template_name, **context):
context.setdefault("user", current_user())
context.setdefault("error", None)
context.setdefault("notice", None)
context.setdefault("csrf_field", csrf_field)
context.setdefault("render_markdown_links", render_markdown_links)
context.setdefault("render_repo_description", render_repo_description)
context.setdefault("format_ref_label", format_ref_label)
context.setdefault("url_with_ref", url_with_ref)
context.setdefault("current_url_with_ref", current_url_with_ref)
context.setdefault("ref_option_label", ref_option_label)
return template(template_name, **context)
def login_user(user):
response.set_cookie(
"session",
str(user["id"]),
secret=SECRET_KEY,
httponly=True,
secure=request_is_secure(),
samesite="Lax",
path="/",
)
def logout_user():
response.delete_cookie("session", path="/")
def require_login():
user = current_user()
if user:
return user
next_url = request.fullpath if request.query_string else request.path
redirect("/login?next=" + quote(next_url, safe="/?=&"))
def safe_next_url(value):
if value and value.startswith("/") and not value.startswith("//"):
return value
return "/"
def repo_path(owner_username, repo_name):
path = REPO_ROOT / owner_username / repo_name
root = REPO_ROOT.resolve()
resolved = path.resolve()
if root != resolved and root not in resolved.parents:
abort(400, "Invalid repository path.")
return path
def get_repo(owner_username, repo_name):
with db_connect() as conn:
return conn.execute(
"""
SELECT repositories.*, users.username AS owner_username
FROM repositories
JOIN users ON users.id = repositories.owner_id
WHERE users.username = ? AND repositories.name = ?
""",
(owner_username, repo_name),
).fetchone()
def get_repo_by_id(repo_id):
with db_connect() as conn:
return conn.execute(
"""
SELECT repositories.*, users.username AS owner_username
FROM repositories
JOIN users ON users.id = repositories.owner_id
WHERE repositories.id = ?
""",
(repo_id,),
).fetchone()
def list_public_repos(limit=100):
with db_connect() as conn:
return conn.execute(
"""
SELECT repositories.*, users.username AS owner_username
FROM repositories
JOIN users ON users.id = repositories.owner_id
ORDER BY repositories.updated_at DESC
LIMIT ?
""",
(limit,),
).fetchall()
def text_preview(value, limit=180):
text = " ".join((value or "").split())
if len(text) <= limit:
return text
return text[: limit - 3].rstrip() + "..."
def parse_activity_time(value):
raw = (value or "").strip()
if not raw:
return dt.datetime.min.replace(tzinfo=dt.UTC)
candidates = [raw]
if raw.endswith("Z"):
candidates.append(raw[:-1] + "+00:00")
for candidate in candidates:
try:
parsed = dt.datetime.fromisoformat(candidate)
if parsed.tzinfo is None:
parsed = parsed.replace(tzinfo=dt.UTC)
return parsed.astimezone(dt.UTC)
except ValueError:
pass
for date_format in ("%Y-%m-%d %H:%M:%S %z", "%Y-%m-%d %H:%M %z"):
try:
return dt.datetime.strptime(raw, date_format).astimezone(dt.UTC)
except ValueError:
pass
return dt.datetime.min.replace(tzinfo=dt.UTC)
def activity_sort_key(action):
return (
parse_activity_time(action["occurred_at"]),
action["kind"],
action["target_url"],
)
def normalize_activity_action(row):
action = dict(row)
actor_username = action.get("actor_username") or ""
action["actor_label"] = f"@{actor_username}" if actor_username else ""
action["actor_url"] = f"/{actor_username}" if actor_username else ""
action["detail"] = text_preview(action["detail"])
return action
def list_activity_repositories():
with db_connect() as conn:
return conn.execute(
"""
SELECT repositories.*, users.username AS owner_username
FROM repositories
JOIN users ON users.id = repositories.owner_id
ORDER BY repositories.created_at DESC
"""
).fetchall()
def list_commit_activity_actions(limit):
actions = []
for repo in list_activity_repositories():
path = repo_path(repo["owner_username"], repo["name"])
try:
commits = commit_log(path, limit=limit)
except (HgCommandError, OSError):
continue
for commit in commits:
actions.append(
{
"kind": "commit_created",
"occurred_at": commit["date"],
"actor_username": "",
"actor_label": commit["author"],
"actor_url": "",
"repo_owner_username": repo["owner_username"],
"repo_name": repo["name"],
"target_url": f"/{repo['owner_username']}/{repo['name']}/commits/{commit['node']}",
"target_label": commit["short_node"],
"summary": "committed",
"detail": text_preview(commit["summary"]),
}
)
return actions
def list_recent_actions(limit=50):
limit = max(1, min(int(limit), 100))
with db_connect() as conn:
rows = conn.execute(
"""
SELECT *
FROM (
SELECT
CASE
WHEN repositories.forked_from_repo_id IS NULL THEN 'repo_created'
ELSE 'repo_forked'
END AS kind,
repositories.created_at AS occurred_at,
owner.username AS actor_username,
owner.username AS repo_owner_username,
repositories.name AS repo_name,
'/' || owner.username || '/' || repositories.name AS target_url,
owner.username || '/' || repositories.name AS target_label,
CASE
WHEN repositories.forked_from_repo_id IS NULL THEN 'created repository'
ELSE 'forked repository'
END AS summary,
repositories.description AS detail
FROM repositories
JOIN users AS owner ON owner.id = repositories.owner_id
UNION ALL
SELECT
'repo_starred' AS kind,
repo_stars.created_at AS occurred_at,
actor.username AS actor_username,
owner.username AS repo_owner_username,
repositories.name AS repo_name,
'/' || owner.username || '/' || repositories.name AS target_url,
owner.username || '/' || repositories.name AS target_label,
'starred repository' AS summary,
'' AS detail
FROM repo_stars
JOIN users AS actor ON actor.id = repo_stars.user_id
JOIN repositories ON repositories.id = repo_stars.repo_id
JOIN users AS owner ON owner.id = repositories.owner_id
UNION ALL
SELECT
'contributor_added' AS kind,
repo_contributors.created_at AS occurred_at,
actor.username AS actor_username,
owner.username AS repo_owner_username,
repositories.name AS repo_name,
'/' || owner.username || '/' || repositories.name AS target_url,
owner.username || '/' || repositories.name AS target_label,
'added contributor' AS summary,
contributor.username AS detail
FROM repo_contributors
JOIN users AS actor ON actor.id = repo_contributors.added_by_id
JOIN users AS contributor ON contributor.id = repo_contributors.user_id
JOIN repositories ON repositories.id = repo_contributors.repo_id
JOIN users AS owner ON owner.id = repositories.owner_id
UNION ALL
SELECT
'issue_opened' AS kind,
issues.created_at AS occurred_at,
actor.username AS actor_username,
owner.username AS repo_owner_username,
repositories.name AS repo_name,
'/' || owner.username || '/' || repositories.name || '/issues/' || issues.number AS target_url,
'#' || issues.number || ' ' || issues.title AS target_label,
'opened issue' AS summary,
issues.body AS detail
FROM issues
JOIN users AS actor ON actor.id = issues.author_id
JOIN repositories ON repositories.id = issues.repo_id
JOIN users AS owner ON owner.id = repositories.owner_id
UNION ALL
SELECT
'issue_commented' AS kind,
issue_comments.created_at AS occurred_at,
actor.username AS actor_username,
owner.username AS repo_owner_username,
repositories.name AS repo_name,
'/' || owner.username || '/' || repositories.name || '/issues/' || issues.number AS target_url,
'#' || issues.number || ' ' || issues.title AS target_label,
'commented on issue' AS summary,
issue_comments.body AS detail
FROM issue_comments
JOIN users AS actor ON actor.id = issue_comments.author_id
JOIN issues ON issues.id = issue_comments.issue_id
JOIN repositories ON repositories.id = issues.repo_id
JOIN users AS owner ON owner.id = repositories.owner_id
UNION ALL
SELECT
'pull_request_opened' AS kind,
pull_requests.created_at AS occurred_at,
actor.username AS actor_username,
owner.username AS repo_owner_username,
repositories.name AS repo_name,
'/' || owner.username || '/' || repositories.name || '/pulls/' || pull_requests.number AS target_url,
'#' || pull_requests.number || ' ' || pull_requests.title AS target_label,
'opened pull request' AS summary,
pull_requests.body AS detail
FROM pull_requests
JOIN users AS actor ON actor.id = pull_requests.author_id
JOIN repositories ON repositories.id = pull_requests.target_repo_id
JOIN users AS owner ON owner.id = repositories.owner_id
UNION ALL
SELECT
'pull_request_commented' AS kind,
pull_request_comments.created_at AS occurred_at,
actor.username AS actor_username,
owner.username AS repo_owner_username,
repositories.name AS repo_name,
'/' || owner.username || '/' || repositories.name || '/pulls/' || pull_requests.number AS target_url,
'#' || pull_requests.number || ' ' || pull_requests.title AS target_label,
'commented on pull request' AS summary,
pull_request_comments.body AS detail
FROM pull_request_comments
JOIN users AS actor ON actor.id = pull_request_comments.author_id
JOIN pull_requests ON pull_requests.id = pull_request_comments.pull_request_id
JOIN repositories ON repositories.id = pull_requests.target_repo_id
JOIN users AS owner ON owner.id = repositories.owner_id
UNION ALL
SELECT
'pull_request_merged' AS kind,
pull_requests.merged_at AS occurred_at,
actor.username AS actor_username,
owner.username AS repo_owner_username,
repositories.name AS repo_name,
'/' || owner.username || '/' || repositories.name || '/pulls/' || pull_requests.number AS target_url,
'#' || pull_requests.number || ' ' || pull_requests.title AS target_label,
'merged pull request' AS summary,
pull_requests.merge_node AS detail
FROM pull_requests
JOIN users AS actor ON actor.id = pull_requests.merged_by_id
JOIN repositories ON repositories.id = pull_requests.target_repo_id
JOIN users AS owner ON owner.id = repositories.owner_id
WHERE pull_requests.merged_at IS NOT NULL
UNION ALL
SELECT
'commit_commented' AS kind,
commit_comments.created_at AS occurred_at,
actor.username AS actor_username,
owner.username AS repo_owner_username,
repositories.name AS repo_name,
'/' || owner.username || '/' || repositories.name || '/commits/' || commit_comments.commit_node AS target_url,
substr(commit_comments.commit_node, 1, 12) AS target_label,
'commented on commit' AS summary,
commit_comments.body AS detail
FROM commit_comments
JOIN users AS actor ON actor.id = commit_comments.author_id
JOIN repositories ON repositories.id = commit_comments.repo_id
JOIN users AS owner ON owner.id = repositories.owner_id
UNION ALL
SELECT
'user_joined' AS kind,
users.created_at AS occurred_at,
users.username AS actor_username,
'' AS repo_owner_username,
'' AS repo_name,
'/' || users.username AS target_url,
'@' || users.username AS target_label,
'joined' AS summary,
'' AS detail
FROM users
)
WHERE occurred_at IS NOT NULL
ORDER BY occurred_at DESC
LIMIT ?
""",
(limit,),
).fetchall()
actions = [normalize_activity_action(row) for row in rows]
actions.extend(list_commit_activity_actions(limit))
actions.sort(key=activity_sort_key, reverse=True)
return actions[:limit]
def list_owned_repos(owner_id):
with db_connect() as conn:
return conn.execute(
"""
SELECT repositories.*, users.username AS owner_username, COUNT(repo_stars.user_id) AS star_count
FROM repositories
JOIN users ON users.id = repositories.owner_id
LEFT JOIN repo_stars ON repo_stars.repo_id = repositories.id
WHERE repositories.owner_id = ?
GROUP BY repositories.id
ORDER BY repositories.updated_at DESC
""",
(owner_id,),
).fetchall()
def list_starred_repos(user_id):
with db_connect() as conn:
return conn.execute(
"""
SELECT
repositories.*,
users.username AS owner_username,
repo_stars.created_at AS starred_at,
(
SELECT COUNT(*)
FROM repo_stars AS counts
WHERE counts.repo_id = repositories.id
) AS star_count
FROM repo_stars
JOIN repositories ON repositories.id = repo_stars.repo_id
JOIN users ON users.id = repositories.owner_id
WHERE repo_stars.user_id = ?
ORDER BY repo_stars.created_at DESC
""",
(user_id,),
).fetchall()
def list_user_forks_for_target(user_id, target_repo_id):
with db_connect() as conn:
return conn.execute(
"""
SELECT repositories.*, users.username AS owner_username
FROM repositories
JOIN users ON users.id = repositories.owner_id
WHERE repositories.owner_id = ?
AND repositories.forked_from_repo_id = ?
ORDER BY repositories.updated_at DESC
""",
(user_id, target_repo_id),
).fetchall()
def user_has_fork_for_target(user_id, target_repo_id):
with db_connect() as conn:
row = conn.execute(
"""
SELECT 1
FROM repositories
WHERE owner_id = ?
AND forked_from_repo_id = ?
LIMIT 1
""",
(user_id, target_repo_id),
).fetchone()
return bool(row)
def list_repo_contributors(repo_id):
with db_connect() as conn:
return conn.execute(
"""
SELECT users.*, repo_contributors.created_at AS contributor_since
FROM repo_contributors
JOIN users ON users.id = repo_contributors.user_id
WHERE repo_contributors.repo_id = ?
ORDER BY users.username
""",
(repo_id,),
).fetchall()
def repo_contributor_usernames(repo_id):
return [row["username"] for row in list_repo_contributors(repo_id)]
def repo_star_count(repo_id):
with db_connect() as conn:
return conn.execute(
"SELECT COUNT(*) FROM repo_stars WHERE repo_id = ?",
(repo_id,),
).fetchone()[0]
def user_starred_repo(user, repo):
if not user or not repo:
return False
with db_connect() as conn:
row = conn.execute(
"SELECT 1 FROM repo_stars WHERE repo_id = ? AND user_id = ?",
(repo["id"], user["id"]),
).fetchone()
return bool(row)
def star_repo(user, repo):
with db_connect() as conn:
conn.execute(
"""
INSERT OR IGNORE INTO repo_stars (repo_id, user_id, created_at)
VALUES (?, ?, ?)
""",
(repo["id"], user["id"], utcnow()),
)
def unstar_repo(user, repo):
with db_connect() as conn:
conn.execute(
"DELETE FROM repo_stars WHERE repo_id = ? AND user_id = ?",
(repo["id"], user["id"]),
)
def normalize_website(value):
website = (value or "").strip()
if not website:
return ""
if any(char.isspace() for char in website):
raise ValueError("Website must be a valid http(s) URL.")
if "://" not in website:
website = "https://" + website
parsed = urlparse(website)
if parsed.scheme not in {"http", "https"} or not parsed.netloc:
raise ValueError("Website must be a valid http(s) URL.")
return website[:255]
def profile_form_values(form, fallback=None):
fallback = dict(fallback) if fallback else {}
return {
"username": fallback.get("username", ""),
"display_name": (form.get("display_name", fallback.get("display_name", "")) or "").strip()[:80],
"bio": (form.get("bio", fallback.get("bio", "")) or "").strip()[:1000],
"website": (form.get("website", fallback.get("website", "")) or "").strip(),
"created_at": fallback.get("created_at", ""),
}
def write_hgrc(path, owner_username, repo_name, description, allow_push_usernames=None):
hgrc = path / ".hg" / "hgrc"
safe_description = " ".join((description or "").splitlines()).strip()
allowed = allow_push_usernames or [owner_username]
hgrc.write_text(
"\n".join(
[
"[web]",
f"name = {owner_username}/{repo_name}",
f"description = {safe_description}",
f"contact = {owner_username}",
f"allow-push = {' '.join(allowed)}",
"push_ssl = false",
"allow_archive = gz, zip, bz2",
"",
]
),
encoding="utf-8",
)
def sync_repo_hgrc(repo):
path = repo_path(repo["owner_username"], repo["name"])
allowed = [repo["owner_username"], *repo_contributor_usernames(repo["id"])]
write_hgrc(path, repo["owner_username"], repo["name"], repo["description"], allowed)
def hg_env():
env = os.environ.copy()
env["HGPLAIN"] = "1"
return env
def run_hg(args, cwd=None, timeout=15, check=True, text=True):
completed = subprocess.run(
["hg", *args],
cwd=cwd,
capture_output=True,
text=text,
encoding="utf-8" if text else None,
errors="replace" if text else None,
timeout=timeout,
env=hg_env(),
)
if check and completed.returncode != 0:
stderr = completed.stderr if text else completed.stderr.decode("utf-8", "replace")
raise HgCommandError(stderr.strip() or "Mercurial command failed.", completed.returncode)
return completed
def create_repository(owner, name, description):
now = utcnow()
path = repo_path(owner["username"], name)
if path.exists():
raise ValueError("Repository directory already exists.")
path.parent.mkdir(parents=True, exist_ok=True)
with db_connect() as conn:
try:
conn.execute(
"""
INSERT INTO repositories (owner_id, name, description, created_at, updated_at)
VALUES (?, ?, ?, ?, ?)
""",
(owner["id"], name, description, now, now),
)
except sqlite3.IntegrityError as exc:
raise ValueError("Repository already exists.") from exc
try:
run_hg(["init", str(path)], timeout=20)
write_hgrc(path, owner["username"], name, description)
except Exception:
with db_connect() as conn:
conn.execute(
"DELETE FROM repositories WHERE owner_id = ? AND name = ?",
(owner["id"], name),
)
if path.exists():
shutil.rmtree(path)
raise
def fork_repository(owner, source_repo, name, description):
now = utcnow()
source_path = repo_path(source_repo["owner_username"], source_repo["name"])
path = repo_path(owner["username"], name)
if path.exists():
raise ValueError("Repository directory already exists.")
path.parent.mkdir(parents=True, exist_ok=True)
upstream_repo_id = source_repo["forked_from_repo_id"] or source_repo["id"]
forked_from_node = (
source_repo["forked_from_node"]
if source_repo["forked_from_repo_id"] and source_repo["forked_from_node"]
else default_code_ref(source_path).get("node") or NULL_REV
)
with db_connect() as conn:
try:
conn.execute(
"""
INSERT INTO repositories (
owner_id, name, description, forked_from_repo_id,
forked_at, forked_from_node, created_at, updated_at
)
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
""",
(
owner["id"],
name,
description,
upstream_repo_id,
now,
forked_from_node,
now,
now,
),
)
except sqlite3.IntegrityError as exc:
raise ValueError("Repository already exists.") from exc
try:
run_hg(["clone", str(source_path), str(path)], timeout=60)
write_hgrc(path, owner["username"], name, description)
except Exception:
with db_connect() as conn:
conn.execute(
"DELETE FROM repositories WHERE owner_id = ? AND name = ?",
(owner["id"], name),
)
if path.exists():
shutil.rmtree(path)
raise
def delete_repository(repo, path):
with db_connect() as conn:
conn.execute("DELETE FROM repositories WHERE id = ?", (repo["id"],))
if path.exists():
shutil.rmtree(path)
def add_repo_contributor(repo, added_by, username):
username = (username or "").strip().lower()
contributor = get_user_by_username(username)
if not contributor:
raise ValueError("User not found.")
if contributor["id"] == repo["owner_id"]:
raise ValueError("The owner already has access.")
now = utcnow()
with db_connect() as conn:
try:
conn.execute(
"""
INSERT INTO repo_contributors (repo_id, user_id, added_by_id, created_at)
VALUES (?, ?, ?, ?)
""",
(repo["id"], contributor["id"], added_by["id"], now),
)
except sqlite3.IntegrityError as exc:
raise ValueError("User is already a contributor.") from exc
sync_repo_hgrc(repo)
def remove_repo_contributor(repo, user_id):
with db_connect() as conn:
conn.execute(
"DELETE FROM repo_contributors WHERE repo_id = ? AND user_id = ?",
(repo["id"], user_id),
)
sync_repo_hgrc(repo)
def hg_files(path, revision="tip"):
if not revision:
return []
completed = run_hg(["files", "-r", revision], cwd=path, check=False)
if completed.returncode != 0:
stderr = (completed.stderr or "").lower()
stdout = completed.stdout or ""
if (
not stderr and not stdout
) or "unknown revision" in stderr or "empty" in stderr or "repo has no changesets" in stderr:
return []
raise HgCommandError(completed.stderr.strip() or "Unable to list files.", completed.returncode)
return [line.strip() for line in completed.stdout.splitlines() if line.strip()]
def hg_cat(path, file_path, revision="tip", text=True):
completed = run_hg(["cat", "-r", revision, file_path], cwd=path, check=True, text=text)
return completed.stdout if text else completed.stdout
def truncate_bytes_for_render(content):
if not MAX_RENDER_BYTES or len(content) <= MAX_RENDER_BYTES:
return content, False
return content[:MAX_RENDER_BYTES], True
def truncate_text_for_render(content, label="Preview"):
encoded = (content or "").encode("utf-8", "replace")
truncated, was_truncated = truncate_bytes_for_render(encoded)
text = truncated.decode("utf-8", "replace")
if was_truncated:
text = text.rstrip() + f"\n\n[{label} truncated. Use the raw view or local clone for the full content.]\n"
return text, was_truncated
def read_file_bytes(path, file_path, revision="tip"):
if not revision:
raise HgCommandError("File not found.")
completed = subprocess.run(
["hg", "cat", "-r", revision, file_path],
cwd=path,
capture_output=True,
timeout=15,
env=hg_env(),
)
if completed.returncode != 0:
raise HgCommandError(completed.stderr.decode("utf-8", "replace").strip() or "Unable to read file.")
return completed.stdout
def build_tree(files, subpath):
prefix = f"{subpath}/" if subpath else ""
entries = {}
for file_path in files:
if prefix and not file_path.startswith(prefix):
continue
rest = file_path[len(prefix) :]
if not rest:
continue
name, _, _remaining = rest.partition("/")
full_path = f"{prefix}{name}" if prefix else name
entries[name] = {
"name": name,
"path": full_path,
"type": "dir" if "/" in rest else "file",
}
return sorted(entries.values(), key=lambda item: (item["type"] != "dir", item["name"].lower()))
def readme_for_repo(path, files, revision="tip"):
by_lower = {file_path.lower(): file_path for file_path in files}
for candidate in README_CANDIDATES:
actual = by_lower.get(candidate.lower())
if actual:
try:
return actual, hg_cat(path, actual, revision=revision)
except HgCommandError:
return actual, ""
return None, None
def readme_preview_for_repo(path, files, revision="tip"):
name, readme = readme_for_repo(path, files, revision=revision)
if readme is None:
return name, readme, False
readme, truncated = truncate_text_for_render(readme, label="README preview")
return name, readme, truncated
def is_markdown_file(file_path):
return bool(file_path and file_path.lower().endswith((".md", ".markdown", ".mdown")))
def highlight_language_class(file_path):
path = Path(file_path)
by_name = HIGHLIGHT_LANGUAGE_BY_NAME.get(path.name.lower())
if by_name:
return by_name
return HIGHLIGHT_LANGUAGE_BY_EXTENSION.get(path.suffix.lower(), "")
def render_markdown(text):
text = SCRIPT_STYLE_RE.sub("", text or "")
rendered = markdown.markdown(
text,
extensions=MARKDOWN_EXTENSIONS,
output_format="html5",
)
return bleach.clean(
rendered,
tags=MARKDOWN_TAGS,
attributes=MARKDOWN_ATTRIBUTES,
protocols={"http", "https", "mailto"},
strip=True,
)
def render_markdown_links(text):
rendered = markdown.markdown(
html.escape(text or "", quote=False),
output_format="html5",
)
return bleach.clean(
rendered,
tags=MARKDOWN_LINK_TAGS,
attributes=MARKDOWN_LINK_ATTRIBUTES,
protocols={"http", "https", "mailto"},
strip=True,
).strip()
def render_repo_description(text):
return render_markdown_links(text)
def is_null_revision(value):
return (value or "").strip().lower() in {NULL_REV, NULL_NODE}
def ancestors_revset(node):
revision = validate_revision_id(node, allow_null=False)
return f"sort(ancestors({revision}), -rev)"
def newest_revision_sort_key(item):
try:
revision = int(item["rev"])
except (KeyError, TypeError, ValueError):
revision = -1
return (revision, item.get("date", ""), item.get("name", ""))
def commit_log(path, limit=50, revision=None):
if is_null_revision(revision):
return []
if revision is None and repo_tip_node(path) is None:
return []
template_arg = "{rev}\\x1f{node}\\x1f{node|short}\\x1f{author|person}\\x1f{date|isodate}\\x1f{desc|firstline}\\x1e"
args = ["log", "-l", str(limit), "--template", template_arg]
if revision:
args[1:1] = ["-r", ancestors_revset(revision)]
completed = run_hg(args, cwd=path, check=False)
if completed.returncode != 0:
stderr = completed.stderr.lower()
if "empty" in stderr or "no changes found" in stderr:
return []
raise HgCommandError(completed.stderr.strip() or "Unable to read commit log.", completed.returncode)
commits = []
for record in completed.stdout.split("\x1e"):
if not record:
continue
parts = record.split("\x1f")
if len(parts) != 6:
continue
commits.append(
{
"rev": parts[0],
"node": parts[1],
"short_node": parts[2],
"author": parts[3],
"date": parts[4],
"summary": parts[5],
}
)
return commits
def list_repo_tags(path):
template_arg = (
"{tag}\\x1f{rev}\\x1f{node}\\x1f{node|short}\\x1f{date|isodate}\\x1f"
"{desc|firstline}\\x1f{type}\\x1e"
)
completed = run_hg(["tags", "--template", template_arg], cwd=path, check=False)
if completed.returncode != 0:
stderr = (completed.stderr or "").lower()
if "empty" in stderr or "no changes found" in stderr or "repo has no changesets" in stderr:
return []
raise HgCommandError(completed.stderr.strip() or "Unable to read repository tags.", completed.returncode)
tags = []
for record in completed.stdout.split("\x1e"):
if not record:
continue
parts = record.split("\x1f")
if len(parts) != 7 or parts[0] == "tip":
continue
tags.append(
{
"type": REF_TYPE_TAG,
"name": parts[0],
"label": f"tag {parts[0]}",
"shell_name": shlex.quote(parts[0]),
"rev": parts[1],
"node": parts[2],
"short_node": parts[3],
"branch": "",
"active": False,
"closed": False,
"local": parts[6] == "local",
"is_default": False,
"date": parts[4],
"summary": parts[5],
}
)
tags.sort(key=newest_revision_sort_key, reverse=True)
return tags
def hg_bool(value):
return str(value).strip().lower() in {"1", "true", "yes", "on"}
def revision_info(path, revision):
if not revision:
return None
template_arg = (
"{rev}\\x1f{node}\\x1f{node|short}\\x1f{branch}\\x1f{date|isodate}\\x1f{desc|firstline}\\x1e"
)
completed = run_hg(["log", "-r", revision, "--template", template_arg], cwd=path, check=False)
if completed.returncode != 0 or not completed.stdout:
return None
parts = completed.stdout.rstrip("\x1e").split("\x1f")
if len(parts) != 6:
return None
if parts[0] == "-1" or is_null_revision(parts[1]):
return None
return {
"rev": parts[0],
"node": parts[1],
"short_node": parts[2],
"branch": parts[3],
"date": parts[4],
"summary": parts[5],
}
def empty_tip_ref(is_default=False):
return {
"type": REF_TYPE_TIP,
"name": "",
"label": "tip",
"rev": "",
"node": None,
"short_node": "",
"branch": "",
"date": "",
"summary": "",
"active": False,
"closed": False,
"is_default": is_default,
}
def tip_ref(path, is_default=False):
info = revision_info(path, "tip")
if not info:
return empty_tip_ref(is_default=is_default)
info.update(
{
"type": REF_TYPE_TIP,
"name": "",
"label": "tip",
"active": False,
"closed": False,
"is_default": is_default,
}
)
return info
def commit_ref(path, revision):
revision = (revision or "").strip()
if not REV_RE.match(revision) or revision == NULL_REV:
raise ValueError("Commit not found.")
info = revision_info(path, revision)
if not info:
raise ValueError("Commit not found.")
info.update(
{
"type": REF_TYPE_COMMIT,
"name": info["node"],
"label": f"commit {info['short_node']}",
"active": False,
"closed": False,
"is_default": False,
}
)
return info
def list_repo_branches(path):
template_arg = (
"{branch}\\x1f{node}\\x1f{node|short}\\x1f{rev}\\x1f{active}\\x1f{closed}\\x1f"
"{date|isodate}\\x1f{desc|firstline}\\x1e"
)
completed = run_hg(["branches", "-c", "--template", template_arg], cwd=path, check=False)
if completed.returncode != 0:
stderr = (completed.stderr or "").lower()
if "empty" in stderr or "no changes found" in stderr or "repo has no changesets" in stderr:
return []
raise HgCommandError(completed.stderr.strip() or "Unable to read repository branches.", completed.returncode)
branches = []
for record in completed.stdout.split("\x1e"):
if not record:
continue
parts = record.split("\x1f")
if len(parts) != 8:
continue
branches.append(
{
"type": REF_TYPE_BRANCH,
"name": parts[0],
"label": f"branch {parts[0]}",
"node": parts[1],
"short_node": parts[2],
"rev": parts[3],
"active": hg_bool(parts[4]),
"closed": hg_bool(parts[5]),
"date": parts[6],
"summary": parts[7],
"is_default": False,
}
)
branches.sort(key=newest_revision_sort_key, reverse=True)
return branches
def list_repo_bookmarks(path):
template_arg = (
"{bookmark}\\x1f{node}\\x1f{node|short}\\x1f{rev}\\x1f{active}\\x1f"
"{date|isodate}\\x1f{desc|firstline}\\x1e"
)
completed = run_hg(["bookmarks", "--template", template_arg], cwd=path, check=False)
if completed.returncode != 0:
stderr = (completed.stderr or "").lower()
if "empty" in stderr or "no changes found" in stderr or "repo has no changesets" in stderr:
return []
raise HgCommandError(completed.stderr.strip() or "Unable to read repository bookmarks.", completed.returncode)
bookmarks = []
for record in completed.stdout.split("\x1e"):
if not record:
continue
parts = record.split("\x1f")
if len(parts) != 7:
continue
bookmarks.append(
{
"type": REF_TYPE_BOOKMARK,
"name": parts[0],
"label": f"bookmark {parts[0]}",
"node": parts[1],
"short_node": parts[2],
"rev": parts[3],
"active": hg_bool(parts[4]),
"closed": False,
"date": parts[5],
"summary": parts[6],
"is_default": False,
}
)
bookmarks.sort(key=newest_revision_sort_key, reverse=True)
return bookmarks
def default_code_ref(path):
for branch in list_repo_branches(path):
if branch["name"] == "default" and not branch["closed"]:
selected = dict(branch)
selected["is_default"] = True
return selected
return tip_ref(path, is_default=True)
def resolve_repo_ref(path, ref_type, ref_name=""):
ref_type = (ref_type or "").strip().lower()
ref_name = ref_name or ""
if ref_type == REF_TYPE_TIP:
return tip_ref(path)
if ref_type == REF_TYPE_COMMIT:
return commit_ref(path, ref_name)
if ref_type == REF_TYPE_BRANCH:
for branch in list_repo_branches(path):
if branch["name"] == ref_name:
return dict(branch)
raise ValueError("Branch not found.")
if ref_type == REF_TYPE_BOOKMARK:
for bookmark in list_repo_bookmarks(path):
if bookmark["name"] == ref_name:
return dict(bookmark)
raise ValueError("Bookmark not found.")
if ref_type == REF_TYPE_TAG:
for tag in list_repo_tags(path):
if tag["name"] == ref_name:
return dict(tag)
raise ValueError("Tag not found.")
raise ValueError("Ref not found.")
def ref_option_value(ref_type, ref_name=""):
return REF_VALUE_SEPARATOR.join((ref_type, quote(ref_name or "", safe="")))
def source_ref_option_value(repo_id, ref_type, ref_name=""):
return REF_VALUE_SEPARATOR.join((str(repo_id), ref_type, quote(ref_name or "", safe="")))
def parse_ref_option_value(value, allowed_types=REF_TYPES):
parts = (value or "").split(REF_VALUE_SEPARATOR, 1)
if len(parts) != 2 or parts[0] not in allowed_types:
raise ValueError("Invalid ref.")
return parts[0], unquote(parts[1])
def parse_source_ref_option_value(value):
parts = (value or "").split(REF_VALUE_SEPARATOR, 2)
if len(parts) != 3 or parts[1] not in PULL_REQUEST_REF_TYPES:
raise ValueError("Invalid source ref.")
try:
repo_id = int(parts[0])
except ValueError as exc:
raise ValueError("Invalid source repository.") from exc
return repo_id, parts[1], unquote(parts[2])
def selected_repo_ref(path):
ref_value = request.query.get("ref_value")
if ref_value:
try:
ref_type, ref_name = parse_ref_option_value(ref_value)
return resolve_repo_ref(path, ref_type, ref_name)
except ValueError as exc:
abort(404, str(exc))
ref_type = (request.query.get("ref_type") or "").strip().lower()
if not ref_type:
return default_code_ref(path)
if ref_type not in REF_TYPES:
abort(404, "Ref not found.")
try:
return resolve_repo_ref(path, ref_type, request.query.get("ref") or "")
except ValueError as exc:
abort(404, str(exc))
def ref_query_string(ref_info, force=False):
if not ref_info or (ref_info.get("is_default") and not force):
return ""
ref_type = ref_info.get("type") or REF_TYPE_TIP
params = {"ref_type": ref_type}
if ref_type != REF_TYPE_TIP:
params["ref"] = ref_info.get("name", "")
return urlencode(params)
def url_with_ref(url, ref_info=None, force=False):
query = ref_query_string(ref_info, force=force)
if not query:
return url
separator = "&" if "?" in url else "?"
return f"{url}{separator}{query}"
def current_url_with_ref(ref_info=None, force=False):
params = [
(key, value)
for key, value in request.query.allitems()
if key not in REF_QUERY_KEYS
]
query = ref_query_string(ref_info, force=force)
if query:
params.extend(parse_qsl(query, keep_blank_values=True))
encoded = urlencode(params)
return request.path + (f"?{encoded}" if encoded else "")
def format_ref_label(ref_type, ref_name=""):
ref_type = (ref_type or REF_TYPE_TIP).strip().lower()
if ref_type == REF_TYPE_BRANCH:
return f"branch {ref_name}"
if ref_type == REF_TYPE_BOOKMARK:
return f"bookmark {ref_name}"
if ref_type == REF_TYPE_TAG:
return f"tag {ref_name}"
if ref_type == REF_TYPE_COMMIT:
return f"commit {ref_name[:12]}" if ref_name else "commit"
return "tip"
def ref_option_label(ref):
label = format_ref_label(ref["type"], ref.get("name", ""))
if ref["type"] == REF_TYPE_BRANCH and ref.get("closed"):
label += " (closed)"
if ref["type"] == REF_TYPE_TAG and ref.get("local"):
label += " (local)"
return label
def ref_option_from_ref(ref, repo_id=None):
if repo_id is None:
value = ref_option_value(ref["type"], ref.get("name", ""))
else:
value = source_ref_option_value(repo_id, ref["type"], ref.get("name", ""))
return {
"value": value,
"label": ref_option_label(ref),
"ref": ref,
}
def repo_ref_options(path, include_closed_branches=True, include_tip=True, include_tags=True):
branches = list_repo_branches(path)
bookmarks = list_repo_bookmarks(path)
refs = []
for branch in branches:
if include_closed_branches or not branch["closed"]:
refs.append(branch)
refs.extend(bookmarks)
if include_tags:
refs.extend(list_repo_tags(path))
refs.sort(key=newest_revision_sort_key, reverse=True)
options = []
for index, ref in enumerate(refs):
option = ref_option_from_ref(ref)
option["is_initial"] = index < 10
options.append(option)
if include_tip:
tip_option = ref_option_from_ref(tip_ref(path))
tip_option["is_initial"] = False
options.append(tip_option)
return options
def source_repo_ref_options(source_repo):
path = repo_path(source_repo["owner_username"], source_repo["name"])
options = repo_ref_options(path, include_closed_branches=True, include_tip=True, include_tags=False)
for option in options:
option["value"] = source_ref_option_value(
source_repo["id"],
option["ref"]["type"],
option["ref"].get("name", ""),
)
option["label"] = f"{source_repo['owner_username']}/{source_repo['name']} {option['label']}"
return options
def target_repo_ref_options(path):
return repo_ref_options(path, include_closed_branches=False, include_tip=False, include_tags=False)
def revision_branch(path, node):
info = revision_info(path, node)
return info["branch"] if info else ""
def commit_count(path, revision=None):
if is_null_revision(revision):
return 0
if revision is None and repo_tip_node(path) is None:
return 0
args = ["log", "--template", "."]
if revision:
args[1:1] = ["-r", ancestors_revset(revision)]
completed = run_hg(args, cwd=path, check=False)
if completed.returncode != 0:
stderr = (completed.stderr or "").lower()
if "empty" in stderr or "no changes found" in stderr:
return 0
raise HgCommandError(completed.stderr.strip() or "Unable to count commits.", completed.returncode)
return len(completed.stdout)
def validate_revision_id(value, allow_null=True):
revision = (value or "").strip()
if not REV_RE.match(revision) or (is_null_revision(revision) and not allow_null):
abort(404, "Revision not found.")
return revision
def repo_tip_node(path):
completed = run_hg(["log", "-r", "tip", "--template", "{node}"], cwd=path, check=False)
if completed.returncode != 0:
stderr = (completed.stderr or "").lower()
if "empty" in stderr or "no changes found" in stderr:
return None
raise HgCommandError(completed.stderr.strip() or "Unable to read repository tip.", completed.returncode)
node = completed.stdout.strip()
if is_null_revision(node):
return None
return node or None
def repo_has_revision(path, revision):
revision = validate_revision_id(revision)
if is_null_revision(revision):
return True
completed = run_hg(["log", "-r", revision, "--template", "{node}"], cwd=path, check=False)
return completed.returncode == 0 and bool(completed.stdout.strip())
def is_ancestor(path, ancestor_node, descendant_node):
ancestor_node = validate_revision_id(ancestor_node)
descendant_node = validate_revision_id(descendant_node, allow_null=False)
if is_null_revision(ancestor_node) or ancestor_node == descendant_node:
return True
completed = run_hg(
["log", "-r", f"descendants({ancestor_node}) and {descendant_node}", "--template", "{node}"],
cwd=path,
check=False,
)
return completed.returncode == 0 and bool(completed.stdout.strip())
def commit_detail(path, node):
node = validate_revision_id(node, allow_null=False)
template_arg = "{rev}\\x1f{node}\\x1f{node|short}\\x1f{author|person}\\x1f{date|isodate}\\x1f{desc}\\x1f{parents}\\x1e"
completed = run_hg(["log", "-r", node, "--template", template_arg], cwd=path, check=False)
if completed.returncode != 0 or not completed.stdout:
abort(404, "Commit not found.")
parts = completed.stdout.rstrip("\x1e").split("\x1f")
if len(parts) != 7:
abort(404, "Commit not found.")
return {
"rev": parts[0],
"node": parts[1],
"short_node": parts[2],
"author": parts[3],
"date": parts[4],
"description": parts[5],
"parents": parts[6],
}
def commit_diff(path, node):
node = validate_revision_id(node, allow_null=False)
completed = run_hg(["diff", "--git", "-c", node], cwd=path, check=False)
if completed.returncode != 0:
raise HgCommandError(completed.stderr.strip() or "Unable to read commit diff.", completed.returncode)
return truncate_text_for_render(completed.stdout, label="Diff")[0]
def diff_between_revisions(path, base_node, source_node):
base_node = validate_revision_id(base_node)
source_node = validate_revision_id(source_node, allow_null=False)
completed = run_hg(["diff", "--git", "--from", base_node, "--to", source_node], cwd=path, check=False)
if completed.returncode != 0:
raise HgCommandError(completed.stderr.strip() or "Unable to read diff.", completed.returncode)
return truncate_text_for_render(completed.stdout, label="Diff")[0]
def ensure_clean_working_copy(path):
completed = run_hg(["status"], cwd=path, check=False)
if completed.returncode != 0:
raise HgCommandError(completed.stderr.strip() or "Unable to inspect working copy.", completed.returncode)
if completed.stdout.strip():
raise HgCommandError("Target repository has uncommitted working copy changes.")
def issue_counts(repo_id):
with db_connect() as conn:
rows = conn.execute(
"SELECT status, COUNT(*) AS count FROM issues WHERE repo_id = ? GROUP BY status",
(repo_id,),
).fetchall()
counts = {"open": 0, "closed": 0}
for row in rows:
counts[row["status"]] = row["count"]
return counts
def pull_request_counts(repo_id):
with db_connect() as conn:
rows = conn.execute(
"SELECT status, COUNT(*) AS count FROM pull_requests WHERE target_repo_id = ? GROUP BY status",
(repo_id,),
).fetchall()
counts = {"open": 0, "closed": 0, "merged": 0}
for row in rows:
counts[row["status"]] = row["count"]
return counts
def list_issues(repo_id, status="open"):
if status not in {"open", "closed", "all"}:
status = "open"
where = "WHERE issues.repo_id = ?"
params = [repo_id]
if status != "all":
where += " AND issues.status = ?"
params.append(status)
with db_connect() as conn:
return conn.execute(
f"""
SELECT issues.*, users.username AS author_username
FROM issues
JOIN users ON users.id = issues.author_id
{where}
ORDER BY issues.updated_at DESC, issues.number DESC
""",
params,
).fetchall()
def get_issue(repo_id, number):
with db_connect() as conn:
return conn.execute(
"""
SELECT issues.*, users.username AS author_username
FROM issues
JOIN users ON users.id = issues.author_id
WHERE issues.repo_id = ? AND issues.number = ?
""",
(repo_id, number),
).fetchone()
def list_issue_comments(issue_id):
with db_connect() as conn:
return conn.execute(
"""
SELECT issue_comments.*, users.username AS author_username
FROM issue_comments
JOIN users ON users.id = issue_comments.author_id
WHERE issue_comments.issue_id = ?
ORDER BY issue_comments.created_at ASC, issue_comments.id ASC
""",
(issue_id,),
).fetchall()
def list_pull_request_comments(pull_request_id):
with db_connect() as conn:
return conn.execute(
"""
SELECT pull_request_comments.*, users.username AS author_username
FROM pull_request_comments
JOIN users ON users.id = pull_request_comments.author_id
WHERE pull_request_comments.pull_request_id = ?
ORDER BY pull_request_comments.created_at ASC, pull_request_comments.id ASC
""",
(pull_request_id,),
).fetchall()
def list_commit_comments(repo_id, commit_node):
with db_connect() as conn:
return conn.execute(
"""
SELECT commit_comments.*, users.username AS author_username
FROM commit_comments
JOIN users ON users.id = commit_comments.author_id
WHERE commit_comments.repo_id = ? AND commit_comments.commit_node = ?
ORDER BY commit_comments.created_at ASC, commit_comments.id ASC
""",
(repo_id, commit_node),
).fetchall()
def pull_request_select_sql(where_clause):
return f"""
SELECT
pull_requests.*,
author.username AS author_username,
source_repo.name AS source_repo_name,
source_owner.username AS source_owner_username,
target_repo.name AS target_repo_name,
target_owner.username AS target_owner_username,
merged_by.username AS merged_by_username
FROM pull_requests
JOIN users AS author ON author.id = pull_requests.author_id
JOIN repositories AS source_repo ON source_repo.id = pull_requests.source_repo_id
JOIN users AS source_owner ON source_owner.id = source_repo.owner_id
JOIN repositories AS target_repo ON target_repo.id = pull_requests.target_repo_id
JOIN users AS target_owner ON target_owner.id = target_repo.owner_id
LEFT JOIN users AS merged_by ON merged_by.id = pull_requests.merged_by_id
{where_clause}
"""
def list_pull_requests(repo_id, status="open"):
if status not in {"open", "closed", "merged", "all"}:
status = "open"
where = "WHERE pull_requests.target_repo_id = ?"
params = [repo_id]
if status != "all":
where += " AND pull_requests.status = ?"
params.append(status)
with db_connect() as conn:
return conn.execute(
pull_request_select_sql(where) + " ORDER BY pull_requests.updated_at DESC, pull_requests.number DESC",
params,
).fetchall()
def get_pull_request(repo_id, number):
with db_connect() as conn:
return conn.execute(
pull_request_select_sql(
"WHERE pull_requests.target_repo_id = ? AND pull_requests.number = ?"
),
(repo_id, number),
).fetchone()
def stored_ref_name(ref):
return "" if ref["type"] == REF_TYPE_TIP else ref.get("name", "")
def pr_ref_type(pr, prefix):
return pr[f"{prefix}_ref_type"] or REF_TYPE_TIP
def pr_ref_name(pr, prefix):
return pr[f"{prefix}_ref_name"] or ""
def resolve_pr_ref(path, pr, prefix):
return resolve_repo_ref(path, pr_ref_type(pr, prefix), pr_ref_name(pr, prefix))
def pull_request_base_node(target_repo, source_repo, target_ref):
source_path = repo_path(source_repo["owner_username"], source_repo["name"])
target_node = target_ref.get("node") or NULL_REV
if repo_has_revision(source_path, target_node):
return target_node
fork_base = source_repo["forked_from_node"] or NULL_REV
if repo_has_revision(source_path, fork_base):
return fork_base
return NULL_REV
def create_pull_request(
target_repo,
source_repo,
author,
title,
body,
source_ref_type,
source_ref_name,
target_ref_type,
target_ref_name,
):
if not source_repo:
raise ValueError("Choose a source fork.")
if source_repo["id"] == target_repo["id"]:
raise ValueError("Choose a fork as the source repository.")
if source_repo["owner_id"] != author["id"] or source_repo["forked_from_repo_id"] != target_repo["id"]:
raise ValueError("Choose one of your forks of this repository.")
target_path = repo_path(target_repo["owner_username"], target_repo["name"])
source_path = repo_path(source_repo["owner_username"], source_repo["name"])
source_ref = resolve_repo_ref(source_path, source_ref_type, source_ref_name)
target_ref = resolve_repo_ref(target_path, target_ref_type, target_ref_name)
if source_ref["type"] not in PULL_REQUEST_REF_TYPES:
raise ValueError("Choose a branch, bookmark, or tip as the source ref.")
if target_ref["type"] not in TARGET_PULL_REQUEST_REF_TYPES:
raise ValueError("Choose an open target branch or bookmark.")
if target_ref["type"] == REF_TYPE_BRANCH and target_ref.get("closed"):
raise ValueError("Choose an open target branch.")
source_node = source_ref.get("node")
if not source_node:
raise ValueError("Source repository has no commits.")
base_node = pull_request_base_node(target_repo, source_repo, target_ref)
now = utcnow()
with db_connect() as conn:
number = conn.execute(
"SELECT COALESCE(MAX(number), 0) + 1 FROM pull_requests WHERE target_repo_id = ?",
(target_repo["id"],),
).fetchone()[0]
conn.execute(
"""
INSERT INTO pull_requests (
target_repo_id, source_repo_id, author_id, number, title, body,
base_node, source_node, target_ref_type, target_ref_name,
source_ref_type, source_ref_name, created_at, updated_at
)
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
""",
(
target_repo["id"],
source_repo["id"],
author["id"],
number,
title[:200],
body[:5000],
base_node,
source_node,
target_ref["type"],
stored_ref_name(target_ref),
source_ref["type"],
stored_ref_name(source_ref),
now,
now,
),
)
return number
def pull_request_diff(pr):
source_repo = get_repo_by_id(pr["source_repo_id"])
if not source_repo:
raise HgCommandError("Source repository no longer exists.")
source_path = repo_path(source_repo["owner_username"], source_repo["name"])
try:
source_ref = resolve_pr_ref(source_path, pr, "source")
except ValueError as exc:
raise HgCommandError(str(exc)) from exc
source_node = source_ref.get("node")
if not source_node:
raise HgCommandError("Source repository has no commits.")
base_node = pr["base_node"] or NULL_REV
if not repo_has_revision(source_path, base_node):
raise HgCommandError("The pull request base revision is not present in the source repository.")
return diff_between_revisions(source_path, base_node, source_node), source_node, source_ref
def pull_source_into_target(target_path, source_path):
completed = run_hg(["pull", str(source_path)], cwd=target_path, check=False, timeout=60)
if completed.returncode != 0:
combined = f"{completed.stdout}\n{completed.stderr}".lower()
if "no changes found" not in combined:
raise HgCommandError(completed.stderr.strip() or "Unable to pull source changes.", completed.returncode)
def merge_pull_request(pr, user):
if pr["status"] != "open":
raise ValueError("Only open pull requests can be merged.")
target_repo = get_repo_by_id(pr["target_repo_id"])
source_repo = get_repo_by_id(pr["source_repo_id"])
if not target_repo or not source_repo:
raise ValueError("Pull request repositories are no longer available.")
target_path = repo_path(target_repo["owner_username"], target_repo["name"])
source_path = repo_path(source_repo["owner_username"], source_repo["name"])
try:
source_ref = resolve_pr_ref(source_path, pr, "source")
target_ref = resolve_pr_ref(target_path, pr, "target")
except ValueError as exc:
raise ValueError(str(exc)) from exc
if target_ref["type"] == REF_TYPE_BRANCH and target_ref.get("closed"):
raise ValueError("The target branch is closed.")
source_node = source_ref.get("node")
if not source_node:
raise ValueError("Source repository has no commits.")
ensure_clean_working_copy(target_path)
target_node_before = target_ref.get("node") or NULL_REV
source_branch = revision_branch(source_path, source_node)
run_hg(["update", "-C", target_node_before], cwd=target_path, check=False)
pull_source_into_target(target_path, source_path)
if not repo_has_revision(target_path, source_node):
raise HgCommandError("Source revision was not pulled into the target repository.")
can_fast_forward = (
target_ref["type"] != REF_TYPE_BRANCH
or target_node_before == NULL_REV
or source_branch == target_ref["name"]
)
if target_node_before != NULL_REV and is_ancestor(target_path, source_node, target_node_before):
run_hg(["update", "-C", target_node_before], cwd=target_path, check=False)
merge_node = target_node_before
elif can_fast_forward and is_ancestor(target_path, target_node_before, source_node):
run_hg(["update", "-C", source_node], cwd=target_path, check=False)
merge_node = source_node
else:
run_hg(["update", "-C", target_node_before], cwd=target_path, check=False)
merge = run_hg(["merge", "--tool", "internal:merge", source_node], cwd=target_path, check=False, timeout=60)
if merge.returncode != 0:
run_hg(["merge", "--abort"], cwd=target_path, check=False)
run_hg(["update", "-C", target_node_before], cwd=target_path, check=False)
message = merge.stderr.strip() or merge.stdout.strip() or "Merge has conflicts."
raise HgCommandError(message, merge.returncode)
commit_message = f"Merge pull request #{pr['number']} from {source_repo['owner_username']}/{source_repo['name']}\n\n{pr['title']}"
commit = run_hg(
["commit", "-u", user["username"], "-m", commit_message],
cwd=target_path,
check=False,
timeout=60,
)
if commit.returncode != 0:
run_hg(["merge", "--abort"], cwd=target_path, check=False)
run_hg(["update", "-C", target_node_before], cwd=target_path, check=False)
raise HgCommandError(commit.stderr.strip() or "Unable to create merge commit.", commit.returncode)
merge_node = repo_tip_node(target_path) or source_node
if target_ref["type"] == REF_TYPE_BOOKMARK and merge_node != NULL_REV:
run_hg(["bookmark", "-f", "-r", merge_node, target_ref["name"]], cwd=target_path, check=True)
now = utcnow()
with db_connect() as conn:
conn.execute(
"""
UPDATE pull_requests
SET status = 'merged',
source_node = ?,
updated_at = ?,
closed_at = ?,
merged_at = ?,
merged_by_id = ?,
merge_node = ?
WHERE id = ?
""",
(source_node, now, now, now, user["id"], merge_node, pr["id"]),
)
conn.execute(
"UPDATE repositories SET updated_at = ? WHERE id = ?",
(now, target_repo["id"]),
)
return merge_node
def render_pull_request_detail(repo, path, pr, error=None, notice=None, comment_value=""):
diff = ""
diff_error = None
current_source_node = pr["source_node"]
current_source_ref = None
try:
diff, current_source_node, current_source_ref = pull_request_diff(pr)
except HgCommandError as exc:
diff_error = str(exc)
return render(
"pull_request_detail.tpl",
repo=repo,
pr=pr,
comments=list_pull_request_comments(pr["id"]),
comment_value=comment_value,
diff=diff,
diff_error=diff_error,
current_source_node=current_source_node,
current_source_ref=current_source_ref,
error=error,
notice=notice,
**repo_page_context(repo, path),
)
def render_commit_detail(repo, path, commit, error=None, notice=None, comment_value=""):
return render(
"commit_detail.tpl",
repo=repo,
commit=commit,
commit_source_ref=commit_ref(path, commit["node"]),
diff=commit_diff(path, commit["node"]),
comments=list_commit_comments(repo["id"], commit["node"]),
comment_value=comment_value,
error=error,
notice=notice,
**repo_page_context(repo, path),
)
def user_owns_repo(user, repo):
return bool(user and repo and user["id"] == repo["owner_id"])
def user_contributes_to_repo(user, repo):
if not user or not repo:
return False
with db_connect() as conn:
row = conn.execute(
"SELECT 1 FROM repo_contributors WHERE repo_id = ? AND user_id = ?",
(repo["id"], user["id"]),
).fetchone()
return bool(row)
def user_can_maintain_repo(user, repo):
return user_owns_repo(user, repo) or user_contributes_to_repo(user, repo)
def repo_page_context(repo, path=None, selected_ref=None):
if path is None:
path = repo_path(repo["owner_username"], repo["name"])
user = current_user()
active_tab = repo_active_tab(repo)
show_ref_picker = active_tab in REF_PICKER_TABS
if selected_ref is None:
selected_ref = selected_repo_ref(path) if show_ref_picker else default_code_ref(path)
fork_target_id = repo["forked_from_repo_id"] or repo["id"]
source_repo = get_repo_by_id(repo["forked_from_repo_id"]) if repo["forked_from_repo_id"] else None
return {
"commit_count": commit_count(path, selected_ref.get("node") if selected_ref else None),
"issue_counts": issue_counts(repo["id"]),
"pr_counts": pull_request_counts(repo["id"]),
"star_count": repo_star_count(repo["id"]),
"is_starred": user_starred_repo(user, repo),
"is_owner": user_owns_repo(user, repo),
"can_maintain": user_can_maintain_repo(user, repo),
"has_fork": bool(user and user_has_fork_for_target(user["id"], fork_target_id)),
"repo_active_tab": active_tab,
"show_ref_picker": show_ref_picker,
"source_repo": source_repo,
"selected_ref": selected_ref,
"selected_ref_label": ref_option_label(selected_ref) if selected_ref else "",
"ref_options": repo_ref_options(path) if show_ref_picker else [],
"selected_ref_value": ref_option_value(
selected_ref.get("type", REF_TYPE_TIP),
selected_ref.get("name", ""),
)
if selected_ref
else "",
}
def repo_active_tab(repo):
base_path = f"/{repo['owner_username']}/{repo['name']}"
current_path = request.path.rstrip("/")
if current_path == base_path:
return "overview"
for tab, suffix in (
("source", "/src"),
("commits", "/commits"),
("tags", "/tags"),
("branches", "/branches"),
("bookmarks", "/bookmarks"),
("issues", "/issues"),
("pulls", "/pulls"),
("settings", "/settings"),
):
if current_path == base_path + suffix or current_path.startswith(base_path + suffix + "/"):
return tab
return ""
def quote_path(path):
return quote(path, safe="/")
def clone_url(owner_username, repo_name):
scheme = request.get_header("X-Forwarded-Proto") or request.urlparts.scheme
host = request.get_header("Host")
if host and host.startswith("0.0.0.0"):
host = "127.0.0.1" + host[len("0.0.0.0") :]
return f"{scheme}://{host}/hg/{owner_username}/{repo_name}"
def parse_basic_auth(rate_limit_kind=None, clear_on_success=True):
header = request.get_header("Authorization", "")
if not header.lower().startswith("basic "):
return None, None
token = header.split(" ", 1)[1].strip()
try:
decoded = base64.b64decode(token).decode("utf-8")
username, password = decoded.split(":", 1)
except (ValueError, UnicodeDecodeError, base64.binascii.Error):
if rate_limit_kind:
if rate_limit_blocked(rate_limit_kind, ""):
return None, "rate_limited"
record_auth_failure(rate_limit_kind, "")
return None, "invalid"
username = username.strip().lower()
if rate_limit_kind and rate_limit_blocked(rate_limit_kind, username):
return None, "rate_limited"
user = get_user_by_username(username)
if user and verify_password(password, user["password_hash"]):
if rate_limit_kind and clear_on_success:
clear_auth_failures(rate_limit_kind, username)
return user, None
if rate_limit_kind:
record_auth_failure(rate_limit_kind, username)
return None, "invalid"
def hg_command_from_request():
cmd = request.query.get("cmd")
if cmd:
return cmd
path_info = request.environ.get("PATH_INFO", "")
parts = path_info.strip("/").split("/")
if len(parts) >= 4 and parts[0] == "hg":
return parts[3]
return ""
def is_hg_write_request():
return hg_command_from_request() in WRITE_HG_COMMANDS
def basic_auth_challenge(message="Authentication required."):
return HTTPResponse(
message + "\n",
status=401,
headers={"WWW-Authenticate": 'Basic realm="HgHost"', "Connection": "close"},
content_type="text/plain; charset=utf-8",
)
def hg_wsgi_response(repo, auth_user):
path = repo_path(repo["owner_username"], repo["name"])
mount = f"/hg/{repo['owner_username']}/{repo['name']}"
original_path = request.environ.get("PATH_INFO", request.path)
rest = original_path[len(mount) :] if original_path.startswith(mount) else ""
env = request.environ.copy()
env["SCRIPT_NAME"] = (env.get("SCRIPT_NAME", "").rstrip("/") + mount) if env.get("SCRIPT_NAME") else mount
env["PATH_INFO"] = rest or ""
env.pop("REPO_NAME", None)
if auth_user:
env["REMOTE_USER"] = auth_user["username"]
status_headers = {}
body_parts = []
body_size = 0
def write_body(chunk):
nonlocal body_size
chunk = chunk if isinstance(chunk, bytes) else chunk.encode("utf-8")
body_size += len(chunk)
if MAX_HG_RESPONSE_BYTES and body_size > MAX_HG_RESPONSE_BYTES:
raise HgResponseTooLarge()
body_parts.append(chunk)
def start_response(status, headers, exc_info=None):
status_headers["status"] = status
status_headers["headers"] = headers
return write_body
# The public factory runs Mercurial's initialization hook. Importing the
# internal hgweb module directly can leave bundle2 handlers such as
# "replycaps" unregistered under some WSGI loaders.
hg_app = make_hgweb(
str(path).encode("utf-8"),
name=f"{repo['owner_username']}/{repo['name']}".encode("utf-8"),
)
try:
body_iter = hg_app(env, start_response)
for chunk in body_iter:
write_body(chunk)
except HgResponseTooLarge:
return HTTPResponse(
"Mercurial response too large.\n",
status=413,
content_type="text/plain; charset=utf-8",
)
finally:
if "body_iter" in locals():
close = getattr(body_iter, "close", None)
if close:
close()
body = b"".join(body_parts)
raw_status = status_headers.get("status", "500 Internal Server Error")
if isinstance(raw_status, bytes):
raw_status = raw_status.decode("latin-1")
status_code = int(raw_status.split(" ", 1)[0])
headers = {}
for key, value in status_headers.get("headers", []):
if isinstance(key, bytes):
key = key.decode("latin-1")
if isinstance(value, bytes):
value = value.decode("latin-1")
headers[key] = value
return HTTPResponse(body=body, status=status_code, headers=headers)
@app.route("/static/<filename:path>")
def static_assets(filename):
return static_file(filename, root=str(BASE_DIR / "static"))
@app.route("/favicon.ico")
def favicon():
return HTTPResponse(status=204)
@app.route("/")
def index():
return render("index.tpl", actions=list_recent_actions(50))
@app.route("/signup", method=["GET", "POST"])
def signup():
if request.method == "GET":
return render("signup.tpl", next_url=safe_next_url(request.query.get("next")))
username_raw = request.forms.get("username", "")
password = request.forms.get("password", "")
next_url = safe_next_url(request.forms.get("next"))
signup_identifier = username_raw.strip().lower()
if rate_limit_blocked("signup", signup_identifier):
return too_many_requests_response()
try:
username = normalize_slug(username_raw, "Username")
if len(password) < 8:
raise ValueError("Password must be at least 8 characters.")
with db_connect() as conn:
conn.execute(
"INSERT INTO users (username, password_hash, created_at) VALUES (?, ?, ?)",
(username, hash_password(password), utcnow()),
)
user = get_user_by_username(username)
login_user(user)
clear_auth_failures("signup", username)
redirect(next_url)
except (sqlite3.IntegrityError, ValueError) as exc:
record_auth_failure("signup", signup_identifier)
message = "Username already exists." if isinstance(exc, sqlite3.IntegrityError) else str(exc)
return render("signup.tpl", error=message, username=username_raw, next_url=next_url)
@app.route("/login", method=["GET", "POST"])
def login():
if request.method == "GET":
return render("login.tpl", next_url=safe_next_url(request.query.get("next")))
username = request.forms.get("username", "").strip().lower()
password = request.forms.get("password", "")
next_url = safe_next_url(request.forms.get("next"))
if rate_limit_blocked("login", username):
return too_many_requests_response()
user = get_user_by_username(username)
if not user or not verify_password(password, user["password_hash"]):
record_auth_failure("login", username)
return render("login.tpl", error="Invalid username or password.", username=username, next_url=next_url)
clear_auth_failures("login", username)
login_user(user)
redirect(next_url)
@app.post("/logout")
def logout():
logout_user()
redirect("/")
@app.route("/settings/profile", method=["GET", "POST"])
def edit_profile():
user = require_login()
if request.method == "GET":
return render("edit_profile.tpl", profile=user)
values = profile_form_values(request.forms, user)
try:
values["website"] = normalize_website(values["website"])
with db_connect() as conn:
conn.execute(
"""
UPDATE users
SET display_name = ?, bio = ?, website = ?
WHERE id = ?
""",
(values["display_name"], values["bio"], values["website"], user["id"]),
)
refreshed = get_user_by_id(user["id"])
request.environ["hghost.user"] = refreshed
return render("edit_profile.tpl", profile=refreshed, notice="Profile updated.")
except ValueError as exc:
return render("edit_profile.tpl", profile=values, error=str(exc))
@app.route("/new", method=["GET", "POST"])
def new_repo():
user = require_login()
if request.method == "GET":
return render("new_repo.tpl")
repo_name_raw = request.forms.get("name", "")
description = request.forms.get("description", "").strip()[:500]
try:
repo_name = normalize_slug(repo_name_raw, "Repository name")
create_repository(user, repo_name, description)
redirect(f"/{user['username']}/{repo_name}")
except (ValueError, HgCommandError) as exc:
return render("new_repo.tpl", error=str(exc), name=repo_name_raw, description=description)
@app.route("/<username>")
def user_profile(username):
profile_user = get_user_by_username(username.lower())
if not profile_user:
abort(404, "User not found.")
user = current_user()
active_tab = request.query.get("tab", "owned")
if active_tab not in {"owned", "stars"}:
active_tab = "owned"
owned_repos = list_owned_repos(profile_user["id"])
starred_repos = list_starred_repos(profile_user["id"])
return render(
"profile.tpl",
profile_user=profile_user,
owned_repos=owned_repos,
starred_repos=starred_repos,
repos=starred_repos if active_tab == "stars" else owned_repos,
active_tab=active_tab,
is_self=bool(user and user["id"] == profile_user["id"]),
)
@app.route("/<owner>/<repo_name>")
def repo_overview(owner, repo_name):
repo = get_repo(owner, repo_name)
if not repo:
abort(404, "Repository not found.")
path = repo_path(owner, repo_name)
selected_ref = selected_repo_ref(path)
files = hg_files(path, selected_ref.get("node"))
readme_name, readme, readme_truncated = readme_preview_for_repo(path, files, revision=selected_ref.get("node"))
readme_is_markdown = is_markdown_file(readme_name)
context = repo_page_context(repo, path, selected_ref=selected_ref)
return render(
"repo.tpl",
repo=repo,
clone_url=clone_url(owner, repo_name),
readme_name=readme_name,
readme=readme,
readme_html=render_markdown(readme) if readme is not None and readme_is_markdown else None,
readme_is_markdown=readme_is_markdown,
readme_truncated=readme_truncated,
**context,
)
@app.post("/<owner>/<repo_name>/star")
def repo_star(owner, repo_name):
user = require_login()
repo = get_repo(owner, repo_name)
if not repo:
abort(404, "Repository not found.")
action = request.forms.get("action", "star")
if action == "unstar":
unstar_repo(user, repo)
else:
star_repo(user, repo)
redirect(f"/{owner}/{repo_name}")
@app.route("/<owner>/<repo_name>/fork", method=["GET", "POST"])
def fork_repo(owner, repo_name):
user = require_login()
source_repo = get_repo(owner, repo_name)
if not source_repo:
abort(404, "Repository not found.")
default_description = source_repo["description"]
if request.method == "GET":
return render(
"fork_repo.tpl",
source_repo=source_repo,
name=source_repo["name"],
description=default_description,
)
repo_name_raw = request.forms.get("name", "")
description = request.forms.get("description", default_description).strip()[:500]
try:
fork_name = normalize_slug(repo_name_raw, "Repository name")
fork_repository(user, source_repo, fork_name, description)
redirect(f"/{user['username']}/{fork_name}")
except (ValueError, HgCommandError) as exc:
return render(
"fork_repo.tpl",
source_repo=source_repo,
name=repo_name_raw,
description=description,
error=str(exc),
)
@app.route("/<owner>/<repo_name>/settings", method=["GET", "POST"])
def repo_settings(owner, repo_name):
user = require_login()
repo = get_repo(owner, repo_name)
if not repo:
abort(404, "Repository not found.")
if not user_owns_repo(user, repo):
abort(403, "Only the owner can update repository settings.")
path = repo_path(owner, repo_name)
if request.method == "GET":
return render(
"repo_settings.tpl",
repo=repo,
contributors=list_repo_contributors(repo["id"]),
contributor_username="",
**repo_page_context(repo, path),
)
action = request.forms.get("action", "save")
if action == "delete":
confirmation = request.forms.get("confirm_name", "").strip()
if confirmation != repo["name"]:
return render(
"repo_settings.tpl",
repo=repo,
contributors=list_repo_contributors(repo["id"]),
contributor_username="",
error=f'Type "{repo["name"]}" to confirm deletion.',
**repo_page_context(repo, path),
)
delete_repository(repo, path)
redirect(f"/{owner}")
if action == "add_contributor":
contributor_username = request.forms.get("username", "").strip()
try:
add_repo_contributor(repo, user, contributor_username)
redirect(f"/{owner}/{repo_name}/settings")
except ValueError as exc:
return render(
"repo_settings.tpl",
repo=repo,
contributors=list_repo_contributors(repo["id"]),
contributor_username=contributor_username,
error=str(exc),
**repo_page_context(repo, path),
)
if action == "remove_contributor":
try:
contributor_user_id = int(request.forms.get("user_id", ""))
except ValueError:
abort(400, "Invalid contributor.")
remove_repo_contributor(repo, contributor_user_id)
redirect(f"/{owner}/{repo_name}/settings")
description = request.forms.get("description", "").strip()[:500]
with db_connect() as conn:
conn.execute(
"UPDATE repositories SET description = ?, updated_at = ? WHERE id = ?",
(description, utcnow(), repo["id"]),
)
updated_repo = get_repo(owner, repo_name)
sync_repo_hgrc(updated_repo)
return render(
"repo_settings.tpl",
repo=updated_repo,
contributors=list_repo_contributors(updated_repo["id"]),
contributor_username="",
notice="Repository settings updated.",
**repo_page_context(updated_repo, path),
)
@app.route("/<owner>/<repo_name>/src")
@app.route("/<owner>/<repo_name>/src/<file_path:path>")
def repo_source(owner, repo_name, file_path=""):
repo = get_repo(owner, repo_name)
if not repo:
abort(404, "Repository not found.")
file_path = clean_repo_path(file_path)
path = repo_path(owner, repo_name)
selected_ref = selected_repo_ref(path)
files = hg_files(path, selected_ref.get("node"))
if file_path in files:
content = read_file_bytes(path, file_path, revision=selected_ref.get("node"))
is_binary = b"\0" in content[:4096]
preview_content, preview_truncated = truncate_bytes_for_render(content)
return render(
"file.tpl",
repo=repo,
file_path=file_path,
content=preview_content.decode("utf-8", "replace") if not is_binary else "",
is_binary=is_binary,
language_class=highlight_language_class(file_path),
size=len(content),
preview_truncated=preview_truncated,
quote_path=quote_path,
**repo_page_context(repo, path, selected_ref=selected_ref),
)
if file_path and not any(item.startswith(file_path + "/") for item in files):
abort(404, "Path not found.")
return render(
"source.tpl",
repo=repo,
current_path=file_path,
entries=build_tree(files, file_path),
quote_path=quote_path,
**repo_page_context(repo, path, selected_ref=selected_ref),
)
@app.route("/<owner>/<repo_name>/raw/<file_path:path>")
def repo_raw(owner, repo_name, file_path):
repo = get_repo(owner, repo_name)
if not repo:
abort(404, "Repository not found.")
file_path = clean_repo_path(file_path)
path = repo_path(owner, repo_name)
selected_ref = selected_repo_ref(path)
files = hg_files(path, selected_ref.get("node"))
if file_path not in files:
abort(404, "File not found.")
content = read_file_bytes(path, file_path, revision=selected_ref.get("node"))
content_type = mimetypes.guess_type(file_path)[0] or "application/octet-stream"
return HTTPResponse(content, content_type=content_type)
@app.route("/<owner>/<repo_name>/commits")
def repo_commits(owner, repo_name):
repo = get_repo(owner, repo_name)
if not repo:
abort(404, "Repository not found.")
path = repo_path(owner, repo_name)
selected_ref = selected_repo_ref(path)
return render(
"commits.tpl",
repo=repo,
commits=commit_log(path, revision=selected_ref.get("node")),
**repo_page_context(repo, path, selected_ref=selected_ref),
)
@app.route("/<owner>/<repo_name>/tags")
def repo_tags(owner, repo_name):
repo = get_repo(owner, repo_name)
if not repo:
abort(404, "Repository not found.")
path = repo_path(owner, repo_name)
return render(
"tags.tpl",
repo=repo,
tags=list_repo_tags(path),
clone_url=clone_url(owner, repo_name),
**repo_page_context(repo, path),
)
@app.route("/<owner>/<repo_name>/branches")
def repo_branches(owner, repo_name):
repo = get_repo(owner, repo_name)
if not repo:
abort(404, "Repository not found.")
path = repo_path(owner, repo_name)
return render(
"branches.tpl",
repo=repo,
branches=list_repo_branches(path),
clone_url=clone_url(owner, repo_name),
**repo_page_context(repo, path),
)
@app.route("/<owner>/<repo_name>/bookmarks")
def repo_bookmarks(owner, repo_name):
repo = get_repo(owner, repo_name)
if not repo:
abort(404, "Repository not found.")
path = repo_path(owner, repo_name)
return render(
"bookmarks.tpl",
repo=repo,
bookmarks=list_repo_bookmarks(path),
clone_url=clone_url(owner, repo_name),
**repo_page_context(repo, path),
)
@app.route("/<owner>/<repo_name>/commits/<node>", method=["GET", "POST"])
def repo_commit(owner, repo_name, node):
repo = get_repo(owner, repo_name)
if not repo:
abort(404, "Repository not found.")
path = repo_path(owner, repo_name)
commit = commit_detail(path, node)
if request.method == "POST":
user = require_login()
action = request.forms.get("action")
if action == "comment":
body = request.forms.get("body", "").strip()
if not body:
return render_commit_detail(
repo,
path,
commit,
error="Comment body is required.",
comment_value=body,
)
now = utcnow()
with db_connect() as conn:
conn.execute(
"""
INSERT INTO commit_comments (repo_id, commit_node, author_id, body, created_at, updated_at)
VALUES (?, ?, ?, ?, ?, ?)
""",
(repo["id"], commit["node"], user["id"], body[:5000], now, now),
)
redirect(f"/{owner}/{repo_name}/commits/{commit['node']}")
return render_commit_detail(repo, path, commit)
@app.route("/<owner>/<repo_name>/pulls")
def repo_pull_requests(owner, repo_name):
repo = get_repo(owner, repo_name)
if not repo:
abort(404, "Repository not found.")
path = repo_path(owner, repo_name)
status = request.query.get("status", "open")
counts = pull_request_counts(repo["id"])
return render(
"pull_requests.tpl",
repo=repo,
pull_requests=list_pull_requests(repo["id"], status),
status=status if status in {"open", "closed", "merged", "all"} else "open",
counts=counts,
**repo_page_context(repo, path),
)
@app.route("/<owner>/<repo_name>/pulls/new", method=["GET", "POST"])
def new_pull_request(owner, repo_name):
user = require_login()
repo = get_repo(owner, repo_name)
if not repo:
abort(404, "Repository not found.")
path = repo_path(owner, repo_name)
forks = list_user_forks_for_target(user["id"], repo["id"])
source_options = []
for fork in forks:
source_options.extend(source_repo_ref_options(fork))
target_options = target_repo_ref_options(path)
target_option_values = {option["value"] for option in target_options}
selected_source_ref = request.forms.get("source_ref") if request.method == "POST" else request.query.get("source_ref")
selected_target_ref = request.forms.get("target_ref") if request.method == "POST" else request.query.get("target_ref")
if not selected_source_ref and source_options:
selected_source_ref = source_options[0]["value"]
if not selected_target_ref and target_options:
default_target = default_code_ref(path)
selected_target_ref = ref_option_value(default_target["type"], default_target.get("name", ""))
if selected_target_ref not in target_option_values:
selected_target_ref = target_options[0]["value"]
if selected_target_ref and selected_target_ref not in target_option_values and target_options:
selected_target_ref = target_options[0]["value"]
title_value = request.forms.get("title", "") if request.method == "POST" else ""
body_value = request.forms.get("body", "") if request.method == "POST" else ""
if request.method == "POST":
try:
source_repo_id, source_ref_type, source_ref_name = parse_source_ref_option_value(selected_source_ref)
target_ref_type, target_ref_name = parse_ref_option_value(
selected_target_ref,
allowed_types=TARGET_PULL_REQUEST_REF_TYPES,
)
except ValueError as exc:
return render(
"new_pull_request.tpl",
repo=repo,
forks=forks,
source_options=source_options,
target_options=target_options,
selected_source_ref=selected_source_ref,
selected_target_ref=selected_target_ref,
title_value=title_value,
body_value=body_value,
error=str(exc),
**repo_page_context(repo, path),
)
source_repo = get_repo_by_id(source_repo_id) if source_repo_id else None
title = title_value.strip()
body = body_value.strip()
if not title:
return render(
"new_pull_request.tpl",
repo=repo,
forks=forks,
source_options=source_options,
target_options=target_options,
selected_source_ref=selected_source_ref,
selected_target_ref=selected_target_ref,
title_value=title_value,
body_value=body_value,
error="Pull request title is required.",
**repo_page_context(repo, path),
)
try:
number = create_pull_request(
repo,
source_repo,
user,
title,
body,
source_ref_type,
source_ref_name,
target_ref_type,
target_ref_name,
)
redirect(f"/{owner}/{repo_name}/pulls/{number}")
except (ValueError, HgCommandError) as exc:
return render(
"new_pull_request.tpl",
repo=repo,
forks=forks,
source_options=source_options,
target_options=target_options,
selected_source_ref=selected_source_ref,
selected_target_ref=selected_target_ref,
title_value=title_value,
body_value=body_value,
error=str(exc),
**repo_page_context(repo, path),
)
return render(
"new_pull_request.tpl",
repo=repo,
forks=forks,
source_options=source_options,
target_options=target_options,
selected_source_ref=selected_source_ref,
selected_target_ref=selected_target_ref,
title_value=title_value,
body_value=body_value,
**repo_page_context(repo, path),
)
@app.route("/<owner>/<repo_name>/pulls/<number:int>", method=["GET", "POST"])
def pull_request_detail(owner, repo_name, number):
repo = get_repo(owner, repo_name)
if not repo:
abort(404, "Repository not found.")
path = repo_path(owner, repo_name)
pr = get_pull_request(repo["id"], number)
if not pr:
abort(404, "Pull request not found.")
if request.method == "POST":
user = require_login()
action = request.forms.get("action")
now = utcnow()
if action == "comment":
body = request.forms.get("body", "").strip()
if not body:
return render_pull_request_detail(
repo,
path,
pr,
error="Comment body is required.",
comment_value=body,
)
with db_connect() as conn:
conn.execute(
"""
INSERT INTO pull_request_comments (pull_request_id, author_id, body, created_at, updated_at)
VALUES (?, ?, ?, ?, ?)
""",
(pr["id"], user["id"], body[:5000], now, now),
)
conn.execute(
"UPDATE pull_requests SET updated_at = ? WHERE id = ?",
(now, pr["id"]),
)
redirect(f"/{owner}/{repo_name}/pulls/{number}")
if not user_can_maintain_repo(user, repo):
abort(403, "Only maintainers can update pull requests.")
if action == "merge":
try:
merge_pull_request(pr, user)
redirect(f"/{owner}/{repo_name}/pulls/{number}")
except (ValueError, HgCommandError) as exc:
return render_pull_request_detail(repo, path, pr, error=str(exc))
if action == "close" and pr["status"] == "open":
now = utcnow()
with db_connect() as conn:
conn.execute(
"""
UPDATE pull_requests
SET status = 'closed', updated_at = ?, closed_at = ?
WHERE id = ?
""",
(now, now, pr["id"]),
)
redirect(f"/{owner}/{repo_name}/pulls/{number}")
return render_pull_request_detail(repo, path, pr)
@app.route("/<owner>/<repo_name>/issues")
def repo_issues(owner, repo_name):
repo = get_repo(owner, repo_name)
if not repo:
abort(404, "Repository not found.")
path = repo_path(owner, repo_name)
status = request.query.get("status", "open")
counts = issue_counts(repo["id"])
context = repo_page_context(repo, path)
return render(
"issues.tpl",
repo=repo,
issues=list_issues(repo["id"], status),
status=status if status in {"open", "closed", "all"} else "open",
counts=counts,
**context,
)
@app.route("/<owner>/<repo_name>/issues/new", method=["GET", "POST"])
def new_issue(owner, repo_name):
user = require_login()
repo = get_repo(owner, repo_name)
if not repo:
abort(404, "Repository not found.")
path = repo_path(owner, repo_name)
if request.method == "GET":
return render("new_issue.tpl", repo=repo, title_value="", body_value="", **repo_page_context(repo, path))
title = request.forms.get("title", "").strip()
body = request.forms.get("body", "").strip()
if not title:
return render(
"new_issue.tpl",
repo=repo,
title_value=title,
body_value=body,
error="Issue title is required.",
**repo_page_context(repo, path),
)
now = utcnow()
with db_connect() as conn:
number = conn.execute(
"SELECT COALESCE(MAX(number), 0) + 1 FROM issues WHERE repo_id = ?",
(repo["id"],),
).fetchone()[0]
conn.execute(
"""
INSERT INTO issues (repo_id, author_id, number, title, body, created_at, updated_at)
VALUES (?, ?, ?, ?, ?, ?, ?)
""",
(repo["id"], user["id"], number, title[:200], body[:5000], now, now),
)
redirect(f"/{owner}/{repo_name}/issues/{number}")
@app.route("/<owner>/<repo_name>/issues/<number:int>", method=["GET", "POST"])
def issue_detail(owner, repo_name, number):
repo = get_repo(owner, repo_name)
if not repo:
abort(404, "Repository not found.")
path = repo_path(owner, repo_name)
issue = get_issue(repo["id"], number)
if not issue:
abort(404, "Issue not found.")
if request.method == "POST":
user = require_login()
action = request.forms.get("action")
now = utcnow()
if action == "comment":
body = request.forms.get("body", "").strip()
if not body:
return render(
"issue_detail.tpl",
repo=repo,
issue=issue,
comments=list_issue_comments(issue["id"]),
comment_value=body,
error="Comment body is required.",
**repo_page_context(repo, path),
)
with db_connect() as conn:
conn.execute(
"""
INSERT INTO issue_comments (issue_id, author_id, body, created_at, updated_at)
VALUES (?, ?, ?, ?, ?)
""",
(issue["id"], user["id"], body[:5000], now, now),
)
conn.execute(
"UPDATE issues SET updated_at = ? WHERE id = ?",
(now, issue["id"]),
)
else:
if not user_can_maintain_repo(user, repo):
abort(403, "Only maintainers can update issues.")
if action == "close" and issue["status"] != "closed":
with db_connect() as conn:
conn.execute(
"UPDATE issues SET status = 'closed', updated_at = ?, closed_at = ? WHERE id = ?",
(now, now, issue["id"]),
)
elif action == "reopen" and issue["status"] != "open":
with db_connect() as conn:
conn.execute(
"UPDATE issues SET status = 'open', updated_at = ?, closed_at = NULL WHERE id = ?",
(now, issue["id"]),
)
redirect(f"/{owner}/{repo_name}/issues/{number}")
return render(
"issue_detail.tpl",
repo=repo,
issue=issue,
comments=list_issue_comments(issue["id"]),
comment_value="",
**repo_page_context(repo, path),
)
@app.route("/hg/<owner>/<repo_name>", method=["GET", "POST"])
@app.route("/hg/<owner>/<repo_name>/<hg_path:path>", method=["GET", "POST"])
def mercurial_http(owner, repo_name, hg_path=""):
repo = get_repo(owner, repo_name)
if not repo:
abort(404, "Repository not found.")
is_write = is_hg_write_request()
auth_user, auth_error = parse_basic_auth("hg" if is_write else None, clear_on_success=not is_write)
if is_write:
if auth_error == "rate_limited":
return too_many_requests_response(close_connection=True)
if auth_error:
return basic_auth_challenge("Invalid Mercurial credentials.")
if not auth_user:
return basic_auth_challenge()
if not user_can_maintain_repo(auth_user, repo):
record_auth_failure("hg", auth_user["username"])
return HTTPResponse(
"Push not authorized for this repository.\n",
status=403,
headers={"Connection": "close"},
content_type="text/plain; charset=utf-8",
)
clear_auth_failures("hg", auth_user["username"])
return hg_wsgi_response(repo, auth_user)
@app.error(404)
def not_found(error):
return render("error.tpl", title="Not found", message=getattr(error, "body", "Not found."))
@app.error(403)
def forbidden(error):
return render("error.tpl", title="Forbidden", message=getattr(error, "body", "Forbidden."))
@app.error(413)
def request_too_large(error):
return render("error.tpl", title="Request too large", message=getattr(error, "body", "Request body too large."))
@app.error(500)
def server_error(error):
return render("error.tpl", title="Server error", message="Something went wrong.")
validate_startup_config()
init_db()
if __name__ == "__main__":
port = int(os.environ.get("PORT", "8080"))
run(app, host="0.0.0.0", port=port, debug=DEBUG, reloader=DEBUG)
