MaysWind
85 lines
3.8 KiB
Go
85 lines
3.8 KiB
Go
package converters
|
|
|
|
import (
|
|
"github.com/mayswind/ezbookkeeping/pkg/core"
|
|
"github.com/mayswind/ezbookkeeping/pkg/models"
|
|
)
|
|
|
|
// ezBookKeepingTransactionDataPlainTextConverter defines the structure of plain file converter for transaction data
|
|
type ezBookKeepingTransactionDataPlainTextConverter struct {
|
|
DataTableTransactionDataConverter
|
|
columns []DataTableColumn
|
|
}
|
|
|
|
var ezbookkeepingDataColumnNameMapping = map[DataTableColumn]string{
|
|
DATA_TABLE_TRANSACTION_TIME: "Time",
|
|
DATA_TABLE_TRANSACTION_TIMEZONE: "Timezone",
|
|
DATA_TABLE_TRANSACTION_TYPE: "Type",
|
|
DATA_TABLE_CATEGORY: "Category",
|
|
DATA_TABLE_SUB_CATEGORY: "Sub Category",
|
|
DATA_TABLE_ACCOUNT_NAME: "Account",
|
|
DATA_TABLE_ACCOUNT_CURRENCY: "Account Currency",
|
|
DATA_TABLE_AMOUNT: "Amount",
|
|
DATA_TABLE_RELATED_ACCOUNT_NAME: "Account2",
|
|
DATA_TABLE_RELATED_ACCOUNT_CURRENCY: "Account2 Currency",
|
|
DATA_TABLE_RELATED_AMOUNT: "Account2 Amount",
|
|
DATA_TABLE_GEOGRAPHIC_LOCATION: "Geographic Location",
|
|
DATA_TABLE_TAGS: "Tags",
|
|
DATA_TABLE_DESCRIPTION: "Description",
|
|
}
|
|
|
|
var ezbookkeepingTransactionTypeNameMapping = map[models.TransactionDbType]string{
|
|
models.TRANSACTION_DB_TYPE_MODIFY_BALANCE: "Balance Modification",
|
|
models.TRANSACTION_DB_TYPE_INCOME: "Income",
|
|
models.TRANSACTION_DB_TYPE_EXPENSE: "Expense",
|
|
models.TRANSACTION_DB_TYPE_TRANSFER_OUT: "Transfer",
|
|
models.TRANSACTION_DB_TYPE_TRANSFER_IN: "Transfer",
|
|
}
|
|
|
|
var ezbookkeepingNameTransactionTypeMapping = map[string]models.TransactionDbType{
|
|
"Balance Modification": models.TRANSACTION_DB_TYPE_MODIFY_BALANCE,
|
|
"Income": models.TRANSACTION_DB_TYPE_INCOME,
|
|
"Expense": models.TRANSACTION_DB_TYPE_EXPENSE,
|
|
"Transfer": models.TRANSACTION_DB_TYPE_TRANSFER_OUT,
|
|
}
|
|
|
|
var ezbookkeepingDataColumns = []DataTableColumn{
|
|
DATA_TABLE_TRANSACTION_TIME,
|
|
DATA_TABLE_TRANSACTION_TIMEZONE,
|
|
DATA_TABLE_TRANSACTION_TYPE,
|
|
DATA_TABLE_CATEGORY,
|
|
DATA_TABLE_SUB_CATEGORY,
|
|
DATA_TABLE_ACCOUNT_NAME,
|
|
DATA_TABLE_ACCOUNT_CURRENCY,
|
|
DATA_TABLE_AMOUNT,
|
|
DATA_TABLE_RELATED_ACCOUNT_NAME,
|
|
DATA_TABLE_RELATED_ACCOUNT_CURRENCY,
|
|
DATA_TABLE_RELATED_AMOUNT,
|
|
DATA_TABLE_GEOGRAPHIC_LOCATION,
|
|
DATA_TABLE_TAGS,
|
|
DATA_TABLE_DESCRIPTION,
|
|
}
|
|
|
|
// ToExportedContent returns the exported transaction plain text data
|
|
func (c *ezBookKeepingTransactionDataPlainTextConverter) ToExportedContent(ctx core.Context, uid int64, transactions []*models.Transaction, accountMap map[int64]*models.Account, categoryMap map[int64]*models.TransactionCategory, tagMap map[int64]*models.TransactionTag, allTagIndexes map[int64][]int64) ([]byte, error) {
|
|
dataTableBuilder := createNewezbookkeepingTransactionPlainTextDataTableBuilder(len(transactions), c.columns, c.dataColumnMapping, c.columnSeparator, c.lineSeparator)
|
|
err := c.buildExportedContent(ctx, dataTableBuilder, uid, transactions, accountMap, categoryMap, tagMap, allTagIndexes)
|
|
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return []byte(dataTableBuilder.String()), nil
|
|
}
|
|
|
|
// ParseImportedData returns the imported data by parsing the transaction plain text data
|
|
func (c *ezBookKeepingTransactionDataPlainTextConverter) ParseImportedData(ctx core.Context, user *models.User, data []byte, defaultTimezoneOffset int16, accountMap map[string]*models.Account, categoryMap map[string]*models.TransactionCategory, tagMap map[string]*models.TransactionTag) (models.ImportedTransactionSlice, []*models.Account, []*models.TransactionCategory, []*models.TransactionTag, error) {
|
|
dataTable, err := createNewezbookkeepingTransactionPlainTextDataTable(string(data), c.columnSeparator, c.lineSeparator)
|
|
|
|
if err != nil {
|
|
return nil, nil, nil, nil, err
|
|
}
|
|
|
|
return c.parseImportedData(ctx, user, dataTable, defaultTimezoneOffset, accountMap, categoryMap, tagMap)
|
|
}
|