From cb4a3f177d755ad89278ee230fdbdbbadc94485b Mon Sep 17 00:00:00 2001
From: lucasmr <omitted>
Date: Sun, 7 Aug 2022 01:03:29 +0100
Subject: [PATCH] Add max_pages parameter to config file

This makes it easier to track smaller tokens, or reduce the amount
of tokens being tracked.
---
 cointop/cointop.go |  6 +++---
 cointop/config.go  | 12 ++++++++++++
 2 files changed, 15 insertions(+), 3 deletions(-)

diff --git a/cointop/cointop.go b/cointop/cointop.go
index 6af281b..7552c19 100644
--- a/cointop/cointop.go
+++ b/cointop/cointop.go
@@ -67,7 +67,7 @@ type State struct {
 	keepRowFocusOnSort         bool
 	lastSelectedRowIndex       int
 	marketBarHeight            int
-	maxPages                   int
+	maxPages                   uint
 	page                       int
 	perPage                    int
 	portfolio                  *Portfolio
@@ -282,7 +282,7 @@ func NewCointop(config *Config) (*Cointop, error) {
 			hidePortfolioBalances: config.HidePortfolioBalances,
 			keepRowFocusOnSort:    false,
 			marketBarHeight:       1,
-			maxPages:              int(maxPages),
+			maxPages:              maxPages,
 			onlyTable:             config.OnlyTable,
 			onlyChart:             config.OnlyChart,
 			refreshRate:           60 * time.Second,
@@ -423,7 +423,7 @@ func NewCointop(config *Config) (*Cointop, error) {
 	if ct.apiChoice == CoinMarketCap {
 		ct.api = api.NewCMC(ct.apiKeys.cmc)
 	} else if ct.apiChoice == CoinGecko {
-		ct.api = api.NewCG(perPage, maxPages)
+		ct.api = api.NewCG(perPage, ct.State.maxPages)
 	} else {
 		return nil, ErrInvalidAPIChoice
 	}
diff --git a/cointop/config.go b/cointop/config.go
index 10e42ce..99df2ca 100644
--- a/cointop/config.go
+++ b/cointop/config.go
@@ -52,6 +52,7 @@ type ConfigFileConfig struct {
 	CompactNotation   interface{}            `toml:"compact_notation"`
 	EnableMouse       interface{}            `toml:"enable_mouse"`
 	AltCoinLink       interface{}            `toml:"alt_coin_link"` // TODO: should really be in API-specific section
+	MaxPages          interface{}            `toml:"max_pages"`
 	Table             map[string]interface{} `toml:"table"`
 	Chart             map[string]interface{} `toml:"chart"`
 }
@@ -79,6 +80,7 @@ func (ct *Cointop) SetupConfig() error {
 		ct.loadAltCoinLinkFromConfig,
 		ct.loadPriceAlertsFromConfig,
 		ct.loadPortfolioFromConfig,
+		ct.loadMaxPagesFromConfig,
 	}
 
 	for _, f := range loaders {
@@ -302,6 +304,7 @@ func (ct *Cointop) ConfigToToml() ([]byte, error) {
 		CompactNotation:   ct.State.compactNotation,
 		EnableMouse:       ct.State.enableMouse,
 		AltCoinLink:       ct.State.altCoinLink,
+		MaxPages:          ct.State.maxPages,
 	}
 
 	var b bytes.Buffer
@@ -746,6 +749,15 @@ func (ct *Cointop) loadPriceAlertsFromConfig() error {
 	return nil
 }
 
+func (ct *Cointop) loadMaxPagesFromConfig() error {
+	log.Debug("loadMaxPagesFromConfig()")
+	if MaxPages, ok := ct.config.MaxPages.(int64); ok {
+		ct.State.maxPages = uint(MaxPages)
+	}
+
+	return nil
+}
+
 // GetColorschemeColors loads colors from colorscheme file to struct
 func (ct *Cointop) GetColorschemeColors() (map[string]interface{}, error) {
 	log.Debug("GetColorschemeColors()")