package ai.vespa.hosted.plugin;

import ai.vespa.hosted.api.Deployment;
import ai.vespa.hosted.api.DeploymentLog;
import ai.vespa.hosted.api.DeploymentResult;
import com.yahoo.config.provision.ApplicationId;
import com.yahoo.config.provision.zone.ZoneId;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.plugin.MojoFailureException;
import org.apache.maven.plugins.annotations.Mojo;
import org.apache.maven.plugins.annotations.Parameter;

import java.nio.file.Paths;
import java.time.ZoneOffset;
import java.time.format.DateTimeFormatter;

/**
 * Deploys a Vespa application package to the hosted Vespa API.
 *
 * @author jonmv
 */
@Mojo(name = "deploy")
public class DeployMojo extends AbstractVespaDeploymentMojo {

    @Parameter(property = "applicationZip")
    private String applicationZip;

    @Parameter(property = "vespaVersion")
    private String vespaVersion;

    @Parameter(property = "follow", defaultValue = "true")
    private boolean follow;

    @Override
    protected void doExecute() throws MojoFailureException, MojoExecutionException {
        Deployment deployment = Deployment.ofPackage(Paths.get(firstNonBlank(applicationZip,
                                                                             projectPathOf("target", "application.zip"))));
        if (vespaVersion != null) deployment = deployment.atVersion(vespaVersion);

        ZoneId zone = zoneOf(environment, region);
        DeploymentResult result = controller.deploy(deployment, id, zone);
        getLog().info(result.message());

        if (follow) tailLogs(id, zone, result.run());
    }

    private void tailLogs(ApplicationId id, ZoneId zone, long run) throws MojoFailureException, MojoExecutionException {
        long last = -1;
        DeploymentLog log;
        while (true) {
            log = controller.deploymentLog(id, zone, run, last);
            for (DeploymentLog.Entry entry : log.entries())
                print(entry);
            last = log.last().orElse(last);

            if ( ! log.isActive())
                break;

            try {
                Thread.sleep(1000);
            }
            catch (InterruptedException e) {
                Thread.currentThread().interrupt();
                break;
            }
        }
        switch (log.status()) {
            case success:            return;
            case error:              throw new MojoExecutionException("Unexpected error during deployment; see log for details");
            case aborted:            throw new MojoFailureException("Deployment was aborted, probably by a newer deployment");
            case outOfCapacity:      throw new MojoFailureException("No capacity left in zone; please contact the Vespa team");
            case deploymentFailed:   throw new MojoFailureException("Deployment failed; see log for details");
            case installationFailed: throw new MojoFailureException("Installation failed; see Vespa log for details");
            case running:            throw new MojoFailureException("Deployment not completed");
            case testFailure:        throw new IllegalStateException("Unexpected status; tests are not run for manual deployments");
            default:                 throw new IllegalArgumentException("Unexpected status '" + log.status() + "'");
        }
    }

    private static final DateTimeFormatter formatter = DateTimeFormatter.ofPattern("HH:mm:ss").withZone(ZoneOffset.UTC);
    private static final String padding = "\n" + " ".repeat(23);

    private void print(DeploymentLog.Entry entry) {
        String timestamp = formatter.format(entry.at());
        String message = String.join(padding, entry.message().split("\n"))
                               .replaceAll("\\s*\n", "\n").trim();
        switch (entry.level()) {
            case "warning" : getLog().warn(" [" + timestamp + "]  " + message); break;
            case "error" : getLog().error("   [" + timestamp + "]  " + message); break;
            default: getLog().info("    [" + timestamp + "]  " + message); break;
        }
    }

}

@@ -3803,24 +3803,114 @@
 .level()) %7B%0A
+            case %22error%22   : getLog().error(%22   %5B%22 + timestamp + %22%5D  %22 + message); break;%0A
             
@@ -3939,16 +3939,19 @@
 g().warn
+   
 (%22 %5B%22 + 
@@ -4005,22 +4005,24 @@
   case %22
-error%22
+info%22   
  : getLo
@@ -4017,39 +4017,39 @@
 %22    : getLog().
-error
+info
 (%22
+ 
    %5B%22 + timestam
@@ -4096,16 +4096,24 @@
  default
+        
 : getLog
@@ -4107,39 +4107,39 @@
      : getLog().
-info
+debug
 (%22
- 
    %5B%22 + timestam

Make debug log entries (from Vespa log) go to debug log method