OrgGo/unioffice
1
0
Fork 0
mirror of https://github.com/unidoc/unioffice.git synced 2025-05-04 22:17:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

spreadsheet: add saving/loading of themes

Browse Source
This commit is contained in:
Todd 2017-09-02 16:33:40 -05:00
parent 8f45ea019b
commit a06fea99f7
2 changed files with 30 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
document/document.go
View File

@ -20,11 +20,11 @@ import (
"path/filepath" "path/filepath"
"baliance.com/gooxml/common" "baliance.com/gooxml/common"
"baliance.com/gooxml/zippkg"
dml "baliance.com/gooxml/schema/schemas.openxmlformats.org/drawingml" dml "baliance.com/gooxml/schema/schemas.openxmlformats.org/drawingml"
st "baliance.com/gooxml/schema/schemas.openxmlformats.org/officeDocument/2006/sharedTypes" st "baliance.com/gooxml/schema/schemas.openxmlformats.org/officeDocument/2006/sharedTypes"
wml "baliance.com/gooxml/schema/schemas.openxmlformats.org/wordprocessingml" wml "baliance.com/gooxml/schema/schemas.openxmlformats.org/wordprocessingml"
"baliance.com/gooxml/zippkg"
) )
// Document is a text document that can be written out in the OOXML .docx // Document is a text document that can be written out in the OOXML .docx
@ -189,7 +189,6 @@ func (d *Document) Save(w io.Writer) error {
if err := zippkg.MarshalXML(z, fmt.Sprintf("word/theme/theme%d.xml", i+1), thm); err != nil { if err := zippkg.MarshalXML(z, fmt.Sprintf("word/theme/theme%d.xml", i+1), thm); err != nil {
return err return err
} }
} }
for i, hdr := range d.headers { for i, hdr := range d.headers {
fn := fmt.Sprintf("word/header%d.xml", i+1) fn := fmt.Sprintf("word/header%d.xml", i+1)

43
spreadsheet/workbook.go
View File

@ -20,27 +20,30 @@ import (
"path/filepath" "path/filepath"
"baliance.com/gooxml/common" "baliance.com/gooxml/common"
"baliance.com/gooxml/schema/schemas.openxmlformats.org/spreadsheetml"
"baliance.com/gooxml/zippkg" "baliance.com/gooxml/zippkg"
dml "baliance.com/gooxml/schema/schemas.openxmlformats.org/drawingml"
sml "baliance.com/gooxml/schema/schemas.openxmlformats.org/spreadsheetml"
) )
// Workbook is the top level container item for a set of spreadsheets. // Workbook is the top level container item for a set of spreadsheets.
type Workbook struct { type Workbook struct {
common.DocBase common.DocBase
x *spreadsheetml.Workbook x *sml.Workbook
StyleSheet StyleSheet StyleSheet StyleSheet
Theme common.Theme
SharedStrings SharedStrings SharedStrings SharedStrings
xws []*spreadsheetml.Worksheet
xws []*sml.Worksheet
xwsRels []common.Relationships xwsRels []common.Relationships
wbRels common.Relationships wbRels common.Relationships
themes []*dml.Theme
} }
// New constructs a new workbook. // New constructs a new workbook.
func New() *Workbook { func New() *Workbook {
wb := &Workbook{} wb := &Workbook{}
wb.x = spreadsheetml.NewWorkbook() wb.x = sml.NewWorkbook()
wb.AppProperties = common.NewAppProperties() wb.AppProperties = common.NewAppProperties()
wb.CoreProperties = common.NewCoreProperties() wb.CoreProperties = common.NewCoreProperties()
@ -54,8 +57,8 @@ func New() *Workbook {
wb.wbRels.AddRelationship("styles.xml", common.StylesType) wb.wbRels.AddRelationship("styles.xml", common.StylesType)
wb.ContentTypes = common.NewContentTypes() wb.ContentTypes = common.NewContentTypes()
wb.ContentTypes.AddOverride("/xl/workbook.xml", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet.main+xml") wb.ContentTypes.AddOverride("/xl/workbook.xml", "application/vnd.openxmlformats-officedocument.sml.sheet.main+xml")
wb.ContentTypes.AddOverride("/xl/styles.xml", "application/vnd.openxmlformats-officedocument.spreadsheetml.styles+xml") wb.ContentTypes.AddOverride("/xl/styles.xml", "application/vnd.openxmlformats-officedocument.sml.styles+xml")
wb.SharedStrings = NewSharedStrings() wb.SharedStrings = NewSharedStrings()
wb.ContentTypes.AddOverride("/xl/sharedStrings.xml", common.SharedStringsContentType) wb.ContentTypes.AddOverride("/xl/sharedStrings.xml", common.SharedStringsContentType)
@ -115,7 +118,7 @@ func Read(r io.ReaderAt, size int64) (*Workbook, error) {
for _, r := range wb.Rels.Relationships() { for _, r := range wb.Rels.Relationships() {
switch r.Type() { switch r.Type() {
case common.OfficeDocumentType: case common.OfficeDocumentType:
wb.x = spreadsheetml.NewWorkbook() wb.x = sml.NewWorkbook()
decMap[r.Target()] = wb.x decMap[r.Target()] = wb.x
// look for the workbook relationships file as well // look for the workbook relationships file as well
basePath, _ := filepath.Split(r.Target()) basePath, _ := filepath.Split(r.Target())
@ -157,7 +160,7 @@ func Read(r io.ReaderAt, size int64) (*Workbook, error) {
for _, r := range wb.wbRels.Relationships() { for _, r := range wb.wbRels.Relationships() {
switch r.Type() { switch r.Type() {
case common.WorksheetType: case common.WorksheetType:
ws := spreadsheetml.NewWorksheet() ws := sml.NewWorksheet()
wb.xws = append(wb.xws, ws) wb.xws = append(wb.xws, ws)
decMap[basePaths[wb.wbRels]+r.Target()] = ws decMap[basePaths[wb.wbRels]+r.Target()] = ws
@ -171,8 +174,9 @@ func Read(r io.ReaderAt, size int64) (*Workbook, error) {
wb.StyleSheet = NewStyleSheet() wb.StyleSheet = NewStyleSheet()
decMap[basePaths[wb.wbRels]+r.Target()] = wb.StyleSheet.X() decMap[basePaths[wb.wbRels]+r.Target()] = wb.StyleSheet.X()
case common.ThemeType: case common.ThemeType:
wb.Theme = common.NewTheme() thm := dml.NewTheme()
decMap[basePaths[wb.wbRels]+r.Target()] = wb.Theme.X() wb.themes = append(wb.themes, thm)
decMap[basePaths[wb.wbRels]+r.Target()] = thm
case common.SharedStingsType: case common.SharedStingsType:
wb.SharedStrings = NewSharedStrings() wb.SharedStrings = NewSharedStrings()
decMap[basePaths[wb.wbRels]+r.Target()] = wb.SharedStrings.X() decMap[basePaths[wb.wbRels]+r.Target()] = wb.SharedStrings.X()
@ -197,13 +201,13 @@ func Read(r io.ReaderAt, size int64) (*Workbook, error) {
} }
// X returns the inner wrapped XML type. // X returns the inner wrapped XML type.
func (wb *Workbook) X() *spreadsheetml.Workbook { func (wb *Workbook) X() *sml.Workbook {
return wb.x return wb.x
} }
// AddSheet adds a new sheet with a given name to a workbook. // AddSheet adds a new sheet with a given name to a workbook.
func (wb *Workbook) AddSheet() Sheet { func (wb *Workbook) AddSheet() Sheet {
rs := spreadsheetml.NewCT_Sheet() rs := sml.NewCT_Sheet()
// Assign a unique sheet ID // Assign a unique sheet ID
rs.SheetIdAttr = 1 rs.SheetIdAttr = 1
@ -217,12 +221,12 @@ func (wb *Workbook) AddSheet() Sheet {
rs.NameAttr = fmt.Sprintf("Sheet %d", rs.SheetIdAttr) rs.NameAttr = fmt.Sprintf("Sheet %d", rs.SheetIdAttr)
// create the actual worksheet // create the actual worksheet
ws := spreadsheetml.NewWorksheet() ws := sml.NewWorksheet()
ws.Dimension = spreadsheetml.NewCT_SheetDimension() ws.Dimension = sml.NewCT_SheetDimension()
ws.Dimension.RefAttr = "A1" ws.Dimension.RefAttr = "A1"
wb.xws = append(wb.xws, ws) wb.xws = append(wb.xws, ws)
ws.SheetData = spreadsheetml.NewCT_SheetData() ws.SheetData = sml.NewCT_SheetData()
// update the references // update the references
rid := wb.wbRels.AddRelationship(fmt.Sprintf("worksheets/sheet%d.xml", rs.SheetIdAttr), rid := wb.wbRels.AddRelationship(fmt.Sprintf("worksheets/sheet%d.xml", rs.SheetIdAttr),
@ -231,7 +235,7 @@ func (wb *Workbook) AddSheet() Sheet {
// add the content type // add the content type
wb.ContentTypes.AddOverride(fmt.Sprintf("/xl/worksheets/sheet%d.xml", rs.SheetIdAttr), wb.ContentTypes.AddOverride(fmt.Sprintf("/xl/worksheets/sheet%d.xml", rs.SheetIdAttr),
"application/vnd.openxmlformats-officedocument.spreadsheetml.worksheet+xml") "application/vnd.openxmlformats-officedocument.sml.worksheet+xml")
return Sheet{wb, rs, ws} return Sheet{wb, rs, ws}
} }
@ -274,6 +278,11 @@ func (wb *Workbook) Save(w io.Writer) error {
if err := zippkg.MarshalXML(z, "xl/_rels/workbook.xml.rels", wb.wbRels.X()); err != nil { if err := zippkg.MarshalXML(z, "xl/_rels/workbook.xml.rels", wb.wbRels.X()); err != nil {
return err return err
} }
for i, thm := range wb.themes {
if err := zippkg.MarshalXML(z, fmt.Sprintf("xl/theme/theme%d.xml", i+1), thm); err != nil {
return err
}
}
for i, sheet := range wb.xws { for i, sheet := range wb.xws {
wbs := wb.x.Sheets.Sheet[i] wbs := wb.x.Sheets.Sheet[i]
zippkg.MarshalXML(z, fmt.Sprintf("xl/worksheets/sheet%d.xml", wbs.SheetIdAttr), sheet) zippkg.MarshalXML(z, fmt.Sprintf("xl/worksheets/sheet%d.xml", wbs.SheetIdAttr), sheet)