policy.go

package headscale

import (
	"context"
	"net/http"
)

type PolicyResource struct {
	Client HeadscaleClientInterface
}

type Policy struct {
	Policy    string `json:"policy"`
	UpdatedAt string `json:"updated_at"`
}

type UpdatePolicyRequest struct {
	Policy string `json:"policy"`
}

func (p *PolicyResource) Get(ctx context.Context) (Policy, error) {
	var policy Policy

	url := p.Client.buildURL("policy")
	req, err := p.Client.buildRequest(ctx, http.MethodGet, url, requestOptions{})
	if err != nil {
		return policy, err
	}

	err = p.Client.do(ctx, req, &policy)
	return policy, err
}

func (p *PolicyResource) Put(ctx context.Context, policy string) error {
	url := p.Client.buildURL("policy")
	req, err := p.Client.buildRequest(ctx, http.MethodPut, url, requestOptions{
		body: UpdatePolicyRequest{
			Policy: policy,
		},
	})
	if err != nil {
		return err
	}

	return p.Client.do(ctx, req, nil)
}