package document

import (
	"fmt"
)

// Run 表示Word文档中的文本运行
type Run struct {
	Text       string
	Properties *RunProperties
	BreakType  string
	Drawing    *Drawing
	Field      *Field
}

// Field 表示Word文档中的域
type Field struct {
	Type string // begin, separate, end
	Code string // 域代码
}

// RunProperties 表示文本运行的属性
type RunProperties struct {
	Bold             bool     // 粗体
	Italic           bool     // 斜体
	Underline        string   // 下划线类型：single, double, thick, dotted, dash, etc.
	Strike           bool     // 删除线
	DoubleStrike     bool     // 双删除线
	Superscript      bool     // 上标
	Subscript        bool     // 下标
	FontSize         int      // 字号，单位为半点
	FontFamily       string   // 字体
	Color            string   // 颜色，格式为RRGGBB
	Highlight        string   // 突出显示颜色
	Caps             bool     // 全部大写
	SmallCaps        bool     // 小型大写
	CharacterSpacing int      // 字符间距
	Shading          *Shading // 底纹
	VertAlign        string   // 垂直对齐方式：baseline, superscript, subscript
	RTL              bool     // 从右到左文本方向
	Language         string   // 语言
}

// BreakType 表示分隔符类型
const (
	BreakTypePage    = "page"         // 分页符
	BreakTypeColumn  = "column"       // 分栏符
	BreakTypeSection = "section"      // 分节符
	BreakTypeLine    = "textWrapping" // 换行符
)

// NewRun 创建一个新的文本运行
func NewRun() *Run {
	return &Run{
		Text: "",
		Properties: &RunProperties{
			FontSize:   22, // 默认11磅 (22半点)
			FontFamily: "Calibri",
			Color:      "000000",
		},
	}
}

// AddText 向文本运行添加文本
func (r *Run) AddText(text string) *Run {
	r.Text = text
	return r
}

// AddBreak 向文本运行添加分隔符
func (r *Run) AddBreak(breakType string) *Run {
	r.BreakType = breakType
	return r
}

// AddDrawing 向文本运行添加图形
func (r *Run) AddDrawing(drawing *Drawing) *Run {
	r.Drawing = drawing
	return r
}

// SetBold 设置粗体
func (r *Run) SetBold(bold bool) *Run {
	r.Properties.Bold = bold
	return r
}

// SetItalic 设置斜体
func (r *Run) SetItalic(italic bool) *Run {
	r.Properties.Italic = italic
	return r
}

// SetUnderline 设置下划线
func (r *Run) SetUnderline(underline string) *Run {
	r.Properties.Underline = underline
	return r
}

// SetStrike 设置删除线
func (r *Run) SetStrike(strike bool) *Run {
	r.Properties.Strike = strike
	return r
}

// SetDoubleStrike 设置双删除线
func (r *Run) SetDoubleStrike(doubleStrike bool) *Run {
	r.Properties.DoubleStrike = doubleStrike
	return r
}

// SetSuperscript 设置上标
func (r *Run) SetSuperscript(superscript bool) *Run {
	r.Properties.Superscript = superscript
	return r
}

// SetSubscript 设置下标
func (r *Run) SetSubscript(subscript bool) *Run {
	r.Properties.Subscript = subscript
	return r
}

// SetFontSize 设置字号
func (r *Run) SetFontSize(fontSize int) *Run {
	r.Properties.FontSize = fontSize
	return r
}

// SetFontFamily 设置字体
func (r *Run) SetFontFamily(fontFamily string) *Run {
	r.Properties.FontFamily = fontFamily
	return r
}

// SetColor 设置颜色
func (r *Run) SetColor(color string) *Run {
	r.Properties.Color = color
	return r
}

// SetHighlight 设置突出显示颜色
func (r *Run) SetHighlight(highlight string) *Run {
	r.Properties.Highlight = highlight
	return r
}

// SetCaps 设置全部大写
func (r *Run) SetCaps(caps bool) *Run {
	r.Properties.Caps = caps
	return r
}

// SetSmallCaps 设置小型大写
func (r *Run) SetSmallCaps(smallCaps bool) *Run {
	r.Properties.SmallCaps = smallCaps
	return r
}

// SetCharacterSpacing 设置字符间距
func (r *Run) SetCharacterSpacing(spacing int) *Run {
	r.Properties.CharacterSpacing = spacing
	return r
}

// SetShading 设置底纹
func (r *Run) SetShading(fill, color, pattern string) *Run {
	r.Properties.Shading = &Shading{
		Fill:    fill,
		Color:   color,
		Pattern: pattern,
	}
	return r
}

// SetVertAlign 设置垂直对齐方式
func (r *Run) SetVertAlign(vertAlign string) *Run {
	r.Properties.VertAlign = vertAlign
	return r
}

// SetRTL 设置从右到左文本方向
func (r *Run) SetRTL(rtl bool) *Run {
	r.Properties.RTL = rtl
	return r
}

// SetLanguage 设置语言
func (r *Run) SetLanguage(language string) *Run {
	r.Properties.Language = language
	return r
}

// AddField 添加Word域
func (r *Run) AddField(fieldType string, fieldCode string) *Run {
	r.Text = ""
	r.Field = &Field{
		Type: fieldType,
		Code: fieldCode,
	}
	return r
}

// AddPageNumber 添加页码域
func (r *Run) AddPageNumber() *Run {
	return r.AddField("begin", " PAGE ")
}

// ToXML 将Run转换为XML
func (r *Run) ToXML() string {
	xml := "<w:r>"

	// 添加运行属性 - 严格按照Word XML规范顺序
	xml += "<w:rPr>"

	// 运行样式（缺失，预留位置）

	// 1. 字体
	if r.Properties.FontFamily != "" {
		xml += fmt.Sprintf("<w:rFonts w:ascii=\"%s\" w:hAnsi=\"%s\" w:eastAsia=\"%s\" w:cs=\"%s\" />",
			r.Properties.FontFamily,
			r.Properties.FontFamily,
			r.Properties.FontFamily,
			r.Properties.FontFamily)
	}

	// 2. 加粗
	if r.Properties.Bold {
		xml += "<w:b />"
		// 3. 复杂文本加粗
		xml += "<w:bCs />"
	}

	// 4. 斜体
	if r.Properties.Italic {
		xml += "<w:i />"
		// 5. 复杂文本斜体
		xml += "<w:iCs />"
	}

	// 6. 全部大写
	if r.Properties.Caps {
		xml += "<w:caps />"
	}

	// 7. 小型大写
	if r.Properties.SmallCaps {
		xml += "<w:smallCaps />"
	}

	// 8. 删除线
	if r.Properties.Strike {
		xml += "<w:strike />"
	}

	// 9. 双删除线
	if r.Properties.DoubleStrike {
		xml += "<w:dstrike />"
	}

	// 10-18. 其他格式（outline, shadow, emboss等，缺失，预留位置）

	// 19. 颜色
	if r.Properties.Color != "" {
		xml += fmt.Sprintf("<w:color w:val=\"%s\" />", r.Properties.Color)
	}

	// 20. 字符间距
	if r.Properties.CharacterSpacing != 0 {
		xml += fmt.Sprintf("<w:spacing w:val=\"%d\" />", r.Properties.CharacterSpacing)
	}

	// 21-23. 其他属性（缺失，预留位置）

	// 24. 字号
	if r.Properties.FontSize > 0 {
		xml += fmt.Sprintf("<w:sz w:val=\"%d\" />", r.Properties.FontSize)
		// 25. 复杂文本字号
		xml += fmt.Sprintf("<w:szCs w:val=\"%d\" />", r.Properties.FontSize)
	}

	// 26. 突出显示
	if r.Properties.Highlight != "" {
		xml += fmt.Sprintf("<w:highlight w:val=\"%s\" />", r.Properties.Highlight)
	}

	// 27. 下划线
	if r.Properties.Underline != "" {
		xml += fmt.Sprintf("<w:u w:val=\"%s\" />", r.Properties.Underline)
	}

	// 28-31. 其他属性（缺失，预留位置）

	// 32. 上标/下标
	if r.Properties.Superscript {
		xml += "<w:vertAlign w:val=\"superscript\" />"
	} else if r.Properties.Subscript {
		xml += "<w:vertAlign w:val=\"subscript\" />"
	} else if r.Properties.VertAlign != "" {
		xml += fmt.Sprintf("<w:vertAlign w:val=\"%s\" />", r.Properties.VertAlign)
	}

	// 33. 从右到左文本方向
	if r.Properties.RTL {
		xml += "<w:rtl />"
	}

	// 34-35. 其他属性（缺失，预留位置）

	// 36. 语言
	if r.Properties.Language != "" {
		xml += fmt.Sprintf("<w:lang w:val=\"%s\" />", r.Properties.Language)
	}

	// 底纹（shd应该在位置30）
	if r.Properties.Shading != nil {
		xml += fmt.Sprintf("<w:shd w:val=\"%s\" w:fill=\"%s\" w:color=\"%s\" />",
			r.Properties.Shading.Pattern,
			r.Properties.Shading.Fill,
			r.Properties.Shading.Color)
	}

	xml += "</w:rPr>"

	// 添加分隔符
	if r.BreakType != "" {
		xml += fmt.Sprintf("<w:br w:type=\"%s\" />", r.BreakType)
	}

	// 添加文本
	if r.Text != "" {
		xml += fmt.Sprintf("<w:t xml:space=\"preserve\">%s</w:t>", r.Text)
	}

	// 添加图形
	if r.Drawing != nil {
		xml += r.Drawing.ToXML()
	}

	// 添加域
	if r.Field != nil {
		if r.Field.Type == "begin" {
			xml += "<w:fldChar w:fldCharType=\"begin\" />"
		} else if r.Field.Type == "separate" {
			xml += "<w:fldChar w:fldCharType=\"separate\" />"
		} else if r.Field.Type == "end" {
			xml += "<w:fldChar w:fldCharType=\"end\" />"
		}

		if r.Field.Code != "" && r.Field.Type == "begin" {
			// 添加域代码
			xml += "</w:r><w:r><w:instrText xml:space=\"preserve\">" + r.Field.Code + "</w:instrText></w:r><w:r><w:fldChar w:fldCharType=\"separate\" />"
			// 添加域结束标记
			xml += "</w:r><w:r><w:fldChar w:fldCharType=\"end\" />"
		}
	}

	xml += "</w:r>"
	return xml
}