1use hwpforge_foundation::{
25 ApplyPageType, HwpUnit, NumberFormatType, PageNumberPosition, ShowMode, TextDirection,
26};
27use schemars::JsonSchema;
28use serde::{Deserialize, Serialize};
29
30use crate::column::ColumnSettings;
31use crate::page::PageSettings;
32use crate::paragraph::Paragraph;
33
34#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize, JsonSchema)]
43pub struct Visibility {
44 #[serde(default)]
46 pub hide_first_header: bool,
47 #[serde(default)]
49 pub hide_first_footer: bool,
50 #[serde(default)]
52 pub hide_first_master_page: bool,
53 #[serde(default)]
55 pub hide_first_page_num: bool,
56 #[serde(default)]
58 pub hide_first_empty_line: bool,
59 #[serde(default)]
61 pub show_line_number: bool,
62 #[serde(default)]
64 pub border: ShowMode,
65 #[serde(default)]
67 pub fill: ShowMode,
68}
69
70impl Default for Visibility {
71 fn default() -> Self {
72 Self {
73 hide_first_header: false,
74 hide_first_footer: false,
75 hide_first_master_page: false,
76 hide_first_page_num: false,
77 hide_first_empty_line: false,
78 show_line_number: false,
79 border: ShowMode::ShowAll,
80 fill: ShowMode::ShowAll,
81 }
82 }
83}
84
85#[derive(Debug, Clone, Copy, PartialEq, Eq, Default, Serialize, Deserialize, JsonSchema)]
93pub struct LineNumberShape {
94 #[serde(default)]
96 pub restart_type: u8,
97 #[serde(default)]
99 pub count_by: u16,
100 #[serde(default)]
102 pub distance: HwpUnit,
103 #[serde(default)]
105 pub start_number: u32,
106}
107
108#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, JsonSchema)]
117pub struct PageBorderFillEntry {
118 pub apply_type: String,
120 #[serde(default = "PageBorderFillEntry::default_border_fill_id")]
122 pub border_fill_id: u32,
123 #[serde(default = "PageBorderFillEntry::default_text_border")]
125 pub text_border: String,
126 #[serde(default)]
128 pub header_inside: bool,
129 #[serde(default)]
131 pub footer_inside: bool,
132 #[serde(default = "PageBorderFillEntry::default_fill_area")]
134 pub fill_area: String,
135 #[serde(default = "PageBorderFillEntry::default_offset")]
137 pub offset: [HwpUnit; 4],
138}
139
140impl PageBorderFillEntry {
141 fn default_border_fill_id() -> u32 {
142 1
143 }
144 fn default_text_border() -> String {
145 "PAPER".to_string()
146 }
147 fn default_fill_area() -> String {
148 "PAPER".to_string()
149 }
150 fn default_offset() -> [HwpUnit; 4] {
151 [
153 HwpUnit::new(1417).unwrap(),
154 HwpUnit::new(1417).unwrap(),
155 HwpUnit::new(1417).unwrap(),
156 HwpUnit::new(1417).unwrap(),
157 ]
158 }
159}
160
161impl Default for PageBorderFillEntry {
162 fn default() -> Self {
163 Self {
164 apply_type: "BOTH".to_string(),
165 border_fill_id: 1,
166 text_border: "PAPER".to_string(),
167 header_inside: false,
168 footer_inside: false,
169 fill_area: "PAPER".to_string(),
170 offset: Self::default_offset(),
171 }
172 }
173}
174
175#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize, JsonSchema)]
183pub struct BeginNum {
184 #[serde(default = "BeginNum::one")]
186 pub page: u32,
187 #[serde(default = "BeginNum::one")]
189 pub footnote: u32,
190 #[serde(default = "BeginNum::one")]
192 pub endnote: u32,
193 #[serde(default = "BeginNum::one")]
195 pub pic: u32,
196 #[serde(default = "BeginNum::one")]
198 pub tbl: u32,
199 #[serde(default = "BeginNum::one")]
201 pub equation: u32,
202}
203
204impl BeginNum {
205 fn one() -> u32 {
206 1
207 }
208}
209
210impl Default for BeginNum {
211 fn default() -> Self {
212 Self { page: 1, footnote: 1, endnote: 1, pic: 1, tbl: 1, equation: 1 }
213 }
214}
215
216#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, JsonSchema)]
228pub struct MasterPage {
229 pub apply_page_type: ApplyPageType,
231 pub paragraphs: Vec<Paragraph>,
233}
234
235impl MasterPage {
236 pub fn new(apply_page_type: ApplyPageType, paragraphs: Vec<Paragraph>) -> Self {
238 Self { apply_page_type, paragraphs }
239 }
240}
241
242impl std::fmt::Display for MasterPage {
243 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
244 let n = self.paragraphs.len();
245 let word = if n == 1 { "paragraph" } else { "paragraphs" };
246 write!(f, "MasterPage({n} {word}, {:?})", self.apply_page_type)
247 }
248}
249
250#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, JsonSchema)]
275pub struct HeaderFooter {
276 pub paragraphs: Vec<Paragraph>,
278 pub apply_page_type: ApplyPageType,
280}
281
282impl HeaderFooter {
283 pub fn new(paragraphs: Vec<Paragraph>, apply_page_type: ApplyPageType) -> Self {
285 Self { paragraphs, apply_page_type }
286 }
287
288 pub fn all_pages(paragraphs: Vec<Paragraph>) -> Self {
305 Self { paragraphs, apply_page_type: ApplyPageType::Both }
306 }
307
308 #[deprecated(since = "0.2.0", note = "Use `all_pages()` instead")]
310 pub fn both(paragraphs: Vec<Paragraph>) -> Self {
311 Self::all_pages(paragraphs)
312 }
313}
314
315impl std::fmt::Display for HeaderFooter {
316 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
317 let n = self.paragraphs.len();
318 let word = if n == 1 { "paragraph" } else { "paragraphs" };
319 write!(f, "HeaderFooter({n} {word}, {:?})", self.apply_page_type)
320 }
321}
322
323#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, JsonSchema)]
345pub struct PageNumber {
346 pub position: PageNumberPosition,
348 pub number_format: NumberFormatType,
350 pub decoration: String,
353}
354
355impl PageNumber {
356 pub fn new(position: PageNumberPosition, number_format: NumberFormatType) -> Self {
358 Self { position, number_format, decoration: String::new() }
359 }
360
361 pub fn bottom_center() -> Self {
379 Self {
380 position: PageNumberPosition::BottomCenter,
381 number_format: NumberFormatType::Digit,
382 decoration: String::new(),
383 }
384 }
385
386 pub fn with_decoration(
402 position: PageNumberPosition,
403 number_format: NumberFormatType,
404 decoration: impl Into<String>,
405 ) -> Self {
406 Self { position, number_format, decoration: decoration.into() }
407 }
408
409 #[deprecated(since = "0.2.0", note = "Use `with_decoration()` instead")]
411 pub fn with_side_char(
412 position: PageNumberPosition,
413 number_format: NumberFormatType,
414 side_char: impl Into<String>,
415 ) -> Self {
416 Self::with_decoration(position, number_format, side_char)
417 }
418}
419
420impl std::fmt::Display for PageNumber {
421 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
422 write!(f, "PageNumber({:?}, {:?})", self.position, self.number_format)
423 }
424}
425
426#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, JsonSchema)]
447pub struct Section {
448 pub paragraphs: Vec<Paragraph>,
450 pub page_settings: PageSettings,
452 #[serde(default, skip_serializing_if = "Option::is_none")]
454 pub header: Option<HeaderFooter>,
455 #[serde(default, skip_serializing_if = "Option::is_none")]
457 pub footer: Option<HeaderFooter>,
458 #[serde(default, skip_serializing_if = "Option::is_none")]
460 pub page_number: Option<PageNumber>,
461 #[serde(default, skip_serializing_if = "Option::is_none")]
463 pub column_settings: Option<ColumnSettings>,
464 #[serde(default, skip_serializing_if = "Option::is_none")]
467 pub visibility: Option<Visibility>,
468 #[serde(default, skip_serializing_if = "Option::is_none")]
470 pub line_number_shape: Option<LineNumberShape>,
471 #[serde(default, skip_serializing_if = "Option::is_none")]
473 pub page_border_fills: Option<Vec<PageBorderFillEntry>>,
474 #[serde(default, skip_serializing_if = "Option::is_none")]
477 pub master_pages: Option<Vec<MasterPage>>,
478 #[serde(default, skip_serializing_if = "Option::is_none")]
481 pub begin_num: Option<BeginNum>,
482 #[serde(default)]
485 pub text_direction: TextDirection,
486}
487
488impl Section {
489 pub fn new(page_settings: PageSettings) -> Self {
501 Self {
502 paragraphs: Vec::new(),
503 page_settings,
504 header: None,
505 footer: None,
506 page_number: None,
507 column_settings: None,
508 visibility: None,
509 line_number_shape: None,
510 page_border_fills: None,
511 master_pages: None,
512 begin_num: None,
513 text_direction: TextDirection::Horizontal,
514 }
515 }
516
517 pub fn with_paragraphs(paragraphs: Vec<Paragraph>, page_settings: PageSettings) -> Self {
534 Self {
535 paragraphs,
536 page_settings,
537 header: None,
538 footer: None,
539 page_number: None,
540 column_settings: None,
541 visibility: None,
542 line_number_shape: None,
543 page_border_fills: None,
544 master_pages: None,
545 begin_num: None,
546 text_direction: TextDirection::Horizontal,
547 }
548 }
549
550 pub fn with_text_direction(mut self, dir: TextDirection) -> Self {
564 self.text_direction = dir;
565 self
566 }
567
568 pub fn add_paragraph(&mut self, paragraph: Paragraph) {
570 self.paragraphs.push(paragraph);
571 }
572
573 pub fn paragraph_count(&self) -> usize {
575 self.paragraphs.len()
576 }
577
578 pub fn is_empty(&self) -> bool {
580 self.paragraphs.is_empty()
581 }
582
583 pub fn content_counts(&self) -> ContentCounts {
600 let mut tables: usize = 0;
601 let mut images: usize = 0;
602 let mut charts: usize = 0;
603
604 for para in &self.paragraphs {
605 for run in ¶.runs {
606 match &run.content {
607 crate::RunContent::Table(_) => tables += 1,
608 crate::RunContent::Image(_) => images += 1,
609 crate::RunContent::Control(c) => {
610 if matches!(**c, crate::control::Control::Chart { .. }) {
611 charts += 1;
612 }
613 }
614 _ => {}
615 }
616 }
617 }
618
619 ContentCounts { tables, images, charts }
620 }
621}
622
623#[derive(Debug, Clone, Copy, PartialEq, Eq, Default)]
625pub struct ContentCounts {
626 pub tables: usize,
628 pub images: usize,
630 pub charts: usize,
632}
633
634impl std::fmt::Display for Section {
635 fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
636 let n = self.paragraphs.len();
637 let word = if n == 1 { "paragraph" } else { "paragraphs" };
638 write!(f, "Section({n} {word})")
639 }
640}
641
642#[cfg(test)]
643mod tests {
644 use super::*;
645 use crate::run::Run;
646 use hwpforge_foundation::{
647 ApplyPageType, CharShapeIndex, NumberFormatType, PageNumberPosition, ParaShapeIndex,
648 };
649
650 fn simple_paragraph() -> Paragraph {
651 Paragraph::with_runs(
652 vec![Run::text("text", CharShapeIndex::new(0))],
653 ParaShapeIndex::new(0),
654 )
655 }
656
657 #[test]
658 fn new_is_empty() {
659 let section = Section::new(PageSettings::a4());
660 assert!(section.is_empty());
661 assert_eq!(section.paragraph_count(), 0);
662 }
663
664 #[test]
665 fn with_paragraphs() {
666 let section = Section::with_paragraphs(
667 vec![simple_paragraph(), simple_paragraph()],
668 PageSettings::a4(),
669 );
670 assert_eq!(section.paragraph_count(), 2);
671 assert!(!section.is_empty());
672 }
673
674 #[test]
675 fn add_paragraph() {
676 let mut section = Section::new(PageSettings::a4());
677 section.add_paragraph(simple_paragraph());
678 section.add_paragraph(simple_paragraph());
679 assert_eq!(section.paragraph_count(), 2);
680 }
681
682 #[test]
683 fn page_settings_preserved() {
684 let section = Section::new(PageSettings::letter());
685 assert_eq!(section.page_settings, PageSettings::letter());
686 }
687
688 #[test]
689 fn display_singular() {
690 let section = Section::with_paragraphs(vec![simple_paragraph()], PageSettings::a4());
691 assert_eq!(section.to_string(), "Section(1 paragraph)");
692 }
693
694 #[test]
695 fn display_plural() {
696 let section = Section::with_paragraphs(
697 vec![simple_paragraph(), simple_paragraph()],
698 PageSettings::a4(),
699 );
700 assert_eq!(section.to_string(), "Section(2 paragraphs)");
701 }
702
703 #[test]
704 fn equality() {
705 let a = Section::with_paragraphs(vec![simple_paragraph()], PageSettings::a4());
706 let b = Section::with_paragraphs(vec![simple_paragraph()], PageSettings::a4());
707 assert_eq!(a, b);
708 }
709
710 #[test]
711 fn inequality_different_page_settings() {
712 let a = Section::new(PageSettings::a4());
713 let b = Section::new(PageSettings::letter());
714 assert_ne!(a, b);
715 }
716
717 #[test]
718 fn clone_independence() {
719 let section = Section::with_paragraphs(vec![simple_paragraph()], PageSettings::a4());
720 let mut cloned = section.clone();
721 cloned.add_paragraph(simple_paragraph());
722 assert_eq!(section.paragraph_count(), 1);
723 assert_eq!(cloned.paragraph_count(), 2);
724 }
725
726 #[test]
727 fn serde_roundtrip() {
728 let section = Section::with_paragraphs(vec![simple_paragraph()], PageSettings::a4());
729 let json = serde_json::to_string(§ion).unwrap();
730 let back: Section = serde_json::from_str(&json).unwrap();
731 assert_eq!(section, back);
732 }
733
734 #[test]
735 fn serde_empty_section() {
736 let section = Section::new(PageSettings::a4());
737 let json = serde_json::to_string(§ion).unwrap();
738 let back: Section = serde_json::from_str(&json).unwrap();
739 assert_eq!(section, back);
740 }
741
742 #[test]
743 fn serde_letter_page() {
744 let section = Section::new(PageSettings::letter());
745 let json = serde_json::to_string(§ion).unwrap();
746 let back: Section = serde_json::from_str(&json).unwrap();
747 assert_eq!(section, back);
748 }
749
750 #[test]
755 fn header_footer_new() {
756 let hf =
757 HeaderFooter::new(vec![Paragraph::new(ParaShapeIndex::new(0))], ApplyPageType::Both);
758 assert_eq!(hf.paragraphs.len(), 1);
759 assert_eq!(hf.apply_page_type, ApplyPageType::Both);
760 }
761
762 #[test]
763 fn header_footer_even_odd() {
764 let even = HeaderFooter::new(vec![], ApplyPageType::Even);
765 let odd = HeaderFooter::new(vec![], ApplyPageType::Odd);
766 assert_eq!(even.apply_page_type, ApplyPageType::Even);
767 assert_eq!(odd.apply_page_type, ApplyPageType::Odd);
768 assert_ne!(even, odd);
769 }
770
771 #[test]
772 fn header_footer_display() {
773 let hf =
774 HeaderFooter::new(vec![Paragraph::new(ParaShapeIndex::new(0))], ApplyPageType::Both);
775 let s = hf.to_string();
776 assert!(s.contains("1 paragraph"), "display: {s}");
777 assert!(s.contains("Both"), "display: {s}");
778 }
779
780 #[test]
781 fn header_footer_serde_roundtrip() {
782 let hf = HeaderFooter::new(
783 vec![Paragraph::with_runs(
784 vec![Run::text("Header text", CharShapeIndex::new(0))],
785 ParaShapeIndex::new(0),
786 )],
787 ApplyPageType::Both,
788 );
789 let json = serde_json::to_string(&hf).unwrap();
790 let back: HeaderFooter = serde_json::from_str(&json).unwrap();
791 assert_eq!(hf, back);
792 }
793
794 #[test]
795 fn header_footer_clone_independence() {
796 let hf =
797 HeaderFooter::new(vec![Paragraph::new(ParaShapeIndex::new(0))], ApplyPageType::Both);
798 let mut cloned = hf.clone();
799 cloned.paragraphs.push(Paragraph::new(ParaShapeIndex::new(1)));
800 assert_eq!(hf.paragraphs.len(), 1);
801 assert_eq!(cloned.paragraphs.len(), 2);
802 }
803
804 #[test]
809 fn page_number_new() {
810 let pn = PageNumber::new(PageNumberPosition::BottomCenter, NumberFormatType::Digit);
811 assert_eq!(pn.position, PageNumberPosition::BottomCenter);
812 assert_eq!(pn.number_format, NumberFormatType::Digit);
813 assert!(pn.decoration.is_empty());
814 }
815
816 #[test]
817 fn page_number_with_decoration() {
818 let pn = PageNumber::with_decoration(
819 PageNumberPosition::BottomCenter,
820 NumberFormatType::RomanCapital,
821 "- ",
822 );
823 assert_eq!(pn.decoration, "- ");
824 assert_eq!(pn.number_format, NumberFormatType::RomanCapital);
825 }
826
827 #[test]
828 #[allow(deprecated)]
829 fn page_number_with_side_char_deprecated() {
830 let pn = PageNumber::with_side_char(
831 PageNumberPosition::BottomCenter,
832 NumberFormatType::Digit,
833 "- ",
834 );
835 assert_eq!(pn.decoration, "- ");
836 }
837
838 #[test]
839 fn page_number_display() {
840 let pn = PageNumber::new(PageNumberPosition::TopCenter, NumberFormatType::Digit);
841 let s = pn.to_string();
842 assert!(s.contains("TopCenter"), "display: {s}");
843 assert!(s.contains("Digit"), "display: {s}");
844 }
845
846 #[test]
847 fn page_number_serde_roundtrip() {
848 let pn = PageNumber::with_decoration(
849 PageNumberPosition::BottomCenter,
850 NumberFormatType::CircledDigit,
851 "< ",
852 );
853 let json = serde_json::to_string(&pn).unwrap();
854 let back: PageNumber = serde_json::from_str(&json).unwrap();
855 assert_eq!(pn, back);
856 }
857
858 #[test]
859 fn page_number_equality() {
860 let a = PageNumber::new(PageNumberPosition::BottomCenter, NumberFormatType::Digit);
861 let b = PageNumber::new(PageNumberPosition::BottomCenter, NumberFormatType::Digit);
862 assert_eq!(a, b);
863 }
864
865 #[test]
866 fn page_number_inequality() {
867 let a = PageNumber::new(PageNumberPosition::BottomCenter, NumberFormatType::Digit);
868 let b = PageNumber::new(PageNumberPosition::TopCenter, NumberFormatType::Digit);
869 assert_ne!(a, b);
870 }
871
872 #[test]
877 fn section_new_has_none_fields() {
878 let section = Section::new(PageSettings::a4());
879 assert!(section.header.is_none());
880 assert!(section.footer.is_none());
881 assert!(section.page_number.is_none());
882 assert!(section.column_settings.is_none());
883 }
884
885 #[test]
886 fn section_with_header_footer() {
887 let mut section = Section::new(PageSettings::a4());
888 section.header = Some(HeaderFooter::new(
889 vec![Paragraph::with_runs(
890 vec![Run::text("Header", CharShapeIndex::new(0))],
891 ParaShapeIndex::new(0),
892 )],
893 ApplyPageType::Both,
894 ));
895 section.footer = Some(HeaderFooter::new(
896 vec![Paragraph::with_runs(
897 vec![Run::text("Footer", CharShapeIndex::new(0))],
898 ParaShapeIndex::new(0),
899 )],
900 ApplyPageType::Both,
901 ));
902 assert!(section.header.is_some());
903 assert!(section.footer.is_some());
904 }
905
906 #[test]
907 fn section_with_page_number() {
908 let mut section = Section::new(PageSettings::a4());
909 section.page_number =
910 Some(PageNumber::new(PageNumberPosition::BottomCenter, NumberFormatType::Digit));
911 assert!(section.page_number.is_some());
912 }
913
914 #[test]
915 fn section_serde_with_optional_fields() {
916 let mut section = Section::new(PageSettings::a4());
917 section.header = Some(HeaderFooter::new(vec![], ApplyPageType::Both));
918 section.page_number =
919 Some(PageNumber::new(PageNumberPosition::BottomCenter, NumberFormatType::Digit));
920 let json = serde_json::to_string(§ion).unwrap();
921 let back: Section = serde_json::from_str(&json).unwrap();
922 assert_eq!(section, back);
923 }
924
925 #[test]
926 fn section_serde_none_fields_skipped() {
927 let section = Section::new(PageSettings::a4());
928 let json = serde_json::to_string(§ion).unwrap();
929 assert!(!json.contains("\"header\""));
932 assert!(!json.contains("\"footer\""));
933 assert!(!json.contains("\"page_number\""));
934 assert!(!json.contains("\"column_settings\""));
935 let back: Section = serde_json::from_str(&json).unwrap();
936 assert_eq!(section, back);
937 }
938
939 #[test]
944 fn header_footer_all_pages_apply_page_type() {
945 let hf = HeaderFooter::all_pages(vec![Paragraph::new(ParaShapeIndex::new(0))]);
946 assert_eq!(hf.apply_page_type, ApplyPageType::Both);
947 }
948
949 #[test]
950 fn header_footer_all_pages_preserves_paragraphs() {
951 let paras = vec![simple_paragraph(), simple_paragraph()];
952 let hf = HeaderFooter::all_pages(paras);
953 assert_eq!(hf.paragraphs.len(), 2);
954 }
955
956 #[test]
957 fn header_footer_all_pages_empty_paragraphs() {
958 let hf = HeaderFooter::all_pages(vec![]);
959 assert_eq!(hf.apply_page_type, ApplyPageType::Both);
960 assert!(hf.paragraphs.is_empty());
961 }
962
963 #[test]
964 #[allow(deprecated)]
965 fn header_footer_both_deprecated_alias() {
966 let hf = HeaderFooter::both(vec![Paragraph::new(ParaShapeIndex::new(0))]);
967 assert_eq!(hf.apply_page_type, ApplyPageType::Both);
968 }
969
970 #[test]
975 fn page_number_bottom_center_position() {
976 let pn = PageNumber::bottom_center();
977 assert_eq!(pn.position, PageNumberPosition::BottomCenter);
978 }
979
980 #[test]
981 fn page_number_bottom_center_format() {
982 let pn = PageNumber::bottom_center();
983 assert_eq!(pn.number_format, NumberFormatType::Digit);
984 }
985
986 #[test]
987 fn page_number_bottom_center_no_decoration() {
988 let pn = PageNumber::bottom_center();
989 assert!(pn.decoration.is_empty());
990 }
991
992 #[test]
993 fn page_number_bottom_center_equals_explicit() {
994 let shortcut = PageNumber::bottom_center();
995 let explicit = PageNumber::new(PageNumberPosition::BottomCenter, NumberFormatType::Digit);
996 assert_eq!(shortcut, explicit);
997 }
998
999 #[test]
1000 fn section_backward_compat_deserialize() {
1001 let a4 = PageSettings::a4();
1003 let json = serde_json::to_string(&Section::with_paragraphs(vec![], a4)).unwrap();
1004 let section: Section = serde_json::from_str(&json).unwrap();
1005 assert!(section.header.is_none());
1006 assert!(section.footer.is_none());
1007 assert!(section.page_number.is_none());
1008 }
1009
1010 #[test]
1011 fn all_pages_equals_new_with_both() {
1012 let paras = vec![simple_paragraph()];
1013 let from_all_pages = HeaderFooter::all_pages(paras.clone());
1014 let from_new = HeaderFooter::new(paras, ApplyPageType::Both);
1015 assert_eq!(from_all_pages, from_new);
1016 }
1017}