package main

import (
	"fmt"
	"github.com/landaiqing/go-dockit/document"
	"log"
)

func main() {
	// 创建一个新的Word文档
	doc := document.NewDocument()

	// 设置文档属性
	doc.SetTitle("数据库三线表示例")
	doc.SetCreator("go-dockit库")
	doc.SetDescription("使用go-dockit库创建学术论文中常用的数据库三线表")

	// 添加标题
	titlePara := doc.AddParagraph()
	titlePara.SetAlignment("center")
	titlePara.SetSpacingAfter(200)
	titleRun := titlePara.AddRun()
	titleRun.AddText("数据库三线表示例")
	titleRun.SetBold(true)
	titleRun.SetFontSize(32) // 16磅
	titleRun.SetFontFamily("宋体")

	// 添加说明文字
	explainPara := doc.AddParagraph()
	explainPara.SetIndentFirstLine(420)
	explainPara.SetSpacingAfter(300)
	explainPara.AddRun().AddText("数据库三线表是学术论文中常用的一种表格格式，特点是只有三根水平线（顶线、表头分隔线和底线），没有垂直线分隔列。这种表格格式符合APA（美国心理学会）和许多学术期刊的规范。")

	// ========== 示例1：基本三线表 ==========
	// 添加表格标题（通常三线表的标题在表格上方）
	tableTitlePara := doc.AddParagraph()
	tableTitlePara.SetAlignment("center")
	tableTitlePara.SetSpacingAfter(0)
	tableTitlePara.SetSpacingBefore(0)
	tableTitlePara.SetLineSpacing(1.5, "auto") // 设置1.5倍行距
	tableTitleRun := tableTitlePara.AddRun()
	tableTitleRun.AddText("表4-1 文章信息表")
	tableTitleRun.SetBold(true)
	tableTitleRun.SetFontSize(21) // 五号字体约为10.5磅(21)
	tableTitleRun.SetFontFamily("宋体")
	// 表序号设置为Times New Roman
	tableTitleRun.SetFontFamilyForRunes("Times New Roman", []rune("表4-1"))

	// 创建一个表格
	table1 := doc.AddTable(6, 5)
	table1.SetWidth("100%", "pct") // 与文字齐宽
	table1.SetAlignment("center")

	// 设置行高为0.72厘米（固定值）
	for i := 0; i < 6; i++ {
		table1.Rows[i].SetHeight(567, "exact") // 0.72厘米 ≈ 567 twip，"exact"表示固定行高
	}

	// 填充表头
	headers := []string{"字段", "字段名", "类型", "长度", "非空"}
	for i, header := range headers {
		cellPara := table1.Rows[0].Cells[i].AddParagraph()
		cellPara.SetAlignment("center")
		cellPara.SetLineSpacing(1.5, "auto") // 1.5倍行距
		cellRun := cellPara.AddRun()
		cellRun.AddText(header)
		cellRun.SetBold(false)
		cellRun.SetFontSize(21) // 五号字体
		cellRun.SetFontFamily("宋体")
	}

	// 填充数据行
	data := [][]string{
		{"标题", "title", "varchar", "100", "是"},
		{"文章分类", "sort", "varchar", "150", "是"},
		{"作者学号", "author_sn", "varchar", "100", "是"},
		{"作者姓名", "author_name", "varchar", "100", "否"},
		{"文章内容", "description", "longtext", "默认", "否"},
	}

	for i, row := range data {
		for j, cell := range row {
			para := table1.Rows[i+1].Cells[j].AddParagraph()
			para.SetAlignment("center")
			para.SetLineSpacing(1.5, "auto") // 1.5倍行距
			cellRun := para.AddRun()
			cellRun.AddText(cell)
			cellRun.SetFontSize(21) // 五号字体
			cellRun.SetFontFamily("宋体")
			// 英文字体设置为Times New Roman
			if j == 1 || j == 2 { // 字段名和类型列通常包含英文
				cellRun.SetFontFamilyForRunes("Times New Roman", []rune(cell))
			}
		}
	}

	// 设置三线表样式
	// 1. 首先清除所有默认边框
	table1.SetBorders("all", "", 0, "")

	// 2. 顶线（表格顶部边框），1.5磅
	table1.SetBorders("top", "single", 24, "000000") // 1.5磅 = 24 twip

	// 3. 表头分隔线（第一行底部边框），1磅
	for i := 0; i < 5; i++ {
		table1.Rows[0].Cells[i].SetBorders("bottom", "single", 16, "000000") // 1磅 = 16 twip
	}

	// 4. 底线（表格底部边框），1.5磅
	table1.SetBorders("bottom", "single", 24, "000000") // 1.5磅 = 24 twip

	// 显式设置内部边框为"none"，而不是空字符串
	table1.SetBorders("insideH", "none", 0, "000000")
	table1.SetBorders("insideV", "none", 0, "000000")

	// ========== 示例2：带有跨页的三线表 ==========
	doc.AddParagraph().SetSpacingBefore(400) // 添加空白间隔

	// 添加表格标题
	tableTitlePara2 := doc.AddParagraph()
	tableTitlePara2.SetAlignment("center")
	tableTitlePara2.SetSpacingAfter(0)
	tableTitlePara2.SetSpacingBefore(0)
	tableTitlePara2.SetLineSpacing(1.5, "auto") // 设置1.5倍行距
	tableTitleRun2 := tableTitlePara2.AddRun()
	tableTitleRun2.AddText("表4-2 学生成绩信息表")
	tableTitleRun2.SetBold(true)
	tableTitleRun2.SetFontSize(21) // 五号字体
	tableTitleRun2.SetFontFamily("宋体")
	// 表序号设置为Times New Roman
	tableTitleRun2.SetFontFamilyForRunes("Times New Roman", []rune("表4-2"))

	// 创建第二个表格
	table2 := doc.AddTable(6, 4)
	table2.SetWidth("100%", "pct") // 与文字齐宽
	table2.SetAlignment("center")

	// 设置行高为0.72厘米（固定值）
	for i := 0; i < 6; i++ {
		table2.Rows[i].SetHeight(567, "exact") // 0.72厘米 ≈ 567 twip，"exact"表示固定行高
	}

	// 填充表头
	headers2 := []string{"学号", "姓名", "科目", "成绩"}
	for i, header := range headers2 {
		cellPara := table2.Rows[0].Cells[i].AddParagraph()
		cellPara.SetAlignment("center")
		cellPara.SetLineSpacing(1.5, "auto") // 1.5倍行距
		cellRun := cellPara.AddRun()
		cellRun.AddText(header)
		cellRun.SetBold(false)
		cellRun.SetFontSize(21) // 五号字体
		cellRun.SetFontFamily("宋体")
	}

	// 填充数据行
	data2 := [][]string{
		{"2020001", "张三", "数据库", "85"},
		{"2020002", "李四", "数据库", "92"},
		{"2020003", "王五", "数据库", "78"},
		{"2020004", "赵六", "数据库", "88"},
		{"2020005", "钱七", "数据库", "95"},
	}

	for i, row := range data2 {
		for j, cell := range row {
			para := table2.Rows[i+1].Cells[j].AddParagraph()
			para.SetAlignment("center")
			para.SetLineSpacing(1.5, "auto") // 1.5倍行距
			cellRun := para.AddRun()
			cellRun.AddText(cell)
			cellRun.SetFontSize(21) // 五号字体
			cellRun.SetFontFamily("宋体")
			// 英文和数字设置为Times New Roman
			if j == 0 || j == 3 { // 学号和成绩列
				cellRun.SetFontFamilyForRunes("Times New Roman", []rune(cell))
			}
		}
	}

	// 设置三线表样式
	// 1. 首先清除所有默认边框
	table2.SetBorders("all", "", 0, "")

	// 2. 顶线（表格顶部边框），1.5磅
	table2.SetBorders("top", "single", 24, "000000")

	// 3. 表头分隔线（第一行底部边框），1磅
	for i := 0; i < 4; i++ {
		table2.Rows[0].Cells[i].SetBorders("bottom", "single", 16, "000000")
	}

	// 4. 底线（表格底部边框），1.5磅
	table2.SetBorders("bottom", "single", 24, "000000")

	// 显式设置内部边框为"none"，而不是空字符串
	table2.SetBorders("insideH", "none", 0, "000000")
	table2.SetBorders("insideV", "none", 0, "000000")

	// 演示跨页表格续表标题
	doc.AddPageBreak()

	// 添加续表标题
	tableTitlePara3 := doc.AddParagraph()
	tableTitlePara3.SetAlignment("center")
	tableTitlePara3.SetSpacingAfter(0)
	tableTitlePara3.SetSpacingBefore(0)
	tableTitlePara3.SetLineSpacing(1.5, "auto")
	tableTitleRun3 := tableTitlePara3.AddRun()
	tableTitleRun3.AddText("表4-2 学生成绩信息表（续）")
	tableTitleRun3.SetBold(true)
	tableTitleRun3.SetFontSize(21)
	tableTitleRun3.SetFontFamily("宋体")
	tableTitleRun3.SetFontFamilyForRunes("Times New Roman", []rune("表4-2"))

	// 创建续表
	table3 := doc.AddTable(6, 4)
	table3.SetWidth("100%", "pct")
	table3.SetAlignment("center")

	// 设置行高为0.72厘米（固定值）
	for i := 0; i < 6; i++ {
		table3.Rows[i].SetHeight(567, "exact") // 0.72厘米 ≈ 567 twip，"exact"表示固定行高
	}

	// 填充表头
	for i, header := range headers2 {
		cellPara := table3.Rows[0].Cells[i].AddParagraph()
		cellPara.SetAlignment("center")
		cellPara.SetLineSpacing(1.5, "auto")
		cellRun := cellPara.AddRun()
		cellRun.AddText(header)
		cellRun.SetBold(false)
		cellRun.SetFontSize(21)
		cellRun.SetFontFamily("宋体")
	}

	// 填充续表数据
	data3 := [][]string{
		{"2020006", "孙八", "数据库", "82"},
		{"2020007", "周九", "数据库", "90"},
		{"2020008", "吴十", "数据库", "87"},
		{"2020009", "郑十一", "数据库", "91"},
		{"2020010", "王十二", "数据库", "84"},
	}

	for i, row := range data3 {
		for j, cell := range row {
			para := table3.Rows[i+1].Cells[j].AddParagraph()
			para.SetAlignment("center")
			para.SetLineSpacing(1.5, "auto")
			cellRun := para.AddRun()
			cellRun.AddText(cell)
			cellRun.SetFontSize(21)
			cellRun.SetFontFamily("宋体")
			if j == 0 || j == 3 {
				cellRun.SetFontFamilyForRunes("Times New Roman", []rune(cell))
			}
		}
	}

	// 设置三线表样式
	table3.SetBorders("all", "", 0, "")
	table3.SetBorders("top", "single", 24, "000000")
	for i := 0; i < 4; i++ {
		table3.Rows[0].Cells[i].SetBorders("bottom", "single", 16, "000000")
	}
	table3.SetBorders("bottom", "single", 24, "000000")

	// 显式设置内部边框为"none"，而不是空字符串
	table3.SetBorders("insideH", "none", 0, "000000")
	table3.SetBorders("insideV", "none", 0, "000000")

	// 添加页脚（页码）
	footer := doc.AddFooterWithReference("default")
	footer.AddPageNumber()

	// 保存文档
	err := doc.Save("./document/examples/threeline_table/threeline_table_example.docx")
	if err != nil {
		log.Fatalf("保存文档时出错: %v", err)
	}

	fmt.Println("数据库三线表示例文档已成功保存为 threeline_table_example.docx")
}