Handle different flag responses

Pushkar Acharya · Pushkar Acharya · commit 021bb195ac4c · 2026-02-05T15:45:37.000-08:00
diff --git a/internal/crusoe/nfs_flag.go b/internal/crusoe/nfs_flag.go
@@ -24,8 +24,9 @@ var (
 	errUnmarshalFlag     = errors.New("failed to unmarshal flag response")
 )
 
-type NfsFlagResponse struct {
-	Status bool `json:"status"`
+type FlagResponse struct {
+	Status  bool `json:"status,omitempty"`
+	Enabled bool `json:"enabled,omitempty"`
 }
 
 // getFlag is a helper function to fetch a boolean flag from the API.
@@ -59,14 +60,14 @@ func getFlag(ctx context.Context, crusoeHTTPClient *http.Client, flagRoute strin
 		return false, fmt.Errorf("%w: HTTP %d: %s", errGetFlag, resp.StatusCode, string(bodyBytes))
 	}
 
-	var flagResponse NfsFlagResponse
+	var flagResponse FlagResponse
 
 	unmarshalErr := json.Unmarshal(bodyBytes, &flagResponse)
 	if unmarshalErr != nil {
 		return false, fmt.Errorf("%w: %w (response body: %q)", errUnmarshalFlag, unmarshalErr, string(bodyBytes))
 	}
 
-	return flagResponse.Status, nil
+	return flagResponse.Status || flagResponse.Enabled, nil
 }
 
 // GetNFSFlag returns true if the project has NFS enabled.

Original file line number	Diff line number	Diff line change
`@@ -24,8 +24,9 @@ var (`
`24`	`24`	`errUnmarshalFlag = errors.New("failed to unmarshal flag response")`
`25`	`25`	`)`
`26`	`26`
`27`		`-type NfsFlagResponse struct {`
`28`		- Status bool `json:"status"`
	`27`	`+type FlagResponse struct {`
	`28`	+ Status bool `json:"status,omitempty"`
	`29`	+ Enabled bool `json:"enabled,omitempty"`
`29`	`30`	`}`
`30`	`31`
`31`	`32`	`// getFlag is a helper function to fetch a boolean flag from the API.`
`@@ -59,14 +60,14 @@ func getFlag(ctx context.Context, crusoeHTTPClient *http.Client, flagRoute strin`
`59`	`60`	`return false, fmt.Errorf("%w: HTTP %d: %s", errGetFlag, resp.StatusCode, string(bodyBytes))`
`60`	`61`	`}`
`61`	`62`
`62`		`- var flagResponse NfsFlagResponse`
	`63`	`+ var flagResponse FlagResponse`
`63`	`64`
`64`	`65`	`unmarshalErr := json.Unmarshal(bodyBytes, &flagResponse)`
`65`	`66`	`if unmarshalErr != nil {`
`66`	`67`	`return false, fmt.Errorf("%w: %w (response body: %q)", errUnmarshalFlag, unmarshalErr, string(bodyBytes))`
`67`	`68`	`}`
`68`	`69`
`69`		`- return flagResponse.Status, nil`
	`70`	`+ return flagResponse.Status \|\| flagResponse.Enabled, nil`
`70`	`71`	`}`
`71`	`72`
`72`	`73`	`// GetNFSFlag returns true if the project has NFS enabled.`