device: allow disabling configuration

Author: nyiyui

cmd/device-client/main.go

@@ -24,7 +24,8 @@ import (
 )
 
 type Config struct {
-	Clients map[string]ClientConfig
+	Clients    map[string]ClientConfig
+	CanFoaward bool
 }
 
 type ClientConfig struct {
@@ -202,12 +203,15 @@ func createGoroutines(m *MachineData, dnsClient dns.Client, config Config) {
 	for clientName, cc := range config.Clients {
 		go func(clientName string, cc ClientConfig) {
 			c, err := device.NewClient(&http.Client{
-				Timeout: 5 * time.Second,
-				//Transport: cc.transport,
+				Timeout:   5 * time.Second,
+				Transport: cc.transport,
 			}, cc.BaseURL, cc.Token, cc.Network, cc.Device, cc.PrivateKey)
 			if err != nil {
 				panic(err)
 			}
+			if !config.CanFoaward {
+				c.SetCanForward(false)
+			}
 			c.Machine = m.Machines[clientName]
 			c.SetDNSClient(dnsClient)
 			continuous := new(device.ContinousClient)

device/client.go

@@ -9,6 +9,7 @@ import (
 	"math/rand"
 	"net/http"
 	"net/url"
+	"slices"
 	"sync"
 
 	"github.com/nyiyui/qrystal/coord"
@@ -29,6 +30,7 @@ type Client struct {
 	goalHandle *goal.Handle
 	dns        dns.Client
 	dnsLock    sync.Mutex
+	canForward bool
 
 	spec       spec.SpecCensored
 	token      util.Token
@@ -65,6 +67,10 @@ func NewClient(httpClient *http.Client, baseURL string, token util.Token, networ
 	}, nil
 }
 
+func (c *Client) SetCanForward(canForward bool) {
+	c.canForward = canForward
+}
+
 func (c *Client) SetDNSClient(client dns.Client) {
 	c.dnsLock.Lock()
 	defer c.dnsLock.Unlock()
@@ -281,23 +287,25 @@ func (c *Client) patchAccessible(nc *spec.NetworkCensored) error {
 	zap.S().Debugf("ndcI = %s", ndcI)
 
 	var accessible []string
-	for _, ndc := range nc.Devices {
-		if ndc.ForwarderAndEndpointChosen {
-			accessible = append(accessible, ndc.Name)
+	if c.canForward {
+		for _, ndc := range nc.Devices {
+			if ndc.ForwarderAndEndpointChosen {
+				accessible = append(accessible, ndc.Name)
+			}
 		}
+		zap.S().Debugf("I can forward for %d devices.", len(accessible))
+	} else {
+		zap.S().Debug("I cannot forward by configuration.")
 	}
-	if len(accessible) != 0 {
+	if !slices.Equal(accessible, nc.Devices[ndcI].Accessible) {
 		nc.Devices[ndcI].Accessible = accessible
-		zap.S().Debugf("I can forward for %d devices.", len(accessible))
 		err := c.patchSpec(coord.PatchReifySpecRequest{
 			Accessible:    accessible,
 			AccessibleSet: true,
 		})
 		if err != nil {
 			return fmt.Errorf("patch spec: %w", err)
 		}
-	} else {
-		zap.S().Infof("I can't forward for any devices.")
 	}
 	return nil
 }