Files
ezbookkeeping/pkg/converters/ezbookkeeping_transaction_data_plain_text_converter.go
T
2024-09-07 03:01:57 +08:00

83 lines
3.6 KiB
Go

package converters
import (
"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 plain text transaction data
func (c *ezBookKeepingTransactionDataPlainTextConverter) ToExportedContent(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(dataTableBuilder, uid, transactions, accountMap, categoryMap, tagMap, allTagIndexes)
if err != nil {
return nil, err
}
return []byte(dataTableBuilder.String()), nil
}
func (c *ezBookKeepingTransactionDataPlainTextConverter) ParseImportedData(user *models.User, data []byte, defaultTimezoneOffset int16, accountMap map[string]*models.Account, categoryMap map[string]*models.TransactionCategory, tagMap map[string]*models.TransactionTag) ([]*models.Transaction, []*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(user, dataTable, defaultTimezoneOffset, accountMap, categoryMap, tagMap)
}