feat: info added

agent/agent.proto

@@ -7,7 +7,7 @@ import "common/common.proto";
 
 service Rpc {
   rpc instances (GetInstancesRequest) returns (GetInstancesReply) {};
-  rpc info (GetInfoRequest) returns (GetInfoReply) {};
+  rpc info (common.GetInfoRequest) returns (common.GetInfoReply) {};
   rpc shell (stream common.ShellRequest) returns (stream common.ShellReply) {};
   rpc launch (common.LaunchRequest) returns (common.LaunchReply) {};
 }

agent/client.go

@@ -15,23 +15,23 @@ type client struct {
 }
 
 type Client interface {
-	Instances(ctx context.Context) (*GetInstancesReply, error)
-	Info(ctx context.Context) (*GetInfoReply, error)
+	Instances(ctx context.Context, request *GetInstancesRequest) (*GetInstancesReply, error)
+	Info(ctx context.Context, request *common.GetInfoRequest) (*common.GetInfoReply, error)
 	Shell(ctx context.Context) (grpc.BidiStreamingClient[common.ShellRequest, common.ShellReply], error)
 	Close() error
-	Launch(ctx context.Context, launchRequest *common.LaunchRequest) (*common.LaunchReply, error)
+	Launch(ctx context.Context, request *common.LaunchRequest) (*common.LaunchReply, error)
 }
 
 func (c *client) Close() error {
 	return c.conn.Close()
 }
 
-func (c *client) Instances(ctx context.Context) (*GetInstancesReply, error) {
-	return c.client.Instances(ctx, &GetInstancesRequest{})
+func (c *client) Instances(ctx context.Context, request *GetInstancesRequest) (*GetInstancesReply, error) {
+	return c.client.Instances(ctx, request)
 }
 
-func (c *client) Info(ctx context.Context) (*GetInfoReply, error) {
-	return c.client.Info(ctx, &GetInfoRequest{})
+func (c *client) Info(ctx context.Context, request *common.GetInfoRequest) (*common.GetInfoReply, error) {
+	return c.client.Info(ctx, request)
 }
 
 func (c *client) Launch(ctx context.Context, launchRequest *common.LaunchRequest) (*common.LaunchReply, error) {

agent/server.go

@@ -87,20 +87,12 @@ func (s *server) Instances(_ context.Context, _ *GetInstancesRequest) (*GetInsta
 	}, nil
 }
 
-func (s *server) Launch(_ context.Context, req *common.LaunchRequest) (*common.LaunchReply, error) {
-	if err := s.multipassClient.Launch(context.Background(), req.InstanceName); err != nil {
-		return nil, err
-	}
-
-	return &common.LaunchReply{}, nil
+func (s *server) Launch(ctx context.Context, req *common.LaunchRequest) (*common.LaunchReply, error) {
+	return s.multipassClient.Launch(ctx, req)
 }
 
-func (s *server) Info(_ context.Context, _ *GetInfoRequest) (*GetInfoReply, error) {
-	resource := s.state.GetState().Resource
-
-	return &GetInfoReply{
-		Resource: resource,
-	}, nil
+func (s *server) Info(ctx context.Context, req *common.GetInfoRequest) (*common.GetInfoReply, error) {
+	return s.multipassClient.Info(ctx, req)
 }
 
 type windowSize struct {

agent/state.go

@@ -59,7 +59,6 @@ func (s *state) updateResources() {
 		log.Printf("error while getting virtual memory: %v", err)
 		return
 	}
-
 	cpuInfoStats, err := cpu.Info()
 	if err != nil {
 		log.Printf("error while getting cpu info: %v", err)
@@ -72,25 +71,12 @@ func (s *state) updateResources() {
 	}
 	totalCore := cpuInfoStats[0].Cores
 	availableCore := totalCore - int32(math.Ceil(float64(totalCore)*percents[0]/100))
-
-	partitions, err := disk.Partitions(true)
+	diskUsage, err := disk.Usage("/")
 	if err != nil {
-		log.Printf("error while getting disk partitions: %v", err)
+		log.Printf("error while getting disk usage: %v", err)
 		return
 	}
 
-	var diskUsageTotal uint64
-	var diskUsageFree uint64
-	for _, partition := range partitions {
-		diskUsageStat, err := disk.Usage(partition.Mountpoint)
-		if err != nil {
-			log.Printf("error while getting disk usage: %v", err)
-			return
-		}
-		diskUsageTotal += diskUsageStat.Total
-		diskUsageFree += diskUsageStat.Free
-	}
-
 	s.Resource = &Resource{
 		Cpu: &CPU{
 			Total:     totalCore,
@@ -101,8 +87,8 @@ func (s *state) updateResources() {
 			Available: virtualMemoryStat.Available,
 		},
 		Disk: &Disk{
-			Total:     diskUsageTotal,
-			Available: diskUsageFree,
+			Total:     diskUsage.Total,
+			Available: diskUsage.Free,
 		},
 	}
 }