Instadapp/trustwallet-assets
1
0
Fork 0
mirror of https://github.com/Instadapp/trustwallet-assets.git synced 2024-07-29 22:37:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Refactor code (#17700)

Browse Source
This commit is contained in:
daniel 2022-01-28 13:00:13 +03:00 committed by GitHub
parent 32d64839da
commit 242c1a4392
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
internal/manager/commands.go
View File

@ -73,20 +73,22 @@ func CreateAssetInfoJSONTemplate(token string) error {
return nil return nil
} }
func AddTokenToTokenListJSON(chain coin.Coin, assetId, tokenID string, tokenListType path.TokenListType) error { func AddTokenToTokenListJSON(chain coin.Coin, assetID, tokenID string, tokenListType path.TokenListType) error {
setup() setup()
// check for duplicates. // Check for duplicates.
tokenListTypes := []path.TokenListType{path.TokenlistDefault, path.TokenlistExtended} tokenListTypes := []path.TokenListType{path.TokenlistDefault, path.TokenlistExtended}
for _, t := range tokenListTypes { for _, t := range tokenListTypes {
tokenListPath := path.GetTokenListPath(chain.Handle, t) tokenListPath := path.GetTokenListPath(chain.Handle, t)
var list tokenlist.Model var list tokenlist.Model
err := libFile.ReadJSONFile(tokenListPath, &list) err := libFile.ReadJSONFile(tokenListPath, &list)
if err != nil { if err != nil {
return fmt.Errorf("failed to read data from %s: %w", tokenListPath, err) return fmt.Errorf("failed to read data from %s: %w", tokenListPath, err)
} }
for _, item := range list.Tokens { for _, item := range list.Tokens {
if item.Asset == assetId { if item.Asset == assetID {
return fmt.Errorf("duplicate asset, already exist in %s", tokenListPath) return fmt.Errorf("duplicate asset, already exist in %s", tokenListPath)
} }
} }
@ -94,6 +96,7 @@ func AddTokenToTokenListJSON(chain coin.Coin, assetId, tokenID string, tokenList
var list tokenlist.Model var list tokenlist.Model
tokenListPath := path.GetTokenListPath(chain.Handle, tokenListType) tokenListPath := path.GetTokenListPath(chain.Handle, tokenListType)
err := libFile.ReadJSONFile(tokenListPath, &list) err := libFile.ReadJSONFile(tokenListPath, &list)
if err != nil { if err != nil {
return fmt.Errorf("failed to read data from %s: %w", tokenListPath, err) return fmt.Errorf("failed to read data from %s: %w", tokenListPath, err)
@ -105,7 +108,7 @@ func AddTokenToTokenListJSON(chain coin.Coin, assetId, tokenID string, tokenList
} }
newToken := tokenlist.Token{ newToken := tokenlist.Token{
Asset: assetId, Asset: assetID,
Type: types.TokenType(*assetInfo.Type), Type: types.TokenType(*assetInfo.Type),
Address: *assetInfo.ID, Address: *assetInfo.ID,
Name: *assetInfo.Name, Name: *assetInfo.Name,
@ -126,8 +129,8 @@ func AddTokenToTokenListJSON(chain coin.Coin, assetId, tokenID string, tokenList
func getAssetInfo(chain coin.Coin, tokenID string) (*info.AssetModel, error) { func getAssetInfo(chain coin.Coin, tokenID string) (*info.AssetModel, error) {
path := path.GetAssetInfoPath(chain.Handle, tokenID) path := path.GetAssetInfoPath(chain.Handle, tokenID)
var assetModel info.AssetModel var assetModel info.AssetModel
err := libFile.ReadJSONFile(path, &assetModel) err := libFile.ReadJSONFile(path, &assetModel)
if err != nil { if err != nil {
return nil, fmt.Errorf("failed to read data from info.json: %w", err) return nil, fmt.Errorf("failed to read data from info.json: %w", err)