mirror of https://github.com/ghostfolio/ghostfolio
Browse Source
* Update changelog Co-Authored-By: Valentin Zickner <valentin@coderworks.de> Co-authored-by: Thomas Kaul <4159106+dtslvr@users.noreply.github.com>pull/413/head
Valentin Zickner
3 years ago
committed by
GitHub
GitHub
15 changed files with 208 additions and 99 deletions
@ -0,0 +1,73 @@ |
|||
import { DataEnhancerInterface } from '@ghostfolio/api/services/data-provider/interfaces/data-enhancer.interface'; |
|||
import { IDataProviderResponse } from '@ghostfolio/api/services/interfaces/interfaces'; |
|||
import bent from 'bent'; |
|||
|
|||
const getJSON = bent('json'); |
|||
|
|||
export class TrackinsightDataEnhancerService implements DataEnhancerInterface { |
|||
private static baseUrl = 'https://data.trackinsight.com/holdings'; |
|||
private static countries = require('countries-list/dist/countries.json'); |
|||
private static sectorsMapping = { |
|||
'Consumer Discretionary': 'Consumer Cyclical', |
|||
'Consumer Defensive': 'Consumer Staples', |
|||
'Health Care': 'Healthcare', |
|||
'Information Technology': 'Technology' |
|||
}; |
|||
|
|||
public async enhance({ |
|||
response, |
|||
symbol |
|||
}: { |
|||
response: IDataProviderResponse; |
|||
symbol: string; |
|||
}): Promise<IDataProviderResponse> { |
|||
if ( |
|||
!(response.assetClass === 'EQUITY' && response.assetSubClass === 'ETF') |
|||
) { |
|||
return response; |
|||
} |
|||
|
|||
const holdings = await getJSON( |
|||
`${TrackinsightDataEnhancerService.baseUrl}/${symbol}.json` |
|||
).catch(() => { |
|||
return getJSON( |
|||
`${TrackinsightDataEnhancerService.baseUrl}/${ |
|||
symbol.split('.')[0] |
|||
}.json` |
|||
); |
|||
}); |
|||
|
|||
if (!response.countries || response.countries.length === 0) { |
|||
response.countries = []; |
|||
for (const [name, value] of Object.entries<any>(holdings.countries)) { |
|||
let countryCode: string; |
|||
|
|||
for (const [key, country] of Object.entries<any>( |
|||
TrackinsightDataEnhancerService.countries |
|||
)) { |
|||
if (country.name === name) { |
|||
countryCode = key; |
|||
break; |
|||
} |
|||
} |
|||
|
|||
response.countries.push({ |
|||
code: countryCode, |
|||
weight: value.weight |
|||
}); |
|||
} |
|||
} |
|||
|
|||
if (!response.sectors || response.sectors.length === 0) { |
|||
response.sectors = []; |
|||
for (const [name, value] of Object.entries<any>(holdings.sectors)) { |
|||
response.sectors.push({ |
|||
name: TrackinsightDataEnhancerService.sectorsMapping[name] ?? name, |
|||
weight: value.weight |
|||
}); |
|||
} |
|||
} |
|||
|
|||
return Promise.resolve(response); |
|||
} |
|||
} |
|||
@ -0,0 +1,11 @@ |
|||
import { IDataProviderResponse } from '@ghostfolio/api/services/interfaces/interfaces'; |
|||
|
|||
export interface DataEnhancerInterface { |
|||
enhance({ |
|||
response, |
|||
symbol |
|||
}: { |
|||
response: IDataProviderResponse; |
|||
symbol: string; |
|||
}): Promise<IDataProviderResponse>; |
|||
} |
|||
Loading…
Reference in new issue