add query for prices

examples/example.go

@@ -3,5 +3,6 @@ package main
 func main() {
 	queryUserExample()
 	queryConsumptionExample()
+	queryPriceExample()
 	setupWebsocket()
 }

examples/queryPrice.go

@@ -0,0 +1,27 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"log/slog"
+	"os"
+
+	"github.com/SiddyP/gotibber/tibber"
+)
+
+func queryPriceExample() {
+
+	ctx := context.Background()
+
+	t := tibber.Client{
+		APIClient: tibber.NewAPIClient(&tibber.APIConfig{
+			Token: os.Getenv("TIBBER_API_TOKEN"),
+			URL:   "https://api.tibber.com/v1-beta/gql",
+		}),
+		Logger: slog.Default(),
+	}
+
+	p := t.QueryPrice(ctx, &tibber.Price{})
+
+	fmt.Printf("Price: %v\n", p.Viewer.Homes)
+}

tibber/apiclient_test.go

@@ -6,7 +6,7 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
-func TestClient(t *testing.T) {
+func TestCreateClient(t *testing.T) {
 	assert := assert.New(t)
 
 	client := NewAPIClient(&APIConfig{

tibber/queries.go

@@ -16,6 +16,9 @@ type Consumption struct {
 	Last       int
 }
 
+type Price struct {
+}
+
 type WebsocketSubscriptionUrl struct {
 }
 
@@ -91,3 +94,43 @@ func (q *Consumption) query(ctx context.Context, t *Client) HomeConsumptionRespo
 	}
 	return h
 }
+
+func (p *Price) query(ctx context.Context, t *Client) PriceResponse {
+	req := graphql.NewRequest(`
+		query {
+				viewer {
+				homes {
+					currentSubscription{
+					priceInfo{
+						current{
+						total
+						energy
+						tax
+						startsAt
+						}
+						today {
+						total
+						energy
+						tax
+						startsAt
+						}
+						tomorrow {
+						total
+						energy
+						tax
+						startsAt
+						}
+					}
+					}
+				}
+			}
+		}
+		`)
+	setHeaders(req, t)
+
+	var price PriceResponse
+	if err := t.APIClient.GQLClient.Run(ctx, req, &price); err != nil {
+		log.Fatal(err)
+	}
+	return price
+}

tibber/response.go

@@ -31,3 +31,32 @@ type HomeConsumptionResponse struct {
 		} `json:"homes"`
 	} `json:"viewer"`
 }
+
+type PriceResponse struct {
+	Viewer struct {
+		Homes []struct {
+			CurrentSubscription struct {
+				PriceInfo struct {
+					Today []struct {
+						StartsAt time.Time `json:"startsAt"`
+						Total    float64   `json:"total"`
+						Energy   float64   `json:"energy"`
+						Tax      float64   `json:"tax"`
+					} `json:"today"`
+					Tomorrow []struct {
+						StartsAt time.Time `json:"startsAt"`
+						Total    float64   `json:"total"`
+						Energy   float64   `json:"energy"`
+						Tax      float64   `json:"tax"`
+					} `json:"tomorrow"`
+					Current struct {
+						StartsAt time.Time `json:"startsAt"`
+						Total    float64   `json:"total"`
+						Energy   float64   `json:"energy"`
+						Tax      float64   `json:"tax"`
+					} `json:"current"`
+				} `json:"priceInfo"`
+			} `json:"currentSubscription"`
+		} `json:"homes"`
+	} `json:"viewer"`
+}

tibber/tibber.go

@@ -32,6 +32,10 @@ func (t *Client) QueryWebsocketSubscriptionUrl(ctx context.Context, w *Websocket
 	return w.query(ctx, t)
 }
 
+func (t *Client) QueryPrice(ctx context.Context, p *Price) PriceResponse {
+	return p.query(ctx, t)
+}
+
 // Subscribe to Tibber websocket
 func (t *Client) Subscribe(ctx context.Context) {
 	defer t.Wg.Done()

tibber/tibber_test.go

@@ -7,7 +7,7 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
-func TestTibberClient(t *testing.T) {
+func TestCreateTibberClient(t *testing.T) {
 	assert := assert.New(t)
 
 	tC := Client{