feat: Add support for GitHub Budgets API

github/billing_budgets.go

@@ -0,0 +1,114 @@
+// Copyright 2025 The go-github AUTHORS. All rights reserved.
+//
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+
+package github
+
+import (
+	"context"
+	"fmt"
+)
+
+// Budget represents a GitHub budget.
+type Budget struct {
+	ID                     *string         `json:"id,omitempty"`
+	BudgetName             *string         `json:"budget_name,omitempty"`
+	TargetSubAccount       *string         `json:"target_sub_account,omitempty"`
+	TargetType             *string         `json:"target_type,omitempty"`
+	TargetID               *int64          `json:"target_id,omitempty"`
+	TargetName             *string         `json:"target_name,omitempty"`
+	PricingModel           *string         `json:"pricing_model,omitempty"`
+	PricingModelID         *string         `json:"pricing_model_id,omitempty"`
+	PricingModelDisplayName *string         `json:"pricing_model_display_name,omitempty"`
+	BudgetType             *string         `json:"budget_type,omitempty"`
+	LimitAmount            *float64        `json:"limit_amount,omitempty"`
+	CurrentAmount          *float64        `json:"current_amount,omitempty"`
+	Currency               *string         `json:"currency,omitempty"`
+	ExcludeCostCenterUsage *bool           `json:"exclude_cost_center_usage,omitempty"`
+	BudgetAlerting         *BudgetAlerting `json:"budget_alerting,omitempty"`
+}
+
+// BudgetAlerting represents the alerting configuration for a budget.
+type BudgetAlerting struct {
+	WillAlert       *bool    `json:"will_alert,omitempty"`
+	AlertRecipients []string `json:"alert_recipients,omitempty"`
+}
+
+// ListOrganizationBudgets lists all budgets for an organization.
+//
+// GitHub API docs: https://docs.github.com/rest/billing/budgets#get-all-budgets-for-an-organization
+//
+// meta:operation GET /organizations/{org}/settings/billing/budgets
+func (s *BillingService) ListOrganizationBudgets(ctx context.Context, org string) ([]*Budget, *Response, error) {
+	u := fmt.Sprintf("organizations/%v/settings/billing/budgets", org)
+	req, err := s.client.NewRequest("GET", u, nil)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	var budgets []*Budget
+	resp, err := s.client.Do(ctx, req, &budgets)
+	if err != nil {
+		return nil, resp, err
+	}
+
+	return budgets, resp, nil
+}
+
+// GetOrganizationBudget gets a specific budget for an organization.
+//
+// GitHub API docs: https://docs.github.com/rest/billing/budgets#get-a-budget-for-an-organization
+//
+// meta:operation GET /organizations/{org}/settings/billing/budgets/{budget_id}
+func (s *BillingService) GetOrganizationBudget(ctx context.Context, org string, budgetID string) (*Budget, *Response, error) {
+	u := fmt.Sprintf("organizations/%v/settings/billing/budgets/%v", org, budgetID)
+	req, err := s.client.NewRequest("GET", u, nil)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	budget := new(Budget)
+	resp, err := s.client.Do(ctx, req, budget)
+	if err != nil {
+		return nil, resp, err
+	}
+
+	return budget, resp, nil
+}
+
+// UpdateOrganizationBudget updates a specific budget for an organization.
+//
+// GitHub API docs: https://docs.github.com/rest/billing/budgets#update-a-budget-for-an-organization
+//
+// meta:operation PATCH /organizations/{org}/settings/billing/budgets/{budget_id}
+func (s *BillingService) UpdateOrganizationBudget(ctx context.Context, org string, budgetID string, budget *Budget) (*Budget, *Response, error) {
+	u := fmt.Sprintf("organizations/%v/settings/billing/budgets/%v", org, budgetID)
+	req, err := s.client.NewRequest("PATCH", u, budget)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	updatedBudget := new(Budget)
+	resp, err := s.client.Do(ctx, req, updatedBudget)
+	if err != nil {
+		return nil, resp, err
+	}
+
+	return updatedBudget, resp, nil
+}
+
+// DeleteOrganizationBudget deletes a specific budget for an organization.
+//
+// GitHub API docs: https://docs.github.com/rest/billing/budgets#delete-a-budget-for-an-organization
+//
+// meta:operation DELETE /organizations/{org}/settings/billing/budgets/{budget_id}
+func (s *BillingService) DeleteOrganizationBudget(ctx context.Context, org string, budgetID string) (*Response, error) {
+	u := fmt.Sprintf("organizations/%v/settings/billing/budgets/%v", org, budgetID)
+	req, err := s.client.NewRequest("DELETE", u, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	return s.client.Do(ctx, req, nil)
+}

github/billing_budgets_test.go

@@ -0,0 +1,130 @@
+// Copyright 2025 The go-github AUTHORS. All rights reserved.
+//
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+
+package github
+
+import (
+	"fmt"
+	"net/http"
+	"testing"
+
+	"github.com/google/go-cmp/cmp"
+)
+
+func TestBillingService_ListOrganizationBudgets(t *testing.T) {
+	t.Parallel()
+	client, mux, _ := setup(t)
+
+	mux.HandleFunc("/organizations/o/settings/billing/budgets", func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, "GET")
+		fmt.Fprint(w, `[
+			{
+				"id": "1",
+				"budget_name": "Budget 1",
+				"limit_amount": 100.5,
+				"budget_alerting": {
+					"will_alert": true,
+					"alert_recipients": ["user1"]
+				}
+			}
+		]`)
+	})
+
+	ctx := t.Context()
+	budgets, _, err := client.Billing.ListOrganizationBudgets(ctx, "o")
+	if err != nil {
+		t.Errorf("Billing.ListOrganizationBudgets returned error: %v", err)
+	}
+
+	want := []*Budget{
+		{
+			ID:          Ptr("1"),
+			BudgetName:  Ptr("Budget 1"),
+			LimitAmount: Ptr(100.5),
+			BudgetAlerting: &BudgetAlerting{
+				WillAlert:       Ptr(true),
+				AlertRecipients: []string{"user1"},
+			},
+		},
+	}
+	if !cmp.Equal(budgets, want) {
+		t.Errorf("Billing.ListOrganizationBudgets returned %+v, want %+v", budgets, want)
+	}
+}
+
+func TestBillingService_GetOrganizationBudget(t *testing.T) {
+	t.Parallel()
+	client, mux, _ := setup(t)
+
+	mux.HandleFunc("/organizations/o/settings/billing/budgets/1", func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, "GET")
+		fmt.Fprint(w, `{
+			"id": "1",
+			"budget_name": "Budget 1"
+		}`)
+	})
+
+	ctx := t.Context()
+	budget, _, err := client.Billing.GetOrganizationBudget(ctx, "o", "1")
+	if err != nil {
+		t.Errorf("Billing.GetOrganizationBudget returned error: %v", err)
+	}
+
+	want := &Budget{
+		ID:         Ptr("1"),
+		BudgetName: Ptr("Budget 1"),
+	}
+	if !cmp.Equal(budget, want) {
+		t.Errorf("Billing.GetOrganizationBudget returned %+v, want %+v", budget, want)
+	}
+}
+
+func TestBillingService_UpdateOrganizationBudget(t *testing.T) {
+	t.Parallel()
+	client, mux, _ := setup(t)
+
+	input := &Budget{
+		BudgetName: Ptr("Updated Budget"),
+	}
+
+	mux.HandleFunc("/organizations/o/settings/billing/budgets/1", func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, "PATCH")
+		testBody(t, r, `{"budget_name":"Updated Budget"}`+"\n")
+		fmt.Fprint(w, `{
+			"id": "1",
+			"budget_name": "Updated Budget"
+		}`)
+	})
+
+	ctx := t.Context()
+	budget, _, err := client.Billing.UpdateOrganizationBudget(ctx, "o", "1", input)
+	if err != nil {
+		t.Errorf("Billing.UpdateOrganizationBudget returned error: %v", err)
+	}
+
+	want := &Budget{
+		ID:         Ptr("1"),
+		BudgetName: Ptr("Updated Budget"),
+	}
+	if !cmp.Equal(budget, want) {
+		t.Errorf("Billing.UpdateOrganizationBudget returned %+v, want %+v", budget, want)
+	}
+}
+
+func TestBillingService_DeleteOrganizationBudget(t *testing.T) {
+	t.Parallel()
+	client, mux, _ := setup(t)
+
+	mux.HandleFunc("/organizations/o/settings/billing/budgets/1", func(w http.ResponseWriter, r *http.Request) {
+		testMethod(t, r, "DELETE")
+		w.WriteHeader(http.StatusNoContent)
+	})
+
+	ctx := t.Context()
+	_, err := client.Billing.DeleteOrganizationBudget(ctx, "o", "1")
+	if err != nil {
+		t.Errorf("Billing.DeleteOrganizationBudget returned error: %v", err)
+	}
+}

github/enterprise_billing_budgets.go

@@ -0,0 +1,110 @@
+// Copyright 2025 The go-github AUTHORS. All rights reserved.
+//
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+
+package github
+
+import (
+	"context"
+	"fmt"
+)
+
+// ListEnterpriseBudgets lists all budgets for an enterprise.
+//
+// GitHub API docs: https://docs.github.com/rest/billing/budgets#get-all-budgets-for-an-enterprise
+//
+// meta:operation GET /enterprises/{enterprise}/settings/billing/budgets
+func (s *EnterpriseService) ListEnterpriseBudgets(ctx context.Context, enterprise string) ([]*Budget, *Response, error) {
+	u := fmt.Sprintf("enterprises/%v/settings/billing/budgets", enterprise)
+	req, err := s.client.NewRequest("GET", u, nil)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	var budgets []*Budget
+	resp, err := s.client.Do(ctx, req, &budgets)
+	if err != nil {
+		return nil, resp, err
+	}
+
+	return budgets, resp, nil
+}
+
+// GetEnterpriseBudget gets a specific budget for an enterprise.
+//
+// GitHub API docs: https://docs.github.com/rest/billing/budgets#get-a-budget-for-an-enterprise
+//
+// meta:operation GET /enterprises/{enterprise}/settings/billing/budgets/{budget_id}
+func (s *EnterpriseService) GetEnterpriseBudget(ctx context.Context, enterprise string, budgetID string) (*Budget, *Response, error) {
+	u := fmt.Sprintf("enterprises/%v/settings/billing/budgets/%v", enterprise, budgetID)
+	req, err := s.client.NewRequest("GET", u, nil)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	budget := new(Budget)
+	resp, err := s.client.Do(ctx, req, budget)
+	if err != nil {
+		return nil, resp, err
+	}
+
+	return budget, resp, nil
+}
+
+// CreateEnterpriseBudget creates a specific budget for an enterprise.
+//
+// GitHub API docs: https://docs.github.com/rest/billing/budgets#create-a-budget
+//
+// meta:operation POST /enterprises/{enterprise}/settings/billing/budgets
+func (s *EnterpriseService) CreateEnterpriseBudget(ctx context.Context, enterprise string, budget *Budget) (*Budget, *Response, error) {
+	u := fmt.Sprintf("enterprises/%v/settings/billing/budgets", enterprise)
+	req, err := s.client.NewRequest("POST", u, budget)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	createdBudget := new(Budget)
+	resp, err := s.client.Do(ctx, req, createdBudget)
+	if err != nil {
+		return nil, resp, err
+	}
+
+	return createdBudget, resp, nil
+}
+
+// UpdateEnterpriseBudget updates a specific budget for an enterprise.
+//
+// GitHub API docs: https://docs.github.com/rest/billing/budgets#update-a-budget-for-an-enterprise
+//
+// meta:operation PATCH /enterprises/{enterprise}/settings/billing/budgets/{budget_id}
+func (s *EnterpriseService) UpdateEnterpriseBudget(ctx context.Context, enterprise string, budgetID string, budget *Budget) (*Budget, *Response, error) {
+	u := fmt.Sprintf("enterprises/%v/settings/billing/budgets/%v", enterprise, budgetID)
+	req, err := s.client.NewRequest("PATCH", u, budget)
+	if err != nil {
+		return nil, nil, err
+	}
+
+	updatedBudget := new(Budget)
+	resp, err := s.client.Do(ctx, req, updatedBudget)
+	if err != nil {
+		return nil, resp, err
+	}
+
+	return updatedBudget, resp, nil
+}
+
+// DeleteEnterpriseBudget deletes a specific budget for an enterprise.
+//
+// GitHub API docs: https://docs.github.com/rest/billing/budgets#delete-a-budget-for-an-enterprise
+//
+// meta:operation DELETE /enterprises/{enterprise}/settings/billing/budgets/{budget_id}
+func (s *EnterpriseService) DeleteEnterpriseBudget(ctx context.Context, enterprise string, budgetID string) (*Response, error) {
+	u := fmt.Sprintf("enterprises/%v/settings/billing/budgets/%v", enterprise, budgetID)
+	req, err := s.client.NewRequest("DELETE", u, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	return s.client.Do(ctx, req, nil)
+}