mirror of
https://github.com/fumiama/go-docx.git
synced 2026-06-04 23:30:25 +08:00
86 lines
2.4 KiB
Go
86 lines
2.4 KiB
Go
/*
|
|
Copyright (c) 2020 gingfrederik
|
|
Copyright (c) 2021 Gonzalo Fernandez-Victorio
|
|
Copyright (c) 2021 Basement Crowd Ltd (https://www.basementcrowd.com)
|
|
Copyright (c) 2023 Fumiama Minamoto (源文雨)
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Affero General Public License as published
|
|
by the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Affero General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Affero General Public License
|
|
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
package docx
|
|
|
|
import (
|
|
"errors"
|
|
"strconv"
|
|
"sync/atomic"
|
|
)
|
|
|
|
var (
|
|
// ErrRefIDNotFound cannot find such reference
|
|
ErrRefIDNotFound = errors.New("ref id not found")
|
|
// ErrRefTargetNotFound cannot find such target
|
|
ErrRefTargetNotFound = errors.New("ref target not found")
|
|
)
|
|
|
|
// when adding an hyperlink we need to store a reference in the relationship field
|
|
//
|
|
// this func is not thread-safe
|
|
func (f *Docx) addLinkRelation(link string) string {
|
|
rel := Relationship{
|
|
ID: "rId" + strconv.Itoa(int(atomic.AddUintptr(&f.rID, 1))),
|
|
Type: REL_HYPERLINK,
|
|
Target: link,
|
|
TargetMode: REL_TARGETMODE,
|
|
}
|
|
|
|
f.docRelation.Relationship = append(f.docRelation.Relationship, rel)
|
|
|
|
return rel.ID
|
|
}
|
|
|
|
// when adding an image we need to store a reference in the relationship field
|
|
//
|
|
// this func is not thread-safe
|
|
func (f *Docx) addImageRelation(m Media) string {
|
|
rel := Relationship{
|
|
ID: "rId" + strconv.Itoa(int(atomic.AddUintptr(&f.rID, 1))),
|
|
Type: REL_IMAGE,
|
|
Target: "media/" + m.Name,
|
|
}
|
|
|
|
f.docRelation.Relationship = append(f.docRelation.Relationship, rel)
|
|
|
|
return rel.ID
|
|
}
|
|
|
|
// ReferTarget gets the target for a reference
|
|
func (f *Docx) ReferTarget(id string) (string, error) {
|
|
for _, a := range f.docRelation.Relationship {
|
|
if a.ID == id {
|
|
return a.Target, nil
|
|
}
|
|
}
|
|
return "", ErrRefIDNotFound
|
|
}
|
|
|
|
// ReferID gets the rId from target
|
|
func (f *Docx) ReferID(target string) (string, error) {
|
|
for _, a := range f.docRelation.Relationship {
|
|
if a.Target == target {
|
|
return a.ID, nil
|
|
}
|
|
}
|
|
return "", ErrRefIDNotFound
|
|
}
|