From 4dc26db12c53f02fdd2e17a143ad769cdd7da2a2 Mon Sep 17 00:00:00 2001
From: Chris Roberts <croberts@hashicorp.com>
Date: Fri, 13 Aug 2021 14:20:22 -0700
Subject: [PATCH] Update index to be free of scope and include factory

---
 internal/core/target_index.go | 93 +++++++++++++++++++++++++++++------
 1 file changed, 77 insertions(+), 16 deletions(-)

diff --git a/internal/core/target_index.go b/internal/core/target_index.go
index 92b5f12f0..01a450da6 100644
--- a/internal/core/target_index.go
+++ b/internal/core/target_index.go
@@ -2,7 +2,9 @@ package core
 
 import (
 	"context"
+	"fmt"
 
+	"github.com/golang/protobuf/ptypes/empty"
 	"github.com/hashicorp/go-hclog"
 	"github.com/hashicorp/vagrant-plugin-sdk/core"
 	"github.com/hashicorp/vagrant-plugin-sdk/proto/vagrant_plugin_sdk"
@@ -15,42 +17,101 @@ type TargetIndex struct {
 	ctx    context.Context
 	logger hclog.Logger
 
-	client  *serverclient.VagrantClient
-	project *Project
+	client *serverclient.VagrantClient
+
+	factory *Factory // used for target loading
 	// The below are resources we need to close when Close is called, if non-nil
 	closers []func() error
 }
 
 func (t *TargetIndex) Delete(uuid string) (err error) {
-	_, err = t.client.DeleteTarget(
-		t.ctx,
-		&vagrant_server.DeleteTargetRequest{
-			Target: &vagrant_plugin_sdk.Ref_Target{ResourceId: uuid},
-		},
-	)
-	return
+	target, err := t.Get(uuid)
+	if err != nil {
+		return
+	}
+
+	return target.(*Target).Destroy()
 }
 
 func (t *TargetIndex) Get(uuid string) (entry core.Target, err error) {
-	return t.project.Target(uuid)
+	// Start with finding the target
+	result, err := t.client.FindTarget(t.ctx, &vagrant_server.FindTargetRequest{
+		Target: &vagrant_server.Target{
+			Uuid: uuid,
+		},
+	})
+	if err != nil {
+		return
+	}
+	return t.loadTarget(result.Target)
 }
 
 func (t *TargetIndex) Includes(uuid string) (exists bool, err error) {
-	_, err = t.project.Target(uuid)
-	if err == nil {
-		return true, nil
+	_, err = t.Get(uuid)
+	if err != nil {
+		// check if 404 error
+		return
 	}
-	return false, nil
+	return true, nil
 }
 
 func (t *TargetIndex) Set(entry core.Target) (updatedEntry core.Target, err error) {
-	updatedEntry = entry.(*Target)
+	updatedEntry, ok := entry.(*Target)
+	if !ok {
+		return nil, fmt.Errorf("cannot save target, invalid type",
+			"target", entry,
+			"type", hclog.Fmt("%T", entry),
+		)
+	}
+
 	err = updatedEntry.Save()
 	return
 }
 
 func (t *TargetIndex) All() (targets []core.Target, err error) {
-	return t.project.Targets()
+	list, err := t.client.ListTargets(t.ctx, &empty.Empty{})
+	if err != nil {
+		return
+	}
+
+	targets = []core.Target{}
+	for _, tInfo := range list.Targets {
+		nt, err := t.loadTarget(&vagrant_server.Target{
+			ResourceId: tInfo.ResourceId,
+		})
+		if err != nil {
+			return nil, err
+		}
+		targets = append(targets, nt)
+	}
+
+	return
+}
+
+func (t *TargetIndex) Close() (err error) {
+	return
+}
+
+func (t *TargetIndex) loadTarget(info *vagrant_server.Target) (target *Target, err error) {
+	// Load the basis
+	b, err := t.factory.New(info.Project.Basis.Name,
+		WithBasisRef(info.Project.Basis))
+	if err != nil {
+		return
+	}
+	// Load the project
+	p, err := b.LoadProject(WithProjectRef(info.Project))
+	if err != nil {
+		return
+	}
+	// Finally, load the target
+	return p.LoadTarget(
+		WithTargetRef(
+			&vagrant_plugin_sdk.Ref_Target{
+				ResourceId: info.ResourceId,
+			},
+		),
+	)
 }
 
 var _ core.TargetIndex = (*TargetIndex)(nil)