Rewrite the client.

agent.go

@@ -96,7 +96,7 @@ func New(cfg Config) (*Agent, error) {
 	if cfg.Identity != nil {
 		id = cfg.Identity
 	}
-	var logger Logger = &defaultLogger{}
+	var logger Logger = new(NoopLogger)
 	if cfg.Logger != nil {
 		logger = cfg.Logger
 	}

client.go

@@ -24,14 +24,14 @@ func NewClient(cfg ClientConfig) Client {
 	return Client{
 		client: http.Client{},
 		config: cfg,
-		logger: &defaultLogger{},
+		logger: new(NoopLogger),
 	}
 }
 
 // NewClientWithLogger creates a new client based on the given configuration and logger.
 func NewClientWithLogger(cfg ClientConfig, logger Logger) Client {
 	if logger == nil {
-		logger = &defaultLogger{}
+		logger = new(NoopLogger)
 	}
 	return Client{
 		client: http.Client{},

logger.go

@@ -4,6 +4,6 @@ type Logger interface {
 	Printf(format string, v ...any)
 }
 
-type defaultLogger struct{}
+type NoopLogger struct{}
 
-func (l defaultLogger) Printf(format string, v ...any) {}
+func (l NoopLogger) Printf(format string, v ...any) {}

pocketic/management.go

@@ -0,0 +1,60 @@
+package pocketic
+
+import (
+	"fmt"
+	"net/http"
+
+	"github.com/aviate-labs/agent-go/candid/idl"
+	"github.com/aviate-labs/agent-go/ic"
+	ic0 "github.com/aviate-labs/agent-go/ic/ic"
+	"github.com/aviate-labs/agent-go/principal"
+)
+
+func (pic PocketIC) AddCycles(canisterID principal.Principal, amount int) (int, error) {
+	var resp struct {
+		Cycles int `json:"cycles"`
+	}
+	if err := pic.do(
+		http.MethodPost,
+		fmt.Sprintf("%s/update/add_cycles", pic.instanceURL()),
+		RawAddCycles{
+			Amount:     amount,
+			CanisterID: canisterID.Raw,
+		},
+		&resp,
+	); err != nil {
+		return 0, err
+	}
+	return resp.Cycles, nil
+}
+
+// CreateCanister creates a canister with default settings as the anonymous principal.
+func (pic PocketIC) CreateCanister() (*principal.Principal, error) {
+	payload, err := idl.Marshal([]any{ProvisionalCreateCanisterArgument{}})
+	if err != nil {
+		return nil, err
+	}
+	raw, err := pic.updateCallWithEP(
+		ic.MANAGEMENT_CANISTER_PRINCIPAL,
+		new(RawEffectivePrincipalNone),
+		principal.AnonymousID,
+		"provisional_create_canister_with_cycles",
+		payload,
+	)
+	if err != nil {
+		return nil, err
+	}
+	var resp struct {
+		CanisterID principal.Principal `ic:"canister_id"`
+	}
+	if err := idl.Unmarshal(raw, []any{&resp}); err != nil {
+		return nil, err
+	}
+	return &resp.CanisterID, nil
+}
+
+type ProvisionalCreateCanisterArgument struct {
+	Settings    *ic0.CanisterSettings `ic:"settings"`
+	SpecifiedID *principal.Principal  `ic:"specified_id"`
+	Amount      *int                  `ic:"amount"`
+}

pocketic/none.go

@@ -0,0 +1,23 @@
+package pocketic
+
+import (
+	"encoding/json"
+	"fmt"
+)
+
+type None struct{}
+
+func (n None) MarshalJSON() ([]byte, error) {
+	return json.Marshal("None")
+}
+
+func (n None) UnmarshalJSON(bytes []byte) error {
+	var s string
+	if err := json.Unmarshal(bytes, &s); err != nil {
+		return err
+	}
+	if s != "None" {
+		return fmt.Errorf("expected None, got %s", s)
+	}
+	return nil
+}

pocketic/none_test.go

@@ -0,0 +1,21 @@
+package pocketic
+
+import (
+	"encoding/json"
+	"testing"
+)
+
+func TestNone(t *testing.T) {
+	var none None
+	raw, err := json.Marshal(none)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if string(raw) != "\"None\"" {
+		t.Fatalf("expected None, got %s", string(raw))
+	}
+	var decoded None
+	if err := json.Unmarshal(raw, &decoded); err != nil {
+		t.Fatal(err)
+	}
+}