package gexcel

import (
	"bytes"
	"encoding/csv"
	"fmt"
	"math"
	"strconv"
)

type csvColumn []string

func (c csvColumn) getRaw(i int, h func(string) string) string {
	if len(c) > i && i >= 0 {
		if h != nil {
			return h(c[i])
		}
		return c[i]
	}
	return ""
}

func (c csvColumn) GetString(i int) string {
	return c.getRaw(i, nil)
}

func (c csvColumn) GetDate(i int) string {
	return FormatDate(c.getRaw(i, Trim))
}

func (c csvColumn) GetInt(i int) int {
	return int(c.GetInt64(i))
}

func (c csvColumn) GetInt64(i int) int64 {
	v := NumberRemoveComma(c.getRaw(i, Trim))
	n, _ := strconv.ParseInt(v, 10, 64)
	return n
}

func (c csvColumn) GetFloat64(i int, nan bool) float64 {
	v := NumberRemoveComma(c.getRaw(i, Trim))
	n, err := strconv.ParseFloat(v, 64)
	if err != nil && nan {
		return math.NaN()
	}
	return n
}

type CsvFileParse struct {
	records [][]string
}

func (x *CsvFileParse) Type() string {
	return CSV
}

func (x *CsvFileParse) Init(buf *bytes.Buffer) *ParseError {
	reader := csv.NewReader(bytes.NewReader(buf.Bytes()))
	reader.FieldsPerRecord = -1
	reader.LazyQuotes = true // 允许懒引号（忘记遇到哪个问题才加的这行）

	var err error
	if x.records, err = reader.ReadAll(); err != nil { // 返回csv中的所有内容
		return &ParseError{Err: fmt.Errorf("文件数据读取失败（File, %s）", err)}
	}
	return nil
}

func (x *CsvFileParse) Len() int {
	return len(x.records)
}

func (x *CsvFileParse) Parse(h FileHandle, ignore bool) *ParseError {
	ret := &ParseError{Sheets: []*ParseSheetError{&ParseSheetError{}}}
	if ch, err := h.Each("", 0); err != nil {
		if err == SheetIgnore {
			return nil
		}
		ret.Sheets[0].Err = err
		return ret
	} else if rowFirst, rowLen := ch.FirstRow(), len(x.records); rowLen > rowFirst {
		xv := csvColumn(x.records[rowFirst])
		if err = ch.Init(rowLen-rowFirst, &xv); err != nil {
			ret.Sheets[0].Err = err
			return ret
		}
		ret.Sheets[0].Rows = make([]*ParseRowError, 0, 8)

		for j := rowFirst; j < rowLen; j++ {
			xv = x.records[j]
			if err = ch.Each(j, &xv); err != nil {
				ret.Sheets[0].Rows = append(ret.Sheets[0].Rows, &ParseRowError{Row: j + 1, Err: err})
				if !ignore {
					break
				}
			}
		}
		if len(ret.Sheets[0].Rows) > 0 {
			return ret
		}
	}
	return nil
}