lib.go 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864
  1. package xlsx
  2. import (
  3. "archive/zip"
  4. "encoding/xml"
  5. "errors"
  6. "fmt"
  7. "io"
  8. "path"
  9. "strconv"
  10. "strings"
  11. )
  12. // XLSXReaderError is the standard error type for otherwise undefined
  13. // errors in the XSLX reading process.
  14. type XLSXReaderError struct {
  15. Err string
  16. }
  17. // String() returns a string value from an XLSXReaderError struct in
  18. // order that it might comply with the os.Error interface.
  19. func (e *XLSXReaderError) Error() string {
  20. return e.Err
  21. }
  22. // getRangeFromString is an internal helper function that converts
  23. // XLSX internal range syntax to a pair of integers. For example,
  24. // the range string "1:3" yield the upper and lower intergers 1 and 3.
  25. func getRangeFromString(rangeString string) (lower int, upper int, error error) {
  26. var parts []string
  27. parts = strings.SplitN(rangeString, ":", 2)
  28. if parts[0] == "" {
  29. error = errors.New(fmt.Sprintf("Invalid range '%s'\n", rangeString))
  30. }
  31. if parts[1] == "" {
  32. error = errors.New(fmt.Sprintf("Invalid range '%s'\n", rangeString))
  33. }
  34. lower, error = strconv.Atoi(parts[0])
  35. if error != nil {
  36. error = errors.New(fmt.Sprintf("Invalid range (not integer in lower bound) %s\n", rangeString))
  37. }
  38. upper, error = strconv.Atoi(parts[1])
  39. if error != nil {
  40. error = errors.New(fmt.Sprintf("Invalid range (not integer in upper bound) %s\n", rangeString))
  41. }
  42. return lower, upper, error
  43. }
  44. // lettersToNumeric is used to convert a character based column
  45. // reference to a zero based numeric column identifier.
  46. func lettersToNumeric(letters string) int {
  47. sum, mul, n := 0, 1, 0
  48. for i := len(letters) - 1; i >= 0; i, mul, n = i-1, mul*26, 1 {
  49. c := letters[i]
  50. switch {
  51. case 'A' <= c && c <= 'Z':
  52. n += int(c - 'A')
  53. case 'a' <= c && c <= 'z':
  54. n += int(c - 'a')
  55. }
  56. sum += n * mul
  57. }
  58. return sum
  59. }
  60. // Get the largestDenominator that is a multiple of a basedDenominator
  61. // and fits at least once into a given numerator.
  62. func getLargestDenominator(numerator, multiple, baseDenominator, power int) (int, int) {
  63. if numerator/multiple == 0 {
  64. return 1, power
  65. }
  66. next, nextPower := getLargestDenominator(
  67. numerator, multiple*baseDenominator, baseDenominator, power+1)
  68. if next > multiple {
  69. return next, nextPower
  70. }
  71. return multiple, power
  72. }
  73. // Convers a list of numbers representing a column into a alphabetic
  74. // representation, as used in the spreadsheet.
  75. func formatColumnName(colId []int) string {
  76. lastPart := len(colId) - 1
  77. result := ""
  78. for n, part := range colId {
  79. if n == lastPart {
  80. // The least significant number is in the
  81. // range 0-25, all other numbers are 1-26,
  82. // hence we use a differente offset for the
  83. // last part.
  84. result += string(part + 65)
  85. } else {
  86. // Don't output leading 0s, as there is no
  87. // representation of 0 in this format.
  88. if part > 0 {
  89. result += string(part + 64)
  90. }
  91. }
  92. }
  93. return result
  94. }
  95. func smooshBase26Slice(b26 []int) []int {
  96. // Smoosh values together, eliminating 0s from all but the
  97. // least significant part.
  98. lastButOnePart := len(b26) - 2
  99. for i := lastButOnePart; i > 0; i-- {
  100. part := b26[i]
  101. if part == 0 {
  102. greaterPart := b26[i-1]
  103. if greaterPart > 0 {
  104. b26[i-1] = greaterPart - 1
  105. b26[i] = 26
  106. }
  107. }
  108. }
  109. return b26
  110. }
  111. func intToBase26(x int) (parts []int) {
  112. // Excel column codes are pure evil - in essence they're just
  113. // base26, but they don't represent the number 0.
  114. b26Denominator, _ := getLargestDenominator(x, 1, 26, 0)
  115. // This loop terminates because integer division of 1 / 26
  116. // returns 0.
  117. for d := b26Denominator; d > 0; d = d / 26 {
  118. value := x / d
  119. remainder := x % d
  120. parts = append(parts, value)
  121. x = remainder
  122. }
  123. return parts
  124. }
  125. // numericToLetters is used to convert a zero based, numeric column
  126. // indentifier into a character code.
  127. func numericToLetters(colRef int) string {
  128. parts := intToBase26(colRef)
  129. return formatColumnName(smooshBase26Slice(parts))
  130. }
  131. // letterOnlyMapF is used in conjunction with strings.Map to return
  132. // only the characters A-Z and a-z in a string
  133. func letterOnlyMapF(rune rune) rune {
  134. switch {
  135. case 'A' <= rune && rune <= 'Z':
  136. return rune
  137. case 'a' <= rune && rune <= 'z':
  138. return rune - 32
  139. }
  140. return -1
  141. }
  142. // intOnlyMapF is used in conjunction with strings.Map to return only
  143. // the numeric portions of a string.
  144. func intOnlyMapF(rune rune) rune {
  145. if rune >= 48 && rune < 58 {
  146. return rune
  147. }
  148. return -1
  149. }
  150. // getCoordsFromCellIDString returns the zero based cartesian
  151. // coordinates from a cell name in Excel format, e.g. the cellIDString
  152. // "A1" returns 0, 0 and the "B3" return 1, 2.
  153. func getCoordsFromCellIDString(cellIDString string) (x, y int, error error) {
  154. var letterPart string = strings.Map(letterOnlyMapF, cellIDString)
  155. y, error = strconv.Atoi(strings.Map(intOnlyMapF, cellIDString))
  156. if error != nil {
  157. return x, y, error
  158. }
  159. y -= 1 // Zero based
  160. x = lettersToNumeric(letterPart)
  161. return x, y, error
  162. }
  163. // getCellIDStringFromCoords returns the Excel format cell name that
  164. // represents a pair of zero based cartesian coordinates.
  165. func getCellIDStringFromCoords(x, y int) string {
  166. letterPart := numericToLetters(x)
  167. numericPart := y + 1
  168. return fmt.Sprintf("%s%d", letterPart, numericPart)
  169. }
  170. // getMaxMinFromDimensionRef return the zero based cartesian maximum
  171. // and minimum coordinates from the dimension reference embedded in a
  172. // XLSX worksheet. For example, the dimension reference "A1:B2"
  173. // returns "0,0", "1,1".
  174. func getMaxMinFromDimensionRef(ref string) (minx, miny, maxx, maxy int, err error) {
  175. var parts []string
  176. parts = strings.Split(ref, ":")
  177. minx, miny, err = getCoordsFromCellIDString(parts[0])
  178. if err != nil {
  179. return -1, -1, -1, -1, err
  180. }
  181. if len(parts) == 1 {
  182. maxx, maxy = minx, miny
  183. return
  184. }
  185. maxx, maxy, err = getCoordsFromCellIDString(parts[1])
  186. if err != nil {
  187. return -1, -1, -1, -1, err
  188. }
  189. return
  190. }
  191. // calculateMaxMinFromWorkSheet works out the dimensions of a spreadsheet
  192. // that doesn't have a DimensionRef set. The only case currently
  193. // known where this is true is with XLSX exported from Google Docs.
  194. func calculateMaxMinFromWorksheet(worksheet *xlsxWorksheet) (minx, miny, maxx, maxy int, err error) {
  195. // Note, this method could be very slow for large spreadsheets.
  196. var x, y int
  197. var maxVal int
  198. maxVal = int(^uint(0) >> 1)
  199. minx = maxVal
  200. miny = maxVal
  201. maxy = 0
  202. maxx = 0
  203. for _, row := range worksheet.SheetData.Row {
  204. for _, cell := range row.C {
  205. x, y, err = getCoordsFromCellIDString(cell.R)
  206. if err != nil {
  207. return -1, -1, -1, -1, err
  208. }
  209. if x < minx {
  210. minx = x
  211. }
  212. if x > maxx {
  213. maxx = x
  214. }
  215. if y < miny {
  216. miny = y
  217. }
  218. if y > maxy {
  219. maxy = y
  220. }
  221. }
  222. }
  223. if minx == maxVal {
  224. minx = 0
  225. }
  226. if miny == maxVal {
  227. miny = 0
  228. }
  229. return
  230. }
  231. // makeRowFromSpan will, when given a span expressed as a string,
  232. // return an empty Row large enough to encompass that span and
  233. // populate it with empty cells. All rows start from cell 1 -
  234. // regardless of the lower bound of the span.
  235. func makeRowFromSpan(spans string) *Row {
  236. var error error
  237. var upper int
  238. var row *Row
  239. var cell *Cell
  240. row = new(Row)
  241. _, upper, error = getRangeFromString(spans)
  242. if error != nil {
  243. panic(error)
  244. }
  245. error = nil
  246. row.Cells = make([]*Cell, upper)
  247. for i := 0; i < upper; i++ {
  248. cell = new(Cell)
  249. cell.Value = ""
  250. row.Cells[i] = cell
  251. }
  252. return row
  253. }
  254. // makeRowFromRaw returns the Row representation of the xlsxRow.
  255. func makeRowFromRaw(rawrow xlsxRow) *Row {
  256. var upper int
  257. var row *Row
  258. var cell *Cell
  259. row = new(Row)
  260. upper = -1
  261. for _, rawcell := range rawrow.C {
  262. if rawcell.R != "" {
  263. x, _, error := getCoordsFromCellIDString(rawcell.R)
  264. if error != nil {
  265. panic(fmt.Sprintf("Invalid Cell Coord, %s\n", rawcell.R))
  266. }
  267. if x > upper {
  268. upper = x
  269. }
  270. continue
  271. }
  272. upper++
  273. }
  274. upper++
  275. row.Cells = make([]*Cell, upper)
  276. for i := 0; i < upper; i++ {
  277. cell = new(Cell)
  278. cell.Value = ""
  279. row.Cells[i] = cell
  280. }
  281. return row
  282. }
  283. func makeEmptyRow() *Row {
  284. row := new(Row)
  285. row.Cells = make([]*Cell, 0)
  286. return row
  287. }
  288. type sharedFormula struct {
  289. x, y int
  290. formula string
  291. }
  292. func formulaForCell(rawcell xlsxC, sharedFormulas map[int]sharedFormula) string {
  293. var res string
  294. f := rawcell.F
  295. if f == nil {
  296. return ""
  297. }
  298. if f.T == "shared" {
  299. x, y, err := getCoordsFromCellIDString(rawcell.R)
  300. if err != nil {
  301. res = f.Content
  302. } else {
  303. if f.Ref != "" {
  304. res = f.Content
  305. sharedFormulas[f.Si] = sharedFormula{x, y, res}
  306. } else {
  307. sharedFormula := sharedFormulas[f.Si]
  308. dx := x - sharedFormula.x
  309. dy := y - sharedFormula.y
  310. orig := []byte(sharedFormula.formula)
  311. var start, end int
  312. for end = 0; end < len(orig); end++ {
  313. c := orig[end]
  314. if c >= 'A' && c <= 'Z' {
  315. res += string(orig[start:end])
  316. start = end
  317. end++
  318. foundNum := false
  319. for ; end < len(orig); end++ {
  320. idc := orig[end]
  321. if idc >= '0' && idc <= '9' {
  322. foundNum = true
  323. } else if idc >= 'A' && idc <= 'Z' {
  324. if foundNum {
  325. break
  326. }
  327. } else {
  328. break
  329. }
  330. }
  331. if foundNum {
  332. fx, fy, _ := getCoordsFromCellIDString(string(orig[start:end]))
  333. fx += dx
  334. fy += dy
  335. res += getCellIDStringFromCoords(fx, fy)
  336. start = end
  337. }
  338. }
  339. }
  340. if start < len(orig) {
  341. res += string(orig[start:end])
  342. }
  343. }
  344. }
  345. } else {
  346. res = f.Content
  347. }
  348. return strings.Trim(res, " \t\n\r")
  349. }
  350. // fillCellData attempts to extract a valid value, usable in
  351. // CSV form from the raw cell value. Note - this is not actually
  352. // general enough - we should support retaining tabs and newlines.
  353. func fillCellData(rawcell xlsxC, reftable *RefTable, sharedFormulas map[int]sharedFormula, cell *Cell) {
  354. var data string = rawcell.V
  355. if len(data) > 0 {
  356. vval := strings.Trim(data, " \t\n\r")
  357. switch rawcell.T {
  358. case "s": // Shared String
  359. ref, error := strconv.Atoi(vval)
  360. if error != nil {
  361. panic(error)
  362. }
  363. cell.Value = reftable.ResolveSharedString(ref)
  364. cell.cellType = CellTypeString
  365. case "b": // Boolean
  366. cell.Value = vval
  367. cell.cellType = CellTypeBool
  368. case "e": // Error
  369. cell.Value = vval
  370. cell.formula = formulaForCell(rawcell, sharedFormulas)
  371. cell.cellType = CellTypeError
  372. default:
  373. if rawcell.F == nil {
  374. // Numeric
  375. cell.Value = vval
  376. cell.cellType = CellTypeNumeric
  377. } else {
  378. // Formula
  379. cell.Value = vval
  380. cell.formula = formulaForCell(rawcell, sharedFormulas)
  381. cell.cellType = CellTypeFormula
  382. }
  383. }
  384. }
  385. }
  386. // readRowsFromSheet is an internal helper function that extracts the
  387. // rows from a XSLXWorksheet, populates them with Cells and resolves
  388. // the value references from the reference table and stores them in
  389. // the rows and columns.
  390. func readRowsFromSheet(Worksheet *xlsxWorksheet, file *File) ([]*Row, []*Col, int, int) {
  391. var rows []*Row
  392. var cols []*Col
  393. var row *Row
  394. var minCol, maxCol, minRow, maxRow, colCount, rowCount int
  395. var reftable *RefTable
  396. var err error
  397. var insertRowIndex, insertColIndex int
  398. sharedFormulas := map[int]sharedFormula{}
  399. if len(Worksheet.SheetData.Row) == 0 {
  400. return nil, nil, 0, 0
  401. }
  402. reftable = file.referenceTable
  403. if len(Worksheet.Dimension.Ref) > 0 {
  404. minCol, minRow, maxCol, maxRow, err = getMaxMinFromDimensionRef(Worksheet.Dimension.Ref)
  405. } else {
  406. minCol, minRow, maxCol, maxRow, err = calculateMaxMinFromWorksheet(Worksheet)
  407. }
  408. if err != nil {
  409. panic(err.Error())
  410. }
  411. rowCount = maxRow + 1
  412. colCount = maxCol + 1
  413. rows = make([]*Row, rowCount)
  414. cols = make([]*Col, colCount)
  415. insertRowIndex = minRow
  416. for i := range cols {
  417. cols[i] = &Col{
  418. Hidden: false,
  419. }
  420. }
  421. // Columns can apply to a range, for convenience we expand the
  422. // ranges out into individual column definitions.
  423. for _, rawcol := range Worksheet.Cols.Col {
  424. // Note, below, that sometimes column definitions can
  425. // exist outside the defined dimensions of the
  426. // spreadsheet - we deliberately exclude these
  427. // columns.
  428. for i := rawcol.Min; i <= rawcol.Max && i <= colCount; i++ {
  429. cols[i-1] = &Col{
  430. Min: rawcol.Min,
  431. Max: rawcol.Max,
  432. Hidden: rawcol.Hidden,
  433. Width: rawcol.Width}
  434. }
  435. }
  436. // insert leading empty rows that is in front of minRow
  437. for rowIndex := 0; rowIndex < minRow; rowIndex++ {
  438. rows[rowIndex] = makeEmptyRow()
  439. }
  440. for rowIndex := 0; rowIndex < len(Worksheet.SheetData.Row); rowIndex++ {
  441. rawrow := Worksheet.SheetData.Row[rowIndex]
  442. // Some spreadsheets will omit blank rows from the
  443. // stored data
  444. for rawrow.R > (insertRowIndex + 1) {
  445. // Put an empty Row into the array
  446. rows[insertRowIndex-minRow] = makeEmptyRow()
  447. insertRowIndex++
  448. }
  449. // range is not empty and only one range exist
  450. if len(rawrow.Spans) != 0 && strings.Count(rawrow.Spans, ":") == 1 {
  451. row = makeRowFromSpan(rawrow.Spans)
  452. } else {
  453. row = makeRowFromRaw(rawrow)
  454. }
  455. row.Hidden = rawrow.Hidden
  456. insertColIndex = minCol
  457. for _, rawcell := range rawrow.C {
  458. x, _, _ := getCoordsFromCellIDString(rawcell.R)
  459. // Some spreadsheets will omit blank cells
  460. // from the data.
  461. for x > insertColIndex {
  462. // Put an empty Cell into the array
  463. row.Cells[insertColIndex] = new(Cell)
  464. insertColIndex++
  465. }
  466. cellX := insertColIndex
  467. cell := row.Cells[cellX]
  468. fillCellData(rawcell, reftable, sharedFormulas, cell)
  469. if file.styles != nil {
  470. cell.style = file.styles.getStyle(rawcell.S)
  471. cell.numFmt = file.styles.getNumberFormat(rawcell.S)
  472. }
  473. cell.date1904 = file.Date1904
  474. // Cell is considered hidden if the row or the column of this cell is hidden
  475. cell.Hidden = rawrow.Hidden || (len(cols) > cellX && cols[cellX].Hidden)
  476. insertColIndex++
  477. }
  478. if len(rows) > insertRowIndex {
  479. rows[insertRowIndex] = row
  480. }
  481. insertRowIndex++
  482. }
  483. return rows, cols, colCount, rowCount
  484. }
  485. type indexedSheet struct {
  486. Index int
  487. Sheet *Sheet
  488. Error error
  489. }
  490. func readSheetViews(xSheetViews xlsxSheetViews) []SheetView {
  491. if xSheetViews.SheetView == nil || len(xSheetViews.SheetView) == 0 {
  492. return nil
  493. }
  494. sheetViews := []SheetView{}
  495. for _, xSheetView := range xSheetViews.SheetView {
  496. sheetView := SheetView{}
  497. if xSheetView.Pane != nil {
  498. xlsxPane := xSheetView.Pane
  499. pane := &Pane{}
  500. pane.XSplit = xlsxPane.XSplit
  501. pane.YSplit = xlsxPane.YSplit
  502. pane.TopLeftCell = xlsxPane.TopLeftCell
  503. pane.ActivePane = xlsxPane.ActivePane
  504. pane.State = xlsxPane.State
  505. sheetView.Pane = pane
  506. }
  507. sheetViews = append(sheetViews, sheetView)
  508. }
  509. return sheetViews
  510. }
  511. // readSheetFromFile is the logic of converting a xlsxSheet struct
  512. // into a Sheet struct. This work can be done in parallel and so
  513. // readSheetsFromZipFile will spawn an instance of this function per
  514. // sheet and get the results back on the provided channel.
  515. func readSheetFromFile(sc chan *indexedSheet, index int, rsheet xlsxSheet, fi *File, sheetXMLMap map[string]string) {
  516. result := &indexedSheet{Index: index, Sheet: nil, Error: nil}
  517. worksheet, error := getWorksheetFromSheet(rsheet, fi.worksheets, sheetXMLMap)
  518. if error != nil {
  519. result.Error = error
  520. sc <- result
  521. return
  522. }
  523. sheet := new(Sheet)
  524. sheet.File = fi
  525. sheet.Rows, sheet.Cols, sheet.MaxCol, sheet.MaxRow = readRowsFromSheet(worksheet, fi)
  526. sheet.Hidden = rsheet.State == sheetStateHidden || rsheet.State == sheetStateVeryHidden
  527. sheet.SheetViews = readSheetViews(worksheet.SheetViews)
  528. sheet.SheetFormat.DefaultColWidth = worksheet.SheetFormatPr.DefaultColWidth
  529. sheet.SheetFormat.DefaultRowHeight = worksheet.SheetFormatPr.DefaultRowHeight
  530. result.Sheet = sheet
  531. sc <- result
  532. }
  533. // readSheetsFromZipFile is an internal helper function that loops
  534. // over the Worksheets defined in the XSLXWorkbook and loads them into
  535. // Sheet objects stored in the Sheets slice of a xlsx.File struct.
  536. func readSheetsFromZipFile(f *zip.File, file *File, sheetXMLMap map[string]string) (map[string]*Sheet, []*Sheet, error) {
  537. var workbook *xlsxWorkbook
  538. var err error
  539. var rc io.ReadCloser
  540. var decoder *xml.Decoder
  541. var sheetCount int
  542. workbook = new(xlsxWorkbook)
  543. rc, err = f.Open()
  544. if err != nil {
  545. return nil, nil, err
  546. }
  547. decoder = xml.NewDecoder(rc)
  548. err = decoder.Decode(workbook)
  549. if err != nil {
  550. return nil, nil, err
  551. }
  552. file.Date1904 = workbook.WorkbookPr.Date1904
  553. // Only try and read sheets that have corresponding files.
  554. // Notably this excludes chartsheets don't right now
  555. var workbookSheets []xlsxSheet
  556. for _, sheet := range workbook.Sheets.Sheet {
  557. if f := worksheetFileForSheet(sheet, file.worksheets, sheetXMLMap); f != nil {
  558. workbookSheets = append(workbookSheets, sheet)
  559. }
  560. }
  561. sheetCount = len(workbookSheets)
  562. sheetsByName := make(map[string]*Sheet, sheetCount)
  563. sheets := make([]*Sheet, sheetCount)
  564. sheetChan := make(chan *indexedSheet, sheetCount)
  565. defer close(sheetChan)
  566. go func() {
  567. err = nil
  568. for i, rawsheet := range workbookSheets {
  569. readSheetFromFile(sheetChan, i, rawsheet, file, sheetXMLMap)
  570. }
  571. }()
  572. for j := 0; j < sheetCount; j++ {
  573. sheet := <-sheetChan
  574. if sheet.Error != nil {
  575. return nil, nil, sheet.Error
  576. }
  577. sheetName := workbookSheets[sheet.Index].Name
  578. sheetsByName[sheetName] = sheet.Sheet
  579. sheet.Sheet.Name = sheetName
  580. sheets[sheet.Index] = sheet.Sheet
  581. }
  582. return sheetsByName, sheets, nil
  583. }
  584. // readSharedStringsFromZipFile() is an internal helper function to
  585. // extract a reference table from the sharedStrings.xml file within
  586. // the XLSX zip file.
  587. func readSharedStringsFromZipFile(f *zip.File) (*RefTable, error) {
  588. var sst *xlsxSST
  589. var error error
  590. var rc io.ReadCloser
  591. var decoder *xml.Decoder
  592. var reftable *RefTable
  593. // In a file with no strings it's possible that
  594. // sharedStrings.xml doesn't exist. In this case the value
  595. // passed as f will be nil.
  596. if f == nil {
  597. return nil, nil
  598. }
  599. rc, error = f.Open()
  600. if error != nil {
  601. return nil, error
  602. }
  603. sst = new(xlsxSST)
  604. decoder = xml.NewDecoder(rc)
  605. error = decoder.Decode(sst)
  606. if error != nil {
  607. return nil, error
  608. }
  609. reftable = MakeSharedStringRefTable(sst)
  610. return reftable, nil
  611. }
  612. // readStylesFromZipFile() is an internal helper function to
  613. // extract a style table from the style.xml file within
  614. // the XLSX zip file.
  615. func readStylesFromZipFile(f *zip.File, theme *theme) (*xlsxStyleSheet, error) {
  616. var style *xlsxStyleSheet
  617. var error error
  618. var rc io.ReadCloser
  619. var decoder *xml.Decoder
  620. rc, error = f.Open()
  621. if error != nil {
  622. return nil, error
  623. }
  624. style = newXlsxStyleSheet(theme)
  625. decoder = xml.NewDecoder(rc)
  626. error = decoder.Decode(style)
  627. if error != nil {
  628. return nil, error
  629. }
  630. buildNumFmtRefTable(style)
  631. return style, nil
  632. }
  633. func buildNumFmtRefTable(style *xlsxStyleSheet) {
  634. for _, numFmt := range style.NumFmts.NumFmt {
  635. // We do this for the side effect of populating the NumFmtRefTable.
  636. style.addNumFmt(numFmt)
  637. }
  638. }
  639. func readThemeFromZipFile(f *zip.File) (*theme, error) {
  640. rc, err := f.Open()
  641. if err != nil {
  642. return nil, err
  643. }
  644. var themeXml xlsxTheme
  645. err = xml.NewDecoder(rc).Decode(&themeXml)
  646. if err != nil {
  647. return nil, err
  648. }
  649. return newTheme(themeXml), nil
  650. }
  651. type WorkBookRels map[string]string
  652. func (w *WorkBookRels) MakeXLSXWorkbookRels() xlsxWorkbookRels {
  653. relCount := len(*w)
  654. xWorkbookRels := xlsxWorkbookRels{}
  655. xWorkbookRels.Relationships = make([]xlsxWorkbookRelation, relCount+3)
  656. for k, v := range *w {
  657. index, err := strconv.Atoi(k[3:])
  658. if err != nil {
  659. panic(err.Error())
  660. }
  661. xWorkbookRels.Relationships[index-1] = xlsxWorkbookRelation{
  662. Id: k,
  663. Target: v,
  664. Type: "http://schemas.openxmlformats.org/officeDocument/2006/relationships/worksheet"}
  665. }
  666. relCount++
  667. sheetId := fmt.Sprintf("rId%d", relCount)
  668. xWorkbookRels.Relationships[relCount-1] = xlsxWorkbookRelation{
  669. Id: sheetId,
  670. Target: "sharedStrings.xml",
  671. Type: "http://schemas.openxmlformats.org/officeDocument/2006/relationships/sharedStrings"}
  672. relCount++
  673. sheetId = fmt.Sprintf("rId%d", relCount)
  674. xWorkbookRels.Relationships[relCount-1] = xlsxWorkbookRelation{
  675. Id: sheetId,
  676. Target: "theme/theme1.xml",
  677. Type: "http://schemas.openxmlformats.org/officeDocument/2006/relationships/theme"}
  678. relCount++
  679. sheetId = fmt.Sprintf("rId%d", relCount)
  680. xWorkbookRels.Relationships[relCount-1] = xlsxWorkbookRelation{
  681. Id: sheetId,
  682. Target: "styles.xml",
  683. Type: "http://schemas.openxmlformats.org/officeDocument/2006/relationships/styles"}
  684. return xWorkbookRels
  685. }
  686. // readWorkbookRelationsFromZipFile is an internal helper function to
  687. // extract a map of relationship ID strings to the name of the
  688. // worksheet.xml file they refer to. The resulting map can be used to
  689. // reliably derefence the worksheets in the XLSX file.
  690. func readWorkbookRelationsFromZipFile(workbookRels *zip.File) (WorkBookRels, error) {
  691. var sheetXMLMap WorkBookRels
  692. var wbRelationships *xlsxWorkbookRels
  693. var rc io.ReadCloser
  694. var decoder *xml.Decoder
  695. var err error
  696. rc, err = workbookRels.Open()
  697. if err != nil {
  698. return nil, err
  699. }
  700. decoder = xml.NewDecoder(rc)
  701. wbRelationships = new(xlsxWorkbookRels)
  702. err = decoder.Decode(wbRelationships)
  703. if err != nil {
  704. return nil, err
  705. }
  706. sheetXMLMap = make(WorkBookRels)
  707. for _, rel := range wbRelationships.Relationships {
  708. if strings.HasSuffix(rel.Target, ".xml") && rel.Type == "http://schemas.openxmlformats.org/officeDocument/2006/relationships/worksheet" {
  709. _, filename := path.Split(rel.Target)
  710. sheetXMLMap[rel.Id] = strings.Replace(filename, ".xml", "", 1)
  711. }
  712. }
  713. return sheetXMLMap, nil
  714. }
  715. // ReadZip() takes a pointer to a zip.ReadCloser and returns a
  716. // xlsx.File struct populated with its contents. In most cases
  717. // ReadZip is not used directly, but is called internally by OpenFile.
  718. func ReadZip(f *zip.ReadCloser) (*File, error) {
  719. defer f.Close()
  720. return ReadZipReader(&f.Reader)
  721. }
  722. // ReadZipReader() can be used to read an XLSX in memory without
  723. // touching the filesystem.
  724. func ReadZipReader(r *zip.Reader) (*File, error) {
  725. var err error
  726. var file *File
  727. var reftable *RefTable
  728. var sharedStrings *zip.File
  729. var sheetXMLMap map[string]string
  730. var sheetsByName map[string]*Sheet
  731. var sheets []*Sheet
  732. var style *xlsxStyleSheet
  733. var styles *zip.File
  734. var themeFile *zip.File
  735. var v *zip.File
  736. var workbook *zip.File
  737. var workbookRels *zip.File
  738. var worksheets map[string]*zip.File
  739. file = NewFile()
  740. // file.numFmtRefTable = make(map[int]xlsxNumFmt, 1)
  741. worksheets = make(map[string]*zip.File, len(r.File))
  742. for _, v = range r.File {
  743. switch v.Name {
  744. case "xl/sharedStrings.xml":
  745. sharedStrings = v
  746. case "xl/workbook.xml":
  747. workbook = v
  748. case "xl/_rels/workbook.xml.rels":
  749. workbookRels = v
  750. case "xl/styles.xml":
  751. styles = v
  752. case "xl/theme/theme1.xml":
  753. themeFile = v
  754. default:
  755. if len(v.Name) > 14 {
  756. if v.Name[0:13] == "xl/worksheets" {
  757. worksheets[v.Name[14:len(v.Name)-4]] = v
  758. }
  759. }
  760. }
  761. }
  762. sheetXMLMap, err = readWorkbookRelationsFromZipFile(workbookRels)
  763. if err != nil {
  764. return nil, err
  765. }
  766. file.worksheets = worksheets
  767. reftable, err = readSharedStringsFromZipFile(sharedStrings)
  768. if err != nil {
  769. return nil, err
  770. }
  771. file.referenceTable = reftable
  772. if themeFile != nil {
  773. theme, err := readThemeFromZipFile(themeFile)
  774. if err != nil {
  775. return nil, err
  776. }
  777. file.theme = theme
  778. }
  779. if styles != nil {
  780. style, err = readStylesFromZipFile(styles, file.theme)
  781. if err != nil {
  782. return nil, err
  783. }
  784. file.styles = style
  785. }
  786. sheetsByName, sheets, err = readSheetsFromZipFile(workbook, file, sheetXMLMap)
  787. if err != nil {
  788. return nil, err
  789. }
  790. if sheets == nil {
  791. readerErr := new(XLSXReaderError)
  792. readerErr.Err = "No sheets found in XLSX File"
  793. return nil, readerErr
  794. }
  795. file.Sheet = sheetsByName
  796. file.Sheets = sheets
  797. return file, nil
  798. }