in internal/cmds/cmds.go [146:269]
func enable(ctx *log.Context, h types.HandlerEnvironment, report *types.RunCommandInstanceView, metadata types.RCMetadata, c types.Cmd) (string, string, error, int) {
// parse the extension handler settings (not available prior to 'enable')
cfg, err1 := handlersettings.GetHandlerSettings(h.HandlerEnvironment.ConfigFolder, metadata.ExtName, metadata.SeqNum, ctx)
if err1 != nil {
return "", "", errors.Wrap(err1, "failed to get configuration"), constants.ExitCode_GetHandlerSettingsFailed
}
exitCode, err := immediatecmds.Enable(ctx, h, metadata.ExtName, metadata.SeqNum, cfg)
// If there is an error or the customer requested to install the script as a service, return the error and exit code immediately.
if err != nil || cfg.InstallAsService() {
return "", "", err, exitCode
}
dir := filepath.Join(metadata.DownloadPath, fmt.Sprintf("%d", metadata.SeqNum))
scriptFilePath, err := downloadScript(ctx, dir, &cfg)
if err != nil {
return "",
"",
errors.Wrap(err, fmt.Sprintf("File downloads failed. Use either a public script URI that points to .sh file, Azure storage blob SAS URI or storage blob accessible by a managed identity and retry. If managed identity is used, make sure it has been given access to container of storage blob '%s' with 'Storage Blob Data Reader' role assignment. In case of user-assigned identity, make sure you add it under VM's identity. For more info, refer https://aka.ms/RunCommandManagedLinux", download.GetUriForLogging(cfg.ScriptURI()))),
constants.ExitCode_ScriptBlobDownloadFailed
}
err = downloadArtifacts(ctx, dir, &cfg)
if err != nil {
return "", "",
errors.Wrap(err, "Artifact downloads failed. Use either a public artifact URI that points to .sh file, Azure storage blob SAS URI, or storage blob accessible by a managed identity and retry."),
constants.ExitCode_DownloadArtifactFailed
}
blobCreateOrReplaceError := "Error creating AppendBlob '%s' using SAS token or Managed identity. Please use a valid blob SAS URI with [read, append, create, write] permissions OR managed identity. If managed identity is used, make sure Azure blob and identity exist, and identity has been given access to storage blob's container with 'Storage Blob Data Contributor' role assignment. In case of user-assigned identity, make sure you add it under VM's identity and provide outputBlobUri / errorBlobUri and corresponding clientId in outputBlobManagedIdentity / errorBlobManagedIdentity parameter(s). In case of system-assigned identity, do not use outputBlobManagedIdentity / errorBlobManagedIdentity parameter(s). For more info, refer https://aka.ms/RunCommandManagedLinux"
var outputBlobSASRef *storage.Blob
var outputBlobAppendClient *appendblob.Client
var outputBlobAppendCreateOrReplaceError error
outputFilePosition := int64(0)
// Create or Replace outputBlobURI if provided. Fail the command if create or replace fails.
if cfg.OutputBlobURI != "" {
outputBlobSASRef, outputBlobAppendClient, outputBlobAppendCreateOrReplaceError = createOrReplaceAppendBlob(cfg.OutputBlobURI,
cfg.ProtectedSettings.OutputBlobSASToken, cfg.ProtectedSettings.OutputBlobManagedIdentity, ctx)
if outputBlobAppendCreateOrReplaceError != nil {
return "",
"",
errors.Wrap(outputBlobAppendCreateOrReplaceError, fmt.Sprintf(blobCreateOrReplaceError, cfg.OutputBlobURI)),
constants.ExitCode_BlobCreateOrReplaceFailed
}
}
var errorBlobSASRef *storage.Blob
var errorBlobAppendClient *appendblob.Client
var errorBlobAppendCreateOrReplaceError error
errorFilePosition := int64(0)
// Create or Replace errorBlobURI if provided. Fail the command if create or replace fails.
if cfg.ErrorBlobURI != "" {
errorBlobSASRef, errorBlobAppendClient, errorBlobAppendCreateOrReplaceError = createOrReplaceAppendBlob(cfg.ErrorBlobURI,
cfg.ProtectedSettings.ErrorBlobSASToken, cfg.ProtectedSettings.ErrorBlobManagedIdentity, ctx)
if errorBlobAppendCreateOrReplaceError != nil {
return "",
"",
errors.Wrap(errorBlobAppendCreateOrReplaceError, fmt.Sprintf(blobCreateOrReplaceError, cfg.ErrorBlobURI)),
constants.ExitCode_BlobCreateOrReplaceFailed
}
}
// AsyncExecution requested by customer means the extension should report successful extension deployment to complete the provisioning state
// Later the full extension output will be reported
statusToReport := types.StatusTransitioning
if cfg.AsyncExecution {
ctx.Log("message", "asycExecution is true - report success")
statusToReport = types.StatusSuccess
instanceview.ReportInstanceView(ctx, h, metadata, statusToReport, c, report)
}
stdoutF, stderrF := exec.LogPaths(dir)
// Implement ticker to update extension status periodically
ticker := time.NewTicker(updateStatusInSeconds * time.Second)
done := make(chan bool)
go func() {
for {
select {
case <-done:
return
case <-ticker.C:
ctx.Log("event", "report partial status")
stdoutTail, stderrTail := getOutput(ctx, stdoutF, stderrF)
report.Output = stdoutTail
report.Error = stderrTail
instanceview.ReportInstanceView(ctx, h, metadata, statusToReport, c, report)
outputFilePosition, err = appendToBlob(stdoutF, outputBlobSASRef, outputBlobAppendClient, outputFilePosition, ctx)
errorFilePosition, err = appendToBlob(stderrF, errorBlobSASRef, errorBlobAppendClient, errorFilePosition, ctx)
}
}
}()
// execute the command, save its error
runErr, exitCode := runCmd(ctx, dir, scriptFilePath, &cfg, metadata)
ticker.Stop()
done <- true
// collect the logs if available
stdoutTail, stderrTail := getOutput(ctx, stdoutF, stderrF)
isSuccess := runErr == nil
telemetryResult("Output", "-- stdout/stderr omitted from telemetry pipeline --", isSuccess, 0)
if isSuccess {
ctx.Log("event", "enabled")
} else {
ctx.Log("event", "enable script failed")
}
// Report the output streams to blobs
outputFilePosition, err = appendToBlob(stdoutF, outputBlobSASRef, outputBlobAppendClient, outputFilePosition, ctx)
errorFilePosition, err = appendToBlob(stderrF, errorBlobSASRef, errorBlobAppendClient, errorFilePosition, ctx)
c.Functions.Cleanup(ctx, metadata, h, cfg.PublicSettings.RunAsUser)
return stdoutTail, stderrTail, runErr, exitCode
}