1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136
|
package runner
import (
"context"
"fmt"
"github.com/hashicorp/go-hclog"
"google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
"github.com/hashicorp/vagrant-plugin-sdk/component"
"github.com/hashicorp/vagrant-plugin-sdk/proto/vagrant_plugin_sdk"
"github.com/hashicorp/vagrant-plugin-sdk/terminal"
"github.com/hashicorp/vagrant/internal/core"
"github.com/hashicorp/vagrant/internal/server/proto/vagrant_server"
)
// executeJob executes an assigned job. This will source the data (if necessary),
// setup the project, execute the job, and return the outcome.
func (r *Runner) executeJob(
ctx context.Context,
log hclog.Logger,
ui terminal.UI,
job *vagrant_server.Job,
wd string,
) (result *vagrant_server.Job_Result, err error) {
// Build our job info
jobInfo := &component.JobInfo{
Id: job.Id,
Local: r.local,
}
log.Debug("processing operation ", "job", job)
// Initial options for setting up the basis
opts := []core.BasisOption{
core.WithLogger(log),
core.WithUI(ui),
core.WithClient(r.client),
core.WithJobInfo(jobInfo),
}
var scope Runs
// Determine our basis reference
var basisRef *vagrant_plugin_sdk.Ref_Basis
var projectRef *vagrant_plugin_sdk.Ref_Project
var targetRef *vagrant_plugin_sdk.Ref_Target
switch s := job.Scope.(type) {
case *vagrant_server.Job_Basis:
basisRef = s.Basis
case *vagrant_server.Job_Project:
projectRef = s.Project
basisRef = s.Project.Basis
case *vagrant_server.Job_Target:
targetRef = s.Target
projectRef = s.Target.Project
basisRef = s.Target.Project.Basis
default:
return nil, fmt.Errorf("invalid job scope %T (%#v)", job.Scope, job.Scope)
}
// Work backwards to setup the basis
opts = append(opts, core.WithBasisRef(basisRef))
// Load our basis
b, err := r.factory.NewBasis(basisRef.ResourceId, opts...)
if err != nil {
return
}
scope = b
// Lets check for a project, and if we have one,
// load it up now
var p *core.Project
if projectRef != nil {
p, err = r.factory.NewProject(
core.WithBasis(b),
core.WithProjectRef(projectRef),
)
if err != nil {
return
}
scope = p
}
// Finally, if we have a target defined, load it up
var m *core.Target
if targetRef != nil {
m, err = r.factory.NewTarget(
core.WithProject(p),
core.WithTargetRef(targetRef),
)
if err != nil {
return
}
scope = m
}
// Execute the operation
log.Info("executing operation", "type", fmt.Sprintf("%T", job.Operation))
switch job.Operation.(type) {
case *vagrant_server.Job_Noop_:
if r.noopCh != nil {
select {
case <-r.noopCh:
case <-ctx.Done():
return nil, ctx.Err()
}
}
log.Debug("noop job success")
return nil, nil
case *vagrant_server.Job_Init:
return r.executeInitOp(ctx, job, b)
case *vagrant_server.Job_Command:
log.Warn("running a run operation", "scope", scope, "job", job)
return r.executeRunOp(ctx, job, scope)
case *vagrant_server.Job_Auth:
return r.executeAuthOp(ctx, log, job, p)
case *vagrant_server.Job_Docs:
return r.executeDocsOp(ctx, log, job, p)
default:
return nil, status.Errorf(codes.Aborted, "unknown operation %T", job.Operation)
}
}
|