#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""מוסיף לראש כל סיפור את מקורו בחז"ל, עם ציטוט מלא.

המקורות שמורים ב‑`scripts/sources.json` (רשומה אחת לסיפור, ובה מקור
אחד או רשימה של כמה) — מפתח לכל סיפור, ובו מראה המקום
בעברית, קישור למקור, והציטוט עצמו. הציטוטים נשלפו משלושה מקומות: ספריא
(רוב הבבלי, המשנה ומדרשי רבה), ויקיטקסט (הירושלמי והפסיקתא, שספריא אינה
מחזיקה בחלוקה שהספר מפנה אליה), וספר האגדה שבפרויקט בן־יהודה (מקומות
שבהם המדרש המקורי אינו זמין דיגיטלית, ושם מצוין הדבר במפורש בתווית). הסקריפט אינו ממציא
מקורות ואינו פונה לרשת: הוא מזריק לספר את מה שכבר אומת ונשמר בקובץ.

הרצה חוזרת מחליפה בלוקים קיימים במקום לשכפל אותם.

    python3 scripts/add_sources.py            # הזרקה
    python3 scripts/add_sources.py --check    # רק דיווח, בלי לכתוב
"""
import html, json, os, re, sys

HERE = os.path.dirname(os.path.abspath(__file__))
BOOK = os.path.join(os.path.dirname(HERE), 'גדולי ישראל לדורותיהם.html')
DATA = os.path.join(HERE, 'sources.json')

CSS = """
/* ═══ בלוק המקור בראש הסיפור ═══════════════════════════════════════════ */
.source-box {
  background: linear-gradient(135deg, #fff8e1 0%, #fff3f8 100%);
  border: 2px dashed #ffb74d;
  border-radius: 16px;
  padding: 1rem 1.2rem;
  margin: 0 0 1.4rem;
}
.source-ref {
  font-weight: 900; color: #b26a00; font-size: 1.02rem;
  margin-bottom: .55rem; line-height: 1.5;
}
.source-quote {
  font-size: 1.05rem; line-height: 2; color: #3d2b00;
  border-right: 4px solid #ffb74d; padding-right: .9rem; margin: 0;
}
.source-link { margin-top: .6rem; font-size: .9rem; }
.source-link a { color: #b26a00; font-weight: 700; }
@media print { .source-box { break-inside: avoid; } .source-link { display: none; } }
"""

BLOCK_OPEN = '<div class="source-box">\n'
BLOCK_ONE = ('<div class="source-ref">📜 {tag}{label}</div>\n'
             '<blockquote class="source-quote">{text}</blockquote>\n'
             '<div class="source-link"><a href="{url}" target="_blank" rel="noopener">'
             'לַמָּקוֹר הַמָּלֵא בְּ{site} ↗</a></div>\n')

SITES = [('sefaria.org', 'סֶפָרְיָא'), ('wikisource.org', 'וִיקִיטֶקְסְט'),
         ('benyehuda.org', 'פְּרוֹיֶקְט בֶּן־יְהוּדָה'),
         ('toratemetfreeware.com', 'תּוֹרַת אֱמֶת'),
         ('yudataub.github.io', 'מַאֲגַר הַסְּפָרִים')]


def site_of(url):
    for host, name in SITES:
        if host in url:
            return name
    return 'מְקוֹרוֹ'
BLOCK_CLOSE = '</div>\n'


def render(entry):
    """בלוק אחד לסיפור, ובתוכו מקור אחד או כמה. סיפור שעיקרו במקום אחד
       ואמרתו במקום אחר לא ייוצג נאמנה במקור יחיד."""
    items = entry if isinstance(entry, list) else [entry]
    out = [BLOCK_OPEN]
    many = len(items) > 1
    for i, s in enumerate(items, 1):
        out.append(BLOCK_ONE.format(
            tag=('הַמָּקוֹר %d: ' % i) if many else 'הַמָּקוֹר: ',
            label=html.escape(s['label']),
            text=html.escape(s['text']),
            url=html.escape(s['url'], quote=True),
            site=site_of(s['url'])))
    out.append(BLOCK_CLOSE)
    return ''.join(out)


EXISTING = re.compile(r'<div class="source-box">.*?</div>\n</div>\n', re.S)


def main():
    check = '--check' in sys.argv
    doc = open(BOOK, encoding='utf-8').read()
    src = json.load(open(DATA, encoding='utf-8'))

    if '.source-box' not in doc:
        doc = doc.replace('\n/* ═══ CONFETTI', CSS + '\n/* ═══ CONFETTI', 1)
        if '.source-box' not in doc:
            sys.exit('לא נמצא מקום לשתול בו את ה‑CSS')

    # מעבר אחד על כל כרטיסי הסיפורים: מסירים בלוק ישן אם יש, ושותלים חדש
    # אם לסיפור הזה יש מקור. סיפור שהמקור שלו הוסר מ‑sources.json מאבד גם
    # את הבלוק שלו, כדי שלא יישאר בספר ציטוט שכבר איננו במפה.
    out, pos = [], 0
    added = replaced = orphan = 0
    seen = set()
    for m in re.finditer(r'<article class="story-card" id="story-(\d+)"', doc):
        sid = m.group(1)
        seen.add(sid)
        j = doc.index('<div class="story-body">', m.start()) + len('<div class="story-body">')
        out.append(doc[pos:j])
        rest = doc[j:]
        lead = len(rest) - len(rest.lstrip('\n'))
        old = EXISTING.match(rest.lstrip('\n'))
        pos = j + lead + old.end() if old else j
        if sid in src:
            s = src[sid]
            out.append('\n' + render(s))
            replaced += 1 if old else 0
            added += 0 if old else 1
        elif old:
            orphan += 1
    out.append(doc[pos:])
    doc = ''.join(out)

    missing = sorted(set(src) - seen, key=int)
    if missing:
        sys.exit('סיפורים שב‑sources.json ואינם בספר: %s' % ', '.join(missing))

    print('בלוקי מקור: %d נוספו, %d הוחלפו, %d יתומים הוסרו. '
          'סיפורים ללא מקור: %d.'
          % (added, replaced, orphan, len(seen) - len(src)))
    if check:
        print('(--check: הקובץ לא נכתב)')
        return
    open(BOOK, 'w', encoding='utf-8').write(doc)
    print('נכתב. הרץ עכשיו: python3 scripts/check_book.py')


if __name__ == '__main__':
    main()