mirror of https://github.com/ghostfolio/ghostfolio
Thomas
4 years ago
6 changed files with 257 additions and 34 deletions
@ -0,0 +1,193 @@ |
|||||
|
import { HttpClient } from '@angular/common/http'; |
||||
|
import { Injectable } from '@angular/core'; |
||||
|
import { CreateOrderDto } from '@ghostfolio/api/app/order/create-order.dto'; |
||||
|
import { Type } from '@prisma/client'; |
||||
|
import { parse } from 'date-fns'; |
||||
|
import { isNumber } from 'lodash'; |
||||
|
import { parse as csvToJson } from 'papaparse'; |
||||
|
import { EMPTY } from 'rxjs'; |
||||
|
import { catchError } from 'rxjs/operators'; |
||||
|
|
||||
|
@Injectable({ |
||||
|
providedIn: 'root' |
||||
|
}) |
||||
|
export class ImportTransactionsService { |
||||
|
private static CURRENCY_KEYS = ['ccy', 'currency']; |
||||
|
private static DATE_KEYS = ['date']; |
||||
|
private static FEE_KEYS = ['fee']; |
||||
|
private static QUANTITY_KEYS = ['qty', 'quantity']; |
||||
|
private static SYMBOL_KEYS = ['code', 'symbol']; |
||||
|
private static TYPE_KEYS = ['action', 'type']; |
||||
|
private static UNIT_PRICE_KEYS = ['price', 'unitprice']; |
||||
|
|
||||
|
public constructor(private http: HttpClient) {} |
||||
|
|
||||
|
public async importCsv({ |
||||
|
defaultAccountId, |
||||
|
fileContent |
||||
|
}: { |
||||
|
defaultAccountId: string; |
||||
|
fileContent: string; |
||||
|
}) { |
||||
|
const content = csvToJson(fileContent, { |
||||
|
dynamicTyping: true, |
||||
|
header: true, |
||||
|
skipEmptyLines: true |
||||
|
}).data; |
||||
|
|
||||
|
const orders: CreateOrderDto[] = []; |
||||
|
|
||||
|
for (const item of content) { |
||||
|
orders.push({ |
||||
|
accountId: defaultAccountId, |
||||
|
currency: this.parseCurrency(item), |
||||
|
dataSource: 'YAHOO', |
||||
|
date: this.parseDate(item), |
||||
|
fee: this.parseFee(item), |
||||
|
quantity: this.parseQuantity(item), |
||||
|
symbol: this.parseSymbol(item), |
||||
|
type: this.parseType(item), |
||||
|
unitPrice: this.parseUnitPrice(item) |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
await this.importJson({ defaultAccountId, content: orders }); |
||||
|
} |
||||
|
|
||||
|
public importJson({ |
||||
|
content, |
||||
|
defaultAccountId |
||||
|
}: { |
||||
|
content: CreateOrderDto[]; |
||||
|
defaultAccountId: string; |
||||
|
}): Promise<void> { |
||||
|
return new Promise((resolve, reject) => { |
||||
|
this.postImport({ |
||||
|
orders: content.map((order) => { |
||||
|
return { ...order, accountId: defaultAccountId }; |
||||
|
}) |
||||
|
}) |
||||
|
.pipe( |
||||
|
catchError((error) => { |
||||
|
reject(error); |
||||
|
return EMPTY; |
||||
|
}) |
||||
|
) |
||||
|
.subscribe({ |
||||
|
next: () => { |
||||
|
resolve(); |
||||
|
} |
||||
|
}); |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
private lowercaseKeys(aObject: any) { |
||||
|
return Object.keys(aObject).reduce((acc, key) => { |
||||
|
acc[key.toLowerCase()] = aObject[key]; |
||||
|
return acc; |
||||
|
}, {}); |
||||
|
} |
||||
|
|
||||
|
private parseCurrency(aItem: any) { |
||||
|
const item = this.lowercaseKeys(aItem); |
||||
|
|
||||
|
for (const key of ImportTransactionsService.CURRENCY_KEYS) { |
||||
|
if (item[key]) { |
||||
|
return item[key]; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
throw new Error('Could not parse currency'); |
||||
|
} |
||||
|
|
||||
|
private parseDate(aItem: any) { |
||||
|
const item = this.lowercaseKeys(aItem); |
||||
|
let date: string; |
||||
|
|
||||
|
for (const key of ImportTransactionsService.DATE_KEYS) { |
||||
|
if (item[key]) { |
||||
|
try { |
||||
|
date = parse(item[key], 'dd-MM-yyyy', new Date()).toISOString(); |
||||
|
} catch {} |
||||
|
|
||||
|
try { |
||||
|
date = parse(item[key], 'dd/MM/yyyy', new Date()).toISOString(); |
||||
|
} catch {} |
||||
|
|
||||
|
if (date) { |
||||
|
return date; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
throw new Error('Could not parse date'); |
||||
|
} |
||||
|
|
||||
|
private parseFee(aItem: any) { |
||||
|
const item = this.lowercaseKeys(aItem); |
||||
|
|
||||
|
for (const key of ImportTransactionsService.FEE_KEYS) { |
||||
|
if ((item[key] || item[key] === 0) && isNumber(item[key])) { |
||||
|
return item[key]; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
throw new Error('Could not parse fee'); |
||||
|
} |
||||
|
|
||||
|
private parseQuantity(aItem: any) { |
||||
|
const item = this.lowercaseKeys(aItem); |
||||
|
|
||||
|
for (const key of ImportTransactionsService.QUANTITY_KEYS) { |
||||
|
if (item[key] && isNumber(item[key])) { |
||||
|
return item[key]; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
throw new Error('Could not parse quantity'); |
||||
|
} |
||||
|
|
||||
|
private parseSymbol(aItem: any) { |
||||
|
const item = this.lowercaseKeys(aItem); |
||||
|
|
||||
|
for (const key of ImportTransactionsService.SYMBOL_KEYS) { |
||||
|
if (item[key]) { |
||||
|
return item[key]; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
throw new Error('Could not parse symbol'); |
||||
|
} |
||||
|
|
||||
|
private parseType(aItem: any) { |
||||
|
const item = this.lowercaseKeys(aItem); |
||||
|
|
||||
|
for (const key of ImportTransactionsService.TYPE_KEYS) { |
||||
|
if (item[key]) { |
||||
|
if (item[key].toLowerCase() === 'buy') { |
||||
|
return Type.BUY; |
||||
|
} else if (item[key].toLowerCase() === 'sell') { |
||||
|
return Type.SELL; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
throw new Error('Could not parse type'); |
||||
|
} |
||||
|
|
||||
|
private parseUnitPrice(aItem: any) { |
||||
|
const item = this.lowercaseKeys(aItem); |
||||
|
|
||||
|
for (const key of ImportTransactionsService.UNIT_PRICE_KEYS) { |
||||
|
if (item[key] && isNumber(item[key])) { |
||||
|
return item[key]; |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
throw new Error('Could not parse unit price (unitPrice)'); |
||||
|
} |
||||
|
|
||||
|
private postImport(aImportData: { orders: CreateOrderDto[] }) { |
||||
|
return this.http.post<void>('/api/import', aImportData); |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue