Initial untidy structure and tests

lib.go

@@ -0,0 +1,180 @@
+package xlsx
+
+import (
+	"archive/zip"
+	"io"
+	"os"
+	"xml"
+
+)
+
+type XLSXV struct {
+	Data string "chardata"
+}
+
+type XLSXC struct {
+	R string "attr"
+	T string "attr"
+	V XLSXV
+}
+
+type XLSXRow struct {
+	R string "attr"
+	Spans string "attr"
+	C []XLSXC
+}
+
+type XLSXSheetData struct {
+	Row []XLSXRow
+}
+
+type XLSXSheetFormatPr struct {
+	BaseColWidth string "attr"
+	DefaultRowHeight string "attr"
+}
+
+type XLSXSelection struct {
+	ActiveCell string "attr"
+	SQRef string "attr"
+}
+
+type XLSXSheetView struct {
+	TabSelected string "attr"
+	WorkbookViewID string "attr"
+	Selection XLSXSelection
+}
+
+type XLSXSheetViews struct {
+	SheetView []XLSXSheetView
+}
+
+type XLSXDimension struct {
+	Ref string "attr"
+}
+
+type XLSXWorksheet struct {
+	Dimension XLSXDimension
+	SheetViews XLSXSheetViews
+	SheetFormatPr XLSXSheetFormatPr
+	SheetData XLSXSheetData
+}
+
+type XLSXT struct {
+	Data string "chardata"
+}
+
+type XLSXSI struct {
+	T XLSXT
+}
+
+type XLSXSST struct {
+	Count string "attr"
+	UniqueCount string "attr"
+	SI []XLSXSI
+}
+
+type XLSXFileVersion struct {
+	AppName string "attr"
+	LastEdited string "attr"
+	LowestEdited string "attr"
+	RupBuild string "attr"
+}
+
+type XLSXWorkbookPr struct {
+	DefaultThemeVersion string "attr"
+}
+
+type XLSXWorkBookView struct {
+	XWindow string "attr"
+	YWindow string "attr"
+	WindowWidth string "attr"
+	WindowHeight string "attr"
+}
+
+type XLSXSheet struct {
+	Name string "attr"
+	SheetId string "attr"
+	Id string "attr"
+}
+
+type XLSXDefinedName struct {
+	Data string "chardata"
+	Name string "attr"
+	LocalSheetID string "attr"
+}
+
+type XLSXCalcPr struct {
+	CalcId string "attr"
+}
+
+type XLSXBookViews struct {
+	WorkBookView []XLSXWorkBookView
+}
+
+type XLSXSheets struct {
+	Sheet []XLSXSheet
+}
+
+type XLSXDefinedNames struct {
+	DefinedName []XLSXDefinedName
+}
+
+type XLSXWorkbook struct {
+	FileVersion XLSXFileVersion
+	WorkbookPr XLSXWorkbookPr
+	BookViews XLSXBookViews
+	Sheets XLSXSheets
+	DefinedNames XLSXDefinedNames
+	CalcPr XLSXCalcPr
+}
+
+type XLSXSheetStruct struct {
+
+}
+
+type XLSXFile struct {
+	Sheets map [string]*XLSXSheetStruct
+}
+
+type XLSXFileInterface interface  {
+	GetSheet(sheetname string) XLSXSheetStruct
+}
+
+
+func readSheetsFromZipFile(f *zip.File) os.Error {
+	var workbook *XLSXWorkbook
+	var error os.Error
+	var rc io.ReadCloser
+	workbook = new(XLSXWorkbook)
+	rc, error = f.Open()
+	if error != nil {
+		return error
+	}	
+	error = xml.Unmarshal(rc, workbook)
+	if error != nil {
+		return error
+	}
+	return nil
+}
+
+func OpenXLSXFile(filename string) (x *XLSXFile, e os.Error) {
+	var f *zip.ReadCloser
+	var error os.Error
+	var xlsxFile *XLSXFile
+	var v *zip.File
+	f, error = zip.OpenReader(filename)
+	if error != nil {
+		return nil, error
+	}
+	for _, v = range f.File {
+		if v.Name == "xl/workbook.xml" {
+			readSheetsFromZipFile(v)
+		}
+
+	}
+	xlsxFile = new(XLSXFile)
+	f.Close()
+	return xlsxFile, nil
+}
+
+

lib_test.go

@@ -0,0 +1,196 @@
+package xlsx
+
+import (
+	"bytes"
+	// "archive/zip"
+	"fmt"
+	"os"
+	"testing"
+	"xml"
+)
+
+func TestOpenXLSXFile(t *testing.T) {
+	var xlsxFile *XLSXFile
+	var error os.Error
+	xlsxFile, error = OpenXLSXFile("testfile.xlsx")
+	if error != nil {
+		t.Error(error.String())
+		return
+	}
+	if xlsxFile == nil {
+		t.Error("OpenXLSXFile returned nil XLSXFileInterface without generating an os.Error")
+		return
+	}
+
+}
+
+func TestExtractSheets(t *testing.T) {
+	var xlsxFile *XLSXFile
+	var smap map[string]*XLSXSheetStruct
+	xlsxFile, _ = OpenXLSXFile("testfile.xlsx")
+	smap = xlsxFile.Sheets
+	if len(smap) == 0 {
+		t.Error("No sheets read from XLSX file")
+		return
+	}
+	fmt.Printf("%v\n", len(smap))
+}
+
+
+func TestUnmarshallSharedStrings(t *testing.T) {
+	var sharedstringsXML = bytes.NewBufferString(`<?xml version="1.0" encoding="UTF-8" standalone="yes"?>
+<sst xmlns="http://schemas.openxmlformats.org/spreadsheetml/2006/main" count="4" uniqueCount="4"><si><t>Foo</t></si><si><t>Bar</t></si><si><t xml:space="preserve">Baz </t></si><si><t>Quuk</t></si></sst>`)
+	sst := new(XLSXSST)
+	error := xml.Unmarshal(sharedstringsXML, sst)
+	if error != nil {
+		t.Error(error.String())
+		return
+	}
+	if sst.Count != "4" {
+		t.Error(`sst.Count != "4"`)
+	}
+	if sst.UniqueCount != "4" {
+		t.Error(`sst.UniqueCount != 4`)
+	}
+	if len(sst.SI) == 0 {
+		t.Error("Expected 4 sst.SI but found none")
+	}
+	si := sst.SI[0]
+	if si.T.Data != "Foo" {
+		t.Error("Expected s.T.Data == 'Foo'")
+	}
+
+}
+
+
+func TestUnmarshallSheet(t *testing.T) {
+	var sheetxml = bytes.NewBufferString(`<?xml version="1.0" encoding="UTF-8" standalone="yes"?>
+<worksheet xmlns="http://schemas.openxmlformats.org/spreadsheetml/2006/main" xmlns:r="http://schemas.openxmlformats.org/officeDocument/2006/relationships"><dimension ref="A1:B2"/><sheetViews><sheetView tabSelected="1" workbookViewId="0"><selection activeCell="C2" sqref="C2"/></sheetView></sheetViews><sheetFormatPr baseColWidth="10" defaultRowHeight="15"/><sheetData><row r="1" spans="1:2"><c r="A1" t="s"><v>0</v></c><c r="B1" t="s"><v>1</v></c></row><row r="2" spans="1:2"><c r="A2" t="s"><v>2</v></c><c r="B2" t="s"><v>3</v></c></row></sheetData><pageMargins left="0.7" right="0.7" top="0.78740157499999996" bottom="0.78740157499999996" header="0.3" footer="0.3"/></worksheet>`)
+	worksheet := new(XLSXWorksheet)
+	error := xml.Unmarshal(sheetxml, worksheet)
+	if error != nil {
+		t.Error(error.String())
+		return
+	}
+	if worksheet.Dimension.Ref != "A1:B2" {
+		t.Error("Expected worksheet.Dimension.Ref == 'A1:B2'")
+	}
+	if len(worksheet.SheetViews.SheetView) == 0 {
+		t.Error("Expected len(worksheet.SheetViews.SheetView) == 1")
+	}
+	sheetview := worksheet.SheetViews.SheetView[0]
+	if sheetview.TabSelected != "1" {
+		t.Error("Expected sheetview.TabSelected == '1'")
+	}
+	if sheetview.WorkbookViewID != "0" {
+		t.Error("Expected sheetview.WorkbookViewID == '0'")
+	}
+	if sheetview.Selection.ActiveCell != "C2" {
+		t.Error("Expeceted sheetview.Selection.ActiveCell == 'C2'")
+	}
+	if sheetview.Selection.SQRef != "C2" {
+		t.Error("Expected sheetview.Selection.SQRef == 'C2'")
+	}
+	if worksheet.SheetFormatPr.BaseColWidth != "10" {
+		t.Error("Expected worksheet.SheetFormatPr.BaseColWidth == '10'")
+	}
+	if worksheet.SheetFormatPr.DefaultRowHeight != "15" {
+		t.Error("Expected worksheet.SheetFormatPr.DefaultRowHeight == '15'")
+	}
+	if len(worksheet.SheetData.Row) == 0 {
+		t.Error("Expected len(worksheet.SheetData.Row) == '2'")
+	}
+	row := worksheet.SheetData.Row[0]
+	if row.R != "1" {
+		t.Error("Expected row.r == '1'")
+	}
+	if row.Spans != "1:2" {
+		t.Error("Expected row.Spans == '1:2'")
+	}
+	if len(row.C) != 2 {
+		t.Error("Expected len(row.C) == 2")
+	}
+	c := row.C[0]
+	if c.R != "A1" {
+		t.Error("Expected c.R == 'A1'")
+	}
+	if c.T != "s" {
+		t.Error("Expected c.T == 's'")
+	}
+	if c.V.Data != "0" {
+		t.Error("Expected c.V.Data == '0'")
+	}
+
+}
+
+func TestUnmarshallXML(t *testing.T) {
+	var error os.Error
+	var buf = bytes.NewBufferString(`<?xml version="1.0" encoding="UTF-8" standalone="yes"?><workbook xmlns="http://schemas.openxmlformats.org/spreadsheetml/2006/main" xmlns:r="http://schemas.openxmlformats.org/officeDocument/2006/relationships"><fileVersion appName="xl" lastEdited="4" lowestEdited="4" rupBuild="4506"/><workbookPr defaultThemeVersion="124226"/><bookViews><workbookView xWindow="120" yWindow="75" windowWidth="15135" windowHeight="7620"/></bookViews><sheets><sheet name="Sheet1" sheetId="1" r:id="rId1"/><sheet name="Sheet2" sheetId="2" r:id="rId2"/><sheet name="Sheet3" sheetId="3" r:id="rId3"/></sheets><definedNames><definedName name="monitors" localSheetId="0">Sheet1!$A$1533</definedName></definedNames><calcPr calcId="125725"/></workbook>`)
+	var workbook *XLSXWorkbook
+	workbook = new(XLSXWorkbook)
+	error = xml.Unmarshal(buf, workbook)
+	if error != nil {
+		t.Error(error.String())
+		return
+	}
+	if workbook.FileVersion.AppName != "xl" {
+		t.Error("Expected FileVersion.AppName == 'xl')")
+	}
+	if workbook.FileVersion.LastEdited != "4" {
+		t.Error("Expected FileVersion.LastEdited == '4'")
+	}
+	if workbook.FileVersion.LowestEdited != "4" {
+		t.Error("Expected FileVersion.LowestEdited == '4'")
+	}
+	if workbook.FileVersion.RupBuild != "4506" {
+		t.Error("Expected FileVersion.RupBuild == '4506'")
+	}
+	if workbook.WorkbookPr.DefaultThemeVersion != "124226" {
+		t.Error("Expected workbook.WorkbookPr.DefaultThemeVersion == '124226'")
+	}
+	if len(workbook.BookViews.WorkBookView) == 0 {
+		t.Error("Expected len(workbook.BookViews.WorkBookView) == 0")
+	}
+	workBookView := workbook.BookViews.WorkBookView[0]
+	if workBookView.XWindow != "120" {
+		t.Error("Expected workBookView.XWindow == '120'")
+	}
+	if workBookView.YWindow != "75" {
+		t.Error("Expected workBookView.YWindow == '75'")
+	}
+	if workBookView.WindowWidth != "15135" {
+		t.Error("Expected workBookView.WindowWidth == '15135'")
+	}
+	if workBookView.WindowHeight != "7620" {
+		t.Error("Expected workBookView.WindowHeight == '7620'")
+	}
+	if len(workbook.Sheets.Sheet) == 0 {
+		t.Error("Expected len(workbook.Sheets.Sheet) == 0")
+	}
+	sheet := workbook.Sheets.Sheet[0]
+	if sheet.Id != "rId1" {
+		t.Error("Expected sheet.Id == 'rID1'")
+	}
+	if sheet.Name != "Sheet1" {
+		t.Error("Expected sheet.Name == 'Sheet1'")
+	}
+	if sheet.SheetId != "1" {
+		t.Error("Expected sheet.SheetId == '1'")
+	}
+	if len(workbook.DefinedNames.DefinedName) == 0 {
+		t.Error("Expected len(workbook.DefinedNames.DefinedName) == 0")
+	}
+	dname := workbook.DefinedNames.DefinedName[0]
+	if dname.Data != "Sheet1!$A$1533" {
+		t.Error("dname.Data == 'Sheet1!$A$1533'")
+	}
+	if dname.LocalSheetID != "0" {
+		t.Error("dname.LocalSheetID == '0'")
+	}
+	if dname.Name != "monitors" {
+		t.Error("Expected dname.Name == 'monitors'")
+	}
+	if workbook.CalcPr.CalcId != "125725" {
+		t.Error("workbook.CalcPr.CalcId != '125725'")
+	}
+}