AddCommand.java
package org.docascode.api;
import org.docascode.api.core.DocAsCodeRepository;
import org.docascode.api.core.chrono.generated.*;
import org.docascode.api.core.chrono.*;
import org.docascode.api.core.errors.DocAsCodeException;
import org.docascode.api.core.errors.ProcessingException;
import org.docascode.api.event.Event;
import java.io.File;
public class AddCommand extends DocAsCodeCommand<Artifact> {
private File file;
private String id;
private String classifier = null;
private String outputName = null;
private String output = null;
private String groupID = null;
private String artifactID = null;
private String version = null;
AddCommand(DocAsCodeRepository repo) {
super(repo);
}
public AddCommand setFile(File location){
return set(id,classifier,
location,
outputName,output,
groupID,artifactID,version);
}
public AddCommand setOutput(String outputName, String output){
return set(id,classifier,
file,
outputName,output,
groupID,artifactID,version);
}
public AddCommand setMaven(String groupID, String artifactID, String version){
return set(id,classifier,
file,
outputName,output,
groupID,artifactID,version);
}
public AddCommand setID(String id, String classifier) {
return set(id,classifier,
file,
outputName,output,
groupID,artifactID,version);
}
public AddCommand set(String id,
String classifier,
File file,
String outputName,
String output,
String groupID,
String artifactID,
String version){
this.id = id;
this.classifier = classifier;
this.file = file;
this.outputName = outputName;
this.output = output;
this.groupID = groupID;
this.artifactID = artifactID;
this.version = version;
return this;
}
private void sanityCheck() throws DocAsCodeException {
if (id == null){
throw new DocAsCodeException("Missing mandatory field 'id'.");
}
if (file == null || !file.exists()){
log(String.format("The file '%s' doesn't exist.",this.file), Event.Level.WARN);
}
}
private Artifact getArtifact(String id) throws ProcessingException {
Delivery delivery = getRepository().chrono().getDelivery();
Artifact a;
if (delivery.getArtifacts() == null) {
delivery.setArtifacts(new ArtifactMap());
}
if (delivery.getArtifacts().containsKey(id)) {
a = delivery.getArtifacts().get(id);
} else {
a = new Artifact();
}
if (a.getAttach() == null) {
a.setAttach(new AttachMap());
}
return a;
}
private void attach(Artifact a){
BaseArtifact attach;
if (a.getAttach().containsKey(this.classifier)){
attach = a.getAttach().get(this.classifier);
} else {
attach = new BaseArtifact();
}
if (attach.getOutputs() == null) {
attach.setOutputs(new OutputMap());
}
if (this.file != null) {
attach.setFile(getRepository().relativize(this.file));
} else {
if (attach.getFile() == null || attach.getFile().equals("")){
attach.setFile(null);
}
}
if (this.output != null) {
attach.getOutputs().put(this.outputName, this.output);
}
a.getAttach().put(this.classifier,attach);
}
private void attachGAV(Artifact a){
if (version != null && groupID != null && artifactID!=null) {
Maven n;
if (a.getMaven() == null) {
n = new Maven();
} else {
n = a.getMaven();
}
n.setVersion(version);
n.setGroupId(groupID);
n.setArtifactId(artifactID);
a.setMaven(n);
}
}
@Override
public Artifact call() throws DocAsCodeException {
sanityCheck();
Delivery delivery = getRepository().chrono().getDelivery();
Artifact a = getArtifact(this.id);
if (this.classifier != null){
attach(a);
} else {
if (a.getOutputs() == null) {
a.setOutputs(new OutputMap());
}
if (this.file != null) {
a.setFile(getRepository().relativize(this.file));
} else {
if (a.getFile() == null || a.getFile().equals("")){
a.setFile(null);
}
}
if (this.output != null) {
a.getOutputs().put(this.outputName, this.output);
}
}
attachGAV(a);
delivery.getArtifacts().put(this.id,a);
getRepository().chrono().save(delivery);
return a;
}
}