mirror of https://github.com/miguelmota/cointop
Add command to print holdings
Miguel Mota
4 years ago
parent
9d15e268c2
commit
97facbe48d
@ -0,0 +1,29 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"github.com/miguelmota/cointop/cointop"
|
||||
"github.com/miguelmota/cointop/cointop/common/filecache"
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
// CleanCmd ...
|
||||
func CleanCmd() *cobra.Command {
|
||||
cacheDir := filecache.DefaultCacheDir
|
||||
|
||||
cleanCmd := &cobra.Command{
|
||||
Use: "clean",
|
||||
Short: "Clear the cache",
|
||||
Long: `The clean command clears the cache`,
|
||||
RunE: func(cmd *cobra.Command, args []string) error {
|
||||
// NOTE: if clean command, clean but don't run cointop
|
||||
return cointop.Clean(&cointop.CleanConfig{
|
||||
Log: true,
|
||||
CacheDir: cacheDir,
|
||||
})
|
||||
},
|
||||
}
|
||||
|
||||
cleanCmd.Flags().StringVarP(&cacheDir, "cache-dir", "", cacheDir, "Cache directory")
|
||||
|
||||
return cleanCmd
|
||||
}
|
||||
@ -0,0 +1,42 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/miguelmota/cointop/cointop"
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
// HoldingsCmd ...
|
||||
func HoldingsCmd() *cobra.Command {
|
||||
var total, noCache bool
|
||||
var config string
|
||||
|
||||
holdingsCmd := &cobra.Command{
|
||||
Use: "holdings",
|
||||
Short: "Displays current holdings",
|
||||
Long: `The holdings command shows your current holdings`,
|
||||
RunE: func(cmd *cobra.Command, args []string) error {
|
||||
ct, err := cointop.NewCointop(&cointop.Config{
|
||||
ConfigFilepath: config,
|
||||
APIChoice: cointop.CoinGecko,
|
||||
CacheDir: cointop.DefaultCacheDir,
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if total {
|
||||
return ct.PrintTotalHoldings()
|
||||
}
|
||||
|
||||
return ct.PrintHoldingsTable()
|
||||
},
|
||||
}
|
||||
|
||||
holdingsCmd.Flags().BoolVarP(&total, "total", "t", false, "Show total only")
|
||||
holdingsCmd.Flags().BoolVarP(&noCache, "no-cache", "", false, "No cache")
|
||||
holdingsCmd.Flags().StringVarP(&config, "config", "c", "", fmt.Sprintf("Config filepath. (default %s)", cointop.DefaultConfigFilepath))
|
||||
|
||||
return holdingsCmd
|
||||
}
|
||||
@ -0,0 +1,30 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"github.com/miguelmota/cointop/cointop"
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
// PriceCmd ...
|
||||
func PriceCmd() *cobra.Command {
|
||||
var apiChoice, coin, currency string
|
||||
|
||||
priceCmd := &cobra.Command{
|
||||
Use: "price",
|
||||
Short: "Displays the current price of a coin",
|
||||
Long: `The price command display the current price of a coin`,
|
||||
RunE: func(cmd *cobra.Command, args []string) error {
|
||||
return cointop.PrintPrice(&cointop.PriceConfig{
|
||||
Coin: coin,
|
||||
Currency: currency,
|
||||
APIChoice: apiChoice,
|
||||
})
|
||||
},
|
||||
}
|
||||
|
||||
priceCmd.Flags().StringVarP(&coin, "coin", "c", "bitcoin", "Full name of the coin (default \"bitcoin\")")
|
||||
priceCmd.Flags().StringVarP(¤cy, "currency", "f", "USD", "The currency to convert to (default \"USD\")")
|
||||
priceCmd.Flags().StringVarP(&apiChoice, "api", "a", cointop.CoinGecko, "API choice. Available choices are \"coinmarketcap\" and \"coingecko\"")
|
||||
|
||||
return priceCmd
|
||||
}
|
||||
@ -0,0 +1,29 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"github.com/miguelmota/cointop/cointop"
|
||||
"github.com/miguelmota/cointop/cointop/common/filecache"
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
// ResetCmd ...
|
||||
func ResetCmd() *cobra.Command {
|
||||
cacheDir := filecache.DefaultCacheDir
|
||||
|
||||
resetCmd := &cobra.Command{
|
||||
Use: "reset",
|
||||
Short: "Resets the config and clear the cache",
|
||||
Long: `The reset command resets the config and clears the cache`,
|
||||
RunE: func(cmd *cobra.Command, args []string) error {
|
||||
// NOTE: if reset command, reset but don't run cointop
|
||||
return cointop.Reset(&cointop.ResetConfig{
|
||||
Log: true,
|
||||
CacheDir: cacheDir,
|
||||
})
|
||||
},
|
||||
}
|
||||
|
||||
resetCmd.Flags().StringVarP(&cacheDir, "cache-dir", "", cacheDir, "Cache directory")
|
||||
|
||||
return resetCmd
|
||||
}
|
||||
@ -0,0 +1,108 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/miguelmota/cointop/cointop"
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
// RootCmd ...
|
||||
func RootCmd() *cobra.Command {
|
||||
var version, test, clean, reset, hideMarketbar, hideChart, hideStatusbar, onlyTable, silent, noCache bool
|
||||
var refreshRate uint
|
||||
var config, cmcAPIKey, apiChoice, colorscheme string
|
||||
perPage := cointop.DefaultPerPage
|
||||
cacheDir := cointop.DefaultCacheDir
|
||||
|
||||
rootCmd := &cobra.Command{
|
||||
Use: "cointop",
|
||||
Short: "Cointop is an interactive terminal based app for tracking cryptocurrencies",
|
||||
Long: `
|
||||
_ _
|
||||
___ ___ (_)_ __ | |_ ___ _ __
|
||||
/ __/ _ \| | '_ \| __/ _ \| '_ \
|
||||
| (_| (_) | | | | | || (_) | |_) |
|
||||
\___\___/|_|_| |_|\__\___/| .__/
|
||||
|_|
|
||||
|
||||
Cointop is a fast and lightweight interactive terminal based UI application for tracking and monitoring cryptocurrency coin stats in real-time.
|
||||
|
||||
See git.io/cointop for more info.`,
|
||||
RunE: func(cmd *cobra.Command, args []string) error {
|
||||
if version {
|
||||
cointop.PrintVersion()
|
||||
return nil
|
||||
}
|
||||
|
||||
if test {
|
||||
// TODO: deprecate test flag, only have test command
|
||||
doTest()
|
||||
return nil
|
||||
}
|
||||
|
||||
// NOTE: if reset flag enabled, reset and run cointop
|
||||
if reset {
|
||||
if err := cointop.Reset(&cointop.ResetConfig{
|
||||
Log: !silent,
|
||||
}); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
// NOTE: if clean flag enabled, clean and run cointop
|
||||
if clean {
|
||||
if err := cointop.Clean(&cointop.CleanConfig{
|
||||
Log: !silent,
|
||||
}); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
var refreshRateP *uint
|
||||
if cmd.Flags().Changed("refresh-rate") {
|
||||
refreshRateP = &refreshRate
|
||||
}
|
||||
|
||||
ct, err := cointop.NewCointop(&cointop.Config{
|
||||
CacheDir: cacheDir,
|
||||
NoCache: noCache,
|
||||
ConfigFilepath: config,
|
||||
CoinMarketCapAPIKey: cmcAPIKey,
|
||||
APIChoice: apiChoice,
|
||||
Colorscheme: colorscheme,
|
||||
HideMarketbar: hideMarketbar,
|
||||
HideChart: hideChart,
|
||||
HideStatusbar: hideStatusbar,
|
||||
OnlyTable: onlyTable,
|
||||
RefreshRate: refreshRateP,
|
||||
PerPage: perPage,
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return ct.Run()
|
||||
},
|
||||
}
|
||||
|
||||
rootCmd.Flags().BoolVarP(&version, "version", "v", false, "Display current version")
|
||||
rootCmd.Flags().BoolVarP(&test, "test", "", false, "Run test (for Homebrew)")
|
||||
rootCmd.Flags().BoolVarP(&clean, "clean", "", false, "Wipe clean the cache")
|
||||
rootCmd.Flags().BoolVarP(&reset, "reset", "", false, "Reset the config. Make sure to backup any relevant changes first!")
|
||||
rootCmd.Flags().BoolVarP(&hideMarketbar, "hide-marketbar", "", false, "Hide the top marketbar")
|
||||
rootCmd.Flags().BoolVarP(&hideChart, "hide-chart", "", false, "Hide the chart view")
|
||||
rootCmd.Flags().BoolVarP(&hideStatusbar, "hide-statusbar", "", false, "Hide the bottom statusbar")
|
||||
rootCmd.Flags().BoolVarP(&onlyTable, "only-table", "", false, "Show only the table. Hides the chart and top and bottom bars")
|
||||
rootCmd.Flags().BoolVarP(&silent, "silent", "s", false, "Silence log ouput")
|
||||
rootCmd.Flags().BoolVarP(&noCache, "no-cache", "", false, "No cache")
|
||||
rootCmd.Flags().UintVarP(&refreshRate, "refresh-rate", "r", 60, "Refresh rate in seconds. Set to 0 to not auto-refresh")
|
||||
rootCmd.Flags().UintVarP(&perPage, "per-page", "", perPage, "Per page")
|
||||
rootCmd.Flags().StringVarP(&config, "config", "c", "", fmt.Sprintf("Config filepath. (default %s)", cointop.DefaultConfigFilepath))
|
||||
rootCmd.Flags().StringVarP(&cmcAPIKey, "coinmarketcap-api-key", "", "", "Set the CoinMarketCap API key")
|
||||
rootCmd.Flags().StringVarP(&apiChoice, "api", "", cointop.CoinGecko, "API choice. Available choices are \"coinmarketcap\" and \"coingecko\"")
|
||||
rootCmd.Flags().StringVarP(&colorscheme, "colorscheme", "", "", "Colorscheme to use (default \"cointop\"). To install standard themes, do:\n\ngit clone git@github.com:cointop-sh/colors.git ~/.config/cointop/colors\n\nSee git.io/cointop#colorschemes for more info.")
|
||||
rootCmd.Flags().StringVarP(&cacheDir, "cache-dir", "", cacheDir, "Cache directory")
|
||||
|
||||
return rootCmd
|
||||
}
|
||||
@ -0,0 +1,44 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
cssh "github.com/miguelmota/cointop/cointop/ssh"
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
// ServerCmd ...
|
||||
func ServerCmd() *cobra.Command {
|
||||
var port uint = 22
|
||||
var address string = "0.0.0.0"
|
||||
var idleTimeout uint = 60
|
||||
var executableBinary string = "cointop"
|
||||
var hostKeyFile string = cssh.DefaultHostKeyFile
|
||||
|
||||
serverCmd := &cobra.Command{
|
||||
Use: "server",
|
||||
Short: "Run cintop SSH Server",
|
||||
Long: `Run cointop SSH server`,
|
||||
RunE: func(cmd *cobra.Command, args []string) error {
|
||||
server := cssh.NewServer(&cssh.Config{
|
||||
Address: address,
|
||||
Port: port,
|
||||
IdleTimeout: time.Duration(int(idleTimeout)) * time.Second,
|
||||
ExecutableBinary: executableBinary,
|
||||
HostKeyFile: hostKeyFile,
|
||||
})
|
||||
|
||||
fmt.Printf("Running SSH server on port %v\n", port)
|
||||
return server.ListenAndServe()
|
||||
},
|
||||
}
|
||||
|
||||
serverCmd.Flags().UintVarP(&port, "port", "p", port, "Port")
|
||||
serverCmd.Flags().StringVarP(&address, "address", "a", address, "Address")
|
||||
serverCmd.Flags().UintVarP(&idleTimeout, "idle-timeout", "t", idleTimeout, "Idle timeout in seconds")
|
||||
serverCmd.Flags().StringVarP(&executableBinary, "binary", "b", executableBinary, "Executable binary path")
|
||||
serverCmd.Flags().StringVarP(&hostKeyFile, "host-key-file", "k", hostKeyFile, "Host key file")
|
||||
|
||||
return serverCmd
|
||||
}
|
||||
@ -0,0 +1,33 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"github.com/miguelmota/cointop/cointop"
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
// TestCmd ...
|
||||
func TestCmd() *cobra.Command {
|
||||
testCmd := &cobra.Command{
|
||||
Use: "test",
|
||||
Short: "Runs tests",
|
||||
Long: `The test command runs tests for Homebrew`,
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
doTest()
|
||||
},
|
||||
}
|
||||
|
||||
return testCmd
|
||||
}
|
||||
|
||||
// DoTest ...
|
||||
func doTest() {
|
||||
ct, err := cointop.NewCointop(&cointop.Config{
|
||||
NoPrompts: true,
|
||||
})
|
||||
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
|
||||
ct.Exit()
|
||||
}
|
||||
@ -0,0 +1,20 @@
|
||||
package cmd
|
||||
|
||||
import (
|
||||
"github.com/miguelmota/cointop/cointop"
|
||||
"github.com/spf13/cobra"
|
||||
)
|
||||
|
||||
// VersionCmd ...
|
||||
func VersionCmd() *cobra.Command {
|
||||
versionCmd := &cobra.Command{
|
||||
Use: "version",
|
||||
Short: "Displays the current version",
|
||||
Long: `The version command displays the current version`,
|
||||
Run: func(cmd *cobra.Command, args []string) {
|
||||
cointop.PrintVersion()
|
||||
},
|
||||
}
|
||||
|
||||
return versionCmd
|
||||
}
|
||||
@ -0,0 +1,62 @@
|
||||
package asciitable
|
||||
|
||||
import (
|
||||
"strings"
|
||||
|
||||
"github.com/olekukonko/tablewriter"
|
||||
)
|
||||
|
||||
// Input ...
|
||||
type Input struct {
|
||||
Data [][]string
|
||||
Headers []string
|
||||
Alignment []int
|
||||
}
|
||||
|
||||
// AsciiTable ...
|
||||
type AsciiTable struct {
|
||||
table *tablewriter.Table
|
||||
tableString *strings.Builder
|
||||
}
|
||||
|
||||
// NewAsciiTable ...
|
||||
func NewAsciiTable(input *Input) *AsciiTable {
|
||||
tableString := &strings.Builder{}
|
||||
alignment := make([]int, len(input.Alignment))
|
||||
for i, value := range input.Alignment {
|
||||
switch value {
|
||||
case -1:
|
||||
alignment[i] = 3
|
||||
case 0:
|
||||
alignment[i] = 1
|
||||
case 1:
|
||||
alignment[i] = 2
|
||||
}
|
||||
}
|
||||
|
||||
table := tablewriter.NewWriter(tableString)
|
||||
table.SetAutoWrapText(false)
|
||||
table.SetAutoFormatHeaders(false)
|
||||
table.SetHeaderAlignment(tablewriter.ALIGN_RIGHT)
|
||||
table.SetColumnAlignment(alignment)
|
||||
table.SetCenterSeparator("")
|
||||
table.SetColumnSeparator("")
|
||||
table.SetRowSeparator("")
|
||||
table.SetHeaderLine(false)
|
||||
table.SetBorder(false)
|
||||
table.SetTablePadding("\t")
|
||||
table.SetNoWhiteSpace(true)
|
||||
table.SetHeader(input.Headers)
|
||||
table.AppendBulk(input.Data)
|
||||
|
||||
return &AsciiTable{
|
||||
table: table,
|
||||
tableString: tableString,
|
||||
}
|
||||
}
|
||||
|
||||
// String ...
|
||||
func (t *AsciiTable) String() string {
|
||||
t.table.Render()
|
||||
return t.tableString.String()
|
||||
}
|
||||
Loading…
Reference in New Issue