add health/provision status related methods (#847)

Author: shreyasj-livekit

agent_client.go

@@ -144,3 +144,19 @@ func (c *AgentClient) ListPrivateLinks(ctx context.Context, req *livekit.ListPri
 	}
 	return c.agentClient.ListPrivateLinks(ctx, req)
 }
+
+func (c *AgentClient) GetPrivateLinkProvisioningStatus(ctx context.Context, req *livekit.GetPrivateLinkProvisioningStatusRequest) (*livekit.GetPrivateLinkProvisioningStatusResponse, error) {
+	ctx, err := c.withAuth(ctx, withAgentGrant{Admin: true})
+	if err != nil {
+		return nil, err
+	}
+	return c.agentClient.GetPrivateLinkProvisioningStatus(ctx, req)
+}
+
+func (c *AgentClient) GetPrivateLinkHealthStatus(ctx context.Context, req *livekit.GetPrivateLinkHealthStatusRequest) (*livekit.GetPrivateLinkHealthStatusResponse, error) {
+	ctx, err := c.withAuth(ctx, withAgentGrant{Admin: true})
+	if err != nil {
+		return nil, err
+	}
+	return c.agentClient.GetPrivateLinkHealthStatus(ctx, req)
+}

pkg/cloudagents/client.go

@@ -137,6 +137,16 @@ func (c *Client) ListPrivateLinks(ctx context.Context, req *lkproto.ListPrivateL
 	return c.AgentClient.ListPrivateLinks(ctx, req)
 }
 
+// GetPrivateLinkProvisioningStatus gets provisioning status for a private link.
+func (c *Client) GetPrivateLinkProvisioningStatus(ctx context.Context, req *lkproto.GetPrivateLinkProvisioningStatusRequest) (*lkproto.GetPrivateLinkProvisioningStatusResponse, error) {
+	return c.AgentClient.GetPrivateLinkProvisioningStatus(ctx, req)
+}
+
+// GetPrivateLinkHealthStatus gets health status for a private link.
+func (c *Client) GetPrivateLinkHealthStatus(ctx context.Context, req *lkproto.GetPrivateLinkHealthStatusRequest) (*lkproto.GetPrivateLinkHealthStatusResponse, error) {
+	return c.AgentClient.GetPrivateLinkHealthStatus(ctx, req)
+}
+
 // uploadAndBuild uploads the source and triggers remote build
 func (c *Client) uploadAndBuild(
 	ctx context.Context,