-
Notifications
You must be signed in to change notification settings - Fork 5
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #34 from bonedaddy/charts
Charts
- Loading branch information
Showing
7 changed files
with
186 additions
and
7 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,17 +1,17 @@ | ||
| FROM golang:1.15-alpine3.12 AS build-env | ||
| ARG VERSION | ||
| RUN apk add build-base linux-headers | ||
| ARG VERSION | ||
| ENV BUILD_HOME=/BUILD | ||
| RUN mkdir -p ${BUILD_HOME} | ||
| ADD . ${BUILD_HOME} | ||
| WORKDIR ${BUILD_HOME} | ||
|
|
||
| ADD . ${BUILD_HOME} | ||
| RUN go mod download | ||
| RUN go build -o /bin/gondx \ | ||
| -ldflags "-X main.Version=$VERSION" \ | ||
| ./cmd/gondx | ||
|
|
||
| FROM alpine:3.12 | ||
| RUN apk add --no-cache tini | ||
| COPY --from=build-env /bin/gondx /bin/gondx | ||
| ENTRYPOINT ["/bin/gondx", "--config", "/config.yml" ] | ||
| CMD [ "--startup.sleep", "discord", "ndx-bot", "--update.database" ] |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,153 @@ | ||
| package discord | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "log" | ||
| "strings" | ||
|
|
||
| "github.com/bonedaddy/dgc" | ||
| "github.com/bonedaddy/go-indexed/db" | ||
| "github.com/wcharczuk/go-chart/v2" | ||
| "github.com/wcharczuk/go-chart/v2/drawing" | ||
| ) | ||
|
|
||
| func (c *Client) priceWindowChart(ctx *dgc.Ctx) { | ||
|
|
||
| arguments := ctx.Arguments | ||
| pair := arguments.Get(0).Raw() | ||
| window, err := arguments.Get(1).AsInt() | ||
| if err != nil { | ||
| ctx.RespondText("window argument is not a number") | ||
| return | ||
| } | ||
| var ( | ||
| prices []*db.Price | ||
| ) | ||
| // valid the allowed currencies | ||
| switch strings.ToLower(pair) { | ||
| case "defi5-dai": | ||
| prices, err = c.db.PricesInRange("defi5", window) | ||
| if err != nil { | ||
| ctx.RespondText("failed to get price") | ||
| log.Println("defi5 dai price change fetch failed: ", err) | ||
| return | ||
| } | ||
| case "cc10-dai": | ||
| prices, err = c.db.PricesInRange("cc10", window) | ||
| if err != nil { | ||
| ctx.RespondText("failed to get price") | ||
| log.Println("cc10 dai price change fetch failed: ", err) | ||
| return | ||
| } | ||
| case "ndx-dai": | ||
| prices, err = c.db.PricesInRange("ndx", window) | ||
| if err != nil { | ||
| ctx.RespondText("failed to get price") | ||
| log.Println("ndx dai price change fetch failed: ", err) | ||
| return | ||
| } | ||
| case "eth-dai": | ||
| prices, err = c.db.PricesInRange("eth", window) | ||
| if err != nil { | ||
| ctx.RespondText("failed to get price") | ||
| log.Println("eth dai price change fetch failed: ", err) | ||
| return | ||
| } | ||
| default: | ||
| ctx.RespondText("invalid currency requested must be one of: defi5-dai, cc10-dai, eth-dai, ndx-dai") | ||
| return | ||
| } | ||
| priceSeries := chart.TimeSeries{ | ||
| Name: "price", | ||
| Style: chart.Style{ | ||
| StrokeColor: chart.GetDefaultColor(0), | ||
| }, | ||
| } | ||
| // records prices at every hour | ||
| priceHourSeries := chart.TimeSeries{ | ||
| Style: chart.Style{ | ||
| StrokeColor: chart.GetDefaultColor(0), | ||
| }, | ||
| } | ||
| var ( | ||
| // set = make(map[int]bool) | ||
| hourSet = make(map[int]map[int]bool) | ||
| minSet = make(map[int]map[int]map[int]bool) | ||
| // track days -> hours | ||
| ) | ||
| for _, price := range prices { | ||
| if hourSet[price.CreatedAt.Day()] == nil { | ||
| hourSet[price.CreatedAt.Day()] = make(map[int]bool) | ||
| } | ||
| if minSet[price.CreatedAt.Day()] == nil { | ||
| minSet[price.CreatedAt.Day()] = make(map[int]map[int]bool) | ||
| } | ||
| if minSet[price.CreatedAt.Day()][price.CreatedAt.Hour()] == nil { | ||
| minSet[price.CreatedAt.Day()][price.CreatedAt.Hour()] = make(map[int]bool) | ||
| } | ||
| // make sure we only update the hour set one per hour | ||
| if !hourSet[price.CreatedAt.Day()][price.CreatedAt.Hour()] { | ||
| priceHourSeries.XValues = append(priceHourSeries.XValues, price.CreatedAt) | ||
| priceHourSeries.YValues = append(priceHourSeries.YValues, price.USDPrice) | ||
| hourSet[price.CreatedAt.Day()][price.CreatedAt.Hour()] = true | ||
| } | ||
| // make sure we only record one price per minute | ||
| if !minSet[price.CreatedAt.Day()][price.CreatedAt.Hour()][price.CreatedAt.Minute()] { | ||
| priceSeries.XValues = append(priceSeries.XValues, price.CreatedAt) | ||
| priceSeries.YValues = append(priceSeries.YValues, price.USDPrice) | ||
| } | ||
| } | ||
|
|
||
| hourlySMASeries := &chart.SMASeries{ | ||
| Name: "hourly sma", | ||
| Period: window, | ||
| Style: chart.Style{ | ||
| StrokeColor: drawing.ColorRed, | ||
| }, | ||
| InnerSeries: priceHourSeries, | ||
| } | ||
| hourlyEMASeries := &chart.EMASeries{ | ||
| Name: "hourly ema", | ||
| Period: window, | ||
| Style: chart.Style{ | ||
| StrokeColor: drawing.ColorGreen, | ||
| }, | ||
| InnerSeries: priceHourSeries, | ||
| } | ||
|
|
||
| graph := chart.Chart{ | ||
| Background: chart.Style{ | ||
| Padding: chart.Box{ | ||
| Left: 45, | ||
| }, | ||
| }, | ||
| XAxis: chart.XAxis{ | ||
| TickPosition: chart.TickPositionBetweenTicks, | ||
| // ensure we render date timestamps with minute granularity | ||
| ValueFormatter: chart.TimeMinuteValueFormatter, | ||
| }, | ||
| Series: []chart.Series{ | ||
| priceSeries, | ||
| hourlySMASeries, | ||
| hourlyEMASeries, | ||
| }, | ||
| } | ||
|
|
||
| //note we have to do this as a separate step because we need a reference to graph | ||
| graph.Elements = []chart.Renderable{ | ||
| chart.LegendLeft(&graph), | ||
| } | ||
|
|
||
| buffer := bytes.NewBuffer(nil) | ||
| if err := graph.Render(chart.PNG, buffer); err != nil { | ||
| log.Println("failed to render SMA: ", err) | ||
| ctx.RespondText("failed to render SMA") | ||
| return | ||
| } | ||
|
|
||
| if _, err := ctx.Session.ChannelFileSend(ctx.Event.ChannelID, "chart.png", buffer); err != nil { | ||
| log.Println("failed to upload chart: ", err) | ||
| ctx.RespondText("failed to upload chart") | ||
| return | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters