1"""Add a new slide to an unpacked PPTX directory.
2 
3Usage: python add_slide.py <unpacked_dir> <source>
4 
5The source can be:
6  - A slide file (e.g., slide2.xml) - duplicates the slide
7  - A layout file (e.g., slideLayout2.xml) - creates from layout
8 
9Examples:
10    python add_slide.py unpacked/ slide2.xml
11    # Duplicates slide2, creates slide5.xml
12 
13    python add_slide.py unpacked/ slideLayout2.xml
14    # Creates slide5.xml from slideLayout2.xml
15 
16To see available layouts: ls unpacked/ppt/slideLayouts/
17 
18Prints the <p:sldId> element to add to presentation.xml.
19"""
20 
21import re
22import shutil
23import sys
24from pathlib import Path
25 
26 
27def get_next_slide_number(slides_dir: Path) -> int:
28    existing = [int(m.group(1)) for f in slides_dir.glob("slide*.xml")
29                if (m := re.match(r"slide(\d+)\.xml", f.name))]
30    return max(existing) + 1 if existing else 1
31 
32 
33def create_slide_from_layout(unpacked_dir: Path, layout_file: str) -> None:
34    slides_dir = unpacked_dir / "ppt" / "slides"
35    rels_dir = slides_dir / "_rels"
36    layouts_dir = unpacked_dir / "ppt" / "slideLayouts"
37 
38    layout_path = layouts_dir / layout_file
39    if not layout_path.exists():
40        print(f"Error: {layout_path} not found", file=sys.stderr)
41        sys.exit(1)
42 
43    next_num = get_next_slide_number(slides_dir)
44    dest = f"slide{next_num}.xml"
45    dest_slide = slides_dir / dest
46    dest_rels = rels_dir / f"{dest}.rels"
47 
48    slide_xml = '''<?xml version="1.0" encoding="UTF-8" standalone="yes"?>
49<p:sld xmlns:a="http://schemas.openxmlformats.org/drawingml/2006/main" xmlns:r="http://schemas.openxmlformats.org/officeDocument/2006/relationships" xmlns:p="http://schemas.openxmlformats.org/presentationml/2006/main">
50  <p:cSld>
51    <p:spTree>
52      <p:nvGrpSpPr>
53        <p:cNvPr id="1" name=""/>
54        <p:cNvGrpSpPr/>
55        <p:nvPr/>
56      </p:nvGrpSpPr>
57      <p:grpSpPr>
58        <a:xfrm>
59          <a:off x="0" y="0"/>
60          <a:ext cx="0" cy="0"/>
61          <a:chOff x="0" y="0"/>
62          <a:chExt cx="0" cy="0"/>
63        </a:xfrm>
64      </p:grpSpPr>
65    </p:spTree>
66  </p:cSld>
67  <p:clrMapOvr>
68    <a:masterClrMapping/>
69  </p:clrMapOvr>
70</p:sld>'''
71    dest_slide.write_text(slide_xml, encoding="utf-8")
72 
73    rels_dir.mkdir(exist_ok=True)
74    rels_xml = f'''<?xml version="1.0" encoding="UTF-8" standalone="yes"?>
75<Relationships xmlns="http://schemas.openxmlformats.org/package/2006/relationships">
76  <Relationship Id="rId1" Type="http://schemas.openxmlformats.org/officeDocument/2006/relationships/slideLayout" Target="../slideLayouts/{layout_file}"/>
77</Relationships>'''
78    dest_rels.write_text(rels_xml, encoding="utf-8")
79 
80    _add_to_content_types(unpacked_dir, dest)
81 
82    rid = _add_to_presentation_rels(unpacked_dir, dest)
83 
84    next_slide_id = _get_next_slide_id(unpacked_dir)
85 
86    print(f"Created {dest} from {layout_file}")
87    print(f'Add to presentation.xml <p:sldIdLst>: <p:sldId id="{next_slide_id}" r:id="{rid}"/>')
88 
89 
90def duplicate_slide(unpacked_dir: Path, source: str) -> None:
91    slides_dir = unpacked_dir / "ppt" / "slides"
92    rels_dir = slides_dir / "_rels"
93 
94    source_slide = slides_dir / source
95 
96    if not source_slide.exists():
97        print(f"Error: {source_slide} not found", file=sys.stderr)
98        sys.exit(1)
99 
100    next_num = get_next_slide_number(slides_dir)
101    dest = f"slide{next_num}.xml"
102    dest_slide = slides_dir / dest
103 
104    source_rels = rels_dir / f"{source}.rels"
105    dest_rels = rels_dir / f"{dest}.rels"
106 
107    shutil.copy2(source_slide, dest_slide)
108 
109    if source_rels.exists():
110        shutil.copy2(source_rels, dest_rels)
111 
112        rels_content = dest_rels.read_text(encoding="utf-8")
113        rels_content = re.sub(
114            r'\s*<Relationship[^>]*Type="[^"]*notesSlide"[^>]*/>\s*',
115            "\n",
116            rels_content,
117        )
118        dest_rels.write_text(rels_content, encoding="utf-8")
119 
120    _add_to_content_types(unpacked_dir, dest)
121 
122    rid = _add_to_presentation_rels(unpacked_dir, dest)
123 
124    next_slide_id = _get_next_slide_id(unpacked_dir)
125 
126    print(f"Created {dest} from {source}")
127    print(f'Add to presentation.xml <p:sldIdLst>: <p:sldId id="{next_slide_id}" r:id="{rid}"/>')
128 
129 
130def _add_to_content_types(unpacked_dir: Path, dest: str) -> None:
131    content_types_path = unpacked_dir / "[Content_Types].xml"
132    content_types = content_types_path.read_text(encoding="utf-8")
133 
134    new_override = f'<Override PartName="/ppt/slides/{dest}" ContentType="application/vnd.openxmlformats-officedocument.presentationml.slide+xml"/>'
135 
136    if f"/ppt/slides/{dest}" not in content_types:
137        content_types = content_types.replace("</Types>", f"  {new_override}\n</Types>")
138        content_types_path.write_text(content_types, encoding="utf-8")
139 
140 
141def _add_to_presentation_rels(unpacked_dir: Path, dest: str) -> str:
142    pres_rels_path = unpacked_dir / "ppt" / "_rels" / "presentation.xml.rels"
143    pres_rels = pres_rels_path.read_text(encoding="utf-8")
144 
145    rids = [int(m) for m in re.findall(r'Id="rId(\d+)"', pres_rels)]
146    next_rid = max(rids) + 1 if rids else 1
147    rid = f"rId{next_rid}"
148 
149    new_rel = f'<Relationship Id="{rid}" Type="http://schemas.openxmlformats.org/officeDocument/2006/relationships/slide" Target="slides/{dest}"/>'
150 
151    if f"slides/{dest}" not in pres_rels:
152        pres_rels = pres_rels.replace("</Relationships>", f"  {new_rel}\n</Relationships>")
153        pres_rels_path.write_text(pres_rels, encoding="utf-8")
154 
155    return rid
156 
157 
158def _get_next_slide_id(unpacked_dir: Path) -> int:
159    pres_path = unpacked_dir / "ppt" / "presentation.xml"
160    pres_content = pres_path.read_text(encoding="utf-8")
161    slide_ids = [int(m) for m in re.findall(r'<p:sldId[^>]*id="(\d+)"', pres_content)]
162    return max(slide_ids) + 1 if slide_ids else 256
163 
164 
165def parse_source(source: str) -> tuple[str, str | None]:
166    if source.startswith("slideLayout") and source.endswith(".xml"):
167        return ("layout", source)
168 
169    return ("slide", None)
170 
171 
172if __name__ == "__main__":
173    if len(sys.argv) != 3:
174        print("Usage: python add_slide.py <unpacked_dir> <source>", file=sys.stderr)
175        print("", file=sys.stderr)
176        print("Source can be:", file=sys.stderr)
177        print("  slide2.xml        - duplicate an existing slide", file=sys.stderr)
178        print("  slideLayout2.xml  - create from a layout template", file=sys.stderr)
179        print("", file=sys.stderr)
180        print("To see available layouts: ls <unpacked_dir>/ppt/slideLayouts/", file=sys.stderr)
181        sys.exit(1)
182 
183    unpacked_dir = Path(sys.argv[1])
184    source = sys.argv[2]
185 
186    if not unpacked_dir.exists():
187        print(f"Error: {unpacked_dir} not found", file=sys.stderr)
188        sys.exit(1)
189 
190    source_type, layout_file = parse_source(source)
191 
192    if source_type == "layout" and layout_file is not None:
193        create_slide_from_layout(unpacked_dir, layout_file)
194    else:
195        duplicate_slide(unpacked_dir, source)
196