// Copyright 2025-2026 肖其顿 (XIAO QI DUN)
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package ofdgo

import (
	"bytes"
	"encoding/xml"
	"fmt"
	"reflect"
)

// sourceInfoXML 更新文档描述字段，保留未识别内容
// 返回: []byte 根索引XML, error 错误信息
func (e *Editor) sourceInfoXML() ([]byte, error) {
	data, err := e.source.reader.readFile("OFD.xml")
	if err != nil {
		return nil, err
	}
	return editorUpdateInfoXML(data, e.source.fallbackDocID, e.source.reader.documentIndex, e.source.info, e.Info)
}

// editorUpdateInfoXML 更新指定文档的描述和自定义字段，保留未修改内容
// 入参: data 根索引XML, docID 缺少描述时使用的标识, index 文档索引, before 原信息, after 新信息
// 返回: []byte 修改后的XML, error 解析错误
func editorUpdateInfoXML(data []byte, docID string, index int, before, after DocInfo) ([]byte, error) {
	data, info, err := editorDocInfoXML(data, docID, index)
	if err != nil {
		return nil, err
	}
	var values [][2]string
	for _, field := range [][3]string{
		{"DocID", before.DocID, after.DocID}, {"Title", before.Title, after.Title}, {"Author", before.Author, after.Author},
		{"Subject", before.Subject, after.Subject}, {"Abstract", before.Abstract, after.Abstract},
		{"CreationDate", before.CreationDate, after.CreationDate}, {"ModDate", before.ModDate, after.ModDate},
		{"Creator", before.Creator, after.Creator}, {"CreatorVersion", before.CreatorVersion, after.CreatorVersion},
	} {
		if field[1] != field[2] {
			values = append(values, [2]string{field[0], field[2]})
		}
	}
	data, err = editorXMLSetText(data, info, values)
	if err != nil {
		return nil, err
	}
	if !reflect.DeepEqual(before.CustomDatas, after.CustomDatas) {
		root, err := parseEditorXML(data)
		if err != nil {
			return nil, err
		}
		return editorCustomDatas(data, root.childAt("DocBody", index).child("DocInfo"), after.CustomDatas)
	}
	return data, nil
}

// editorCreatorXML 更新指定文档的制作软件，移除原软件版本，保留其余根索引内容
// 入参: data 当前根索引XML, docID 缺少文档描述时使用的标识, index 文档索引
// 返回: []byte 修改后的XML, error 解析错误
func editorCreatorXML(data []byte, docID string, index int) ([]byte, error) {
	data, info, err := editorDocInfoXML(data, docID, index)
	if err != nil {
		return nil, err
	}
	return editorXMLSetText(data, info, [][2]string{{"Creator", ofdCreator}, {"CreatorVersion", ""}})
}

// editorDocInfoXML 获取文档描述，修改缺少描述的源文件时补建标准容器及标识
// 入参: data 根索引XML, docID 缺少文档描述时使用的标识, index 文档索引
// 返回: []byte 根索引XML, *editorXML 描述节点, error 解析错误
func editorDocInfoXML(data []byte, docID string, index int) ([]byte, *editorXML, error) {
	root, err := parseEditorXML(data)
	if err != nil {
		return nil, nil, err
	}
	body := root.childAt("DocBody", index)
	if body == nil {
		return nil, nil, fmt.Errorf("document has no DocBody")
	}
	if info := body.child("DocInfo"); info != nil {
		return data, info, nil
	}
	info, err := body.containerXML("DocInfo", nil, body.textXML("DocID", docID))
	if err != nil {
		return nil, nil, err
	}
	if body.open == body.end {
		data = editorPatchXML(data, []editorXMLPatch{editorXMLContent(data, body, info)})
	} else {
		data = editorPatchXML(data, []editorXMLPatch{{body.open, body.open, info}})
	}
	root, err = parseEditorXML(data)
	if err != nil {
		return nil, nil, err
	}
	return data, root.childAt("DocBody", index).child("DocInfo"), nil
}

// editorCustomDatas 更新自定义字段，保留原字段及容器的扩展内容
// 入参: data 原XML, info 文档信息节点, values 新字段
// 返回: []byte 修改后的XML, error 编码错误
func editorCustomDatas(data []byte, info *editorXML, values *CustomDatas) ([]byte, error) {
	container := info.child("CustomDatas")
	context := info
	if container != nil {
		context = container
	}
	var items []CustomData
	if values != nil {
		items = values.CustomData
	}
	byName := make(map[string][]*editorXML)
	var nodes []*editorXML
	if container != nil {
		for _, node := range container.children {
			if node.name.Local == "CustomData" && (node.name.Space == ofdNamespace2016 || node.name.Space == container.name.Space) {
				nodes = append(nodes, node)
				byName[node.attr("Name")] = append(byName[node.attr("Name")], node)
			}
		}
	}
	var encoded [][]byte
	used := make(map[*editorXML]bool)
	for _, item := range items {
		if item.sourceIndex > 0 && item.sourceIndex <= len(nodes) {
			used[nodes[item.sourceIndex-1]] = true
		}
	}
	for _, item := range items {
		var node *editorXML
		if item.sourceIndex > 0 && item.sourceIndex <= len(nodes) {
			node = nodes[item.sourceIndex-1]
		} else {
			for _, candidate := range byName[item.Name] {
				if !used[candidate] {
					node = candidate
					used[node] = true
					break
				}
			}
		}
		if node == nil {
			var value bytes.Buffer
			_ = xml.EscapeText(&value, []byte(item.Value))
			entry, err := context.containerXML("CustomData", ofdAttrs{{Name: xml.Name{Local: "Name"}, Value: item.Name}}, value.Bytes())
			if err != nil {
				return nil, err
			}
			encoded = append(encoded, entry)
			continue
		}
		var original CustomData
		if err := xml.Unmarshal(data[node.start:node.end], &original); err != nil {
			return nil, err
		}
		if original.Name != item.Name {
			entry, err := editorXMLAttribute(data, node, "Name", item.Name)
			if err != nil {
				return nil, err
			}
			renamed, err := parseEditorXML(entry)
			if err != nil {
				return nil, err
			}
			updated, err := editorXMLTextValue(entry, renamed, original.Value, item.Value)
			if err != nil {
				return nil, err
			}
			encoded = append(encoded, updated)
			continue
		}
		entry, err := editorXMLTextValue(data, node, original.Value, item.Value)
		if err != nil {
			return nil, err
		}
		encoded = append(encoded, entry)
	}
	if container == nil {
		if len(encoded) == 0 {
			return data, nil
		}
		content, err := info.containerXML("CustomDatas", nil, bytes.Join(encoded, nil))
		if err != nil {
			return nil, err
		}
		if info.open == info.end {
			return editorPatchXML(data, []editorXMLPatch{editorXMLContent(data, info, content)}), nil
		}
		return editorPatchXML(data, []editorXMLPatch{{info.close, info.close, content}}), nil
	}
	var patches []editorXMLPatch
	for i, node := range nodes {
		var entry []byte
		if i < len(encoded) {
			entry = encoded[i]
		}
		patches = append(patches, editorXMLPatch{node.start, node.end, entry})
	}
	if len(encoded) > len(nodes) {
		added := bytes.Join(encoded[len(nodes):], nil)
		if container.open == container.end {
			patches = append(patches, editorXMLContent(data, container, added))
		} else {
			patches = append(patches, editorXMLPatch{container.close, container.close, added})
		}
	}
	return editorPatchXML(data, patches), nil
}