package com.github.dockerjava.core;

import static com.google.common.base.Preconditions.checkNotNull;

import java.io.Closeable;

import java.io.File;

import java.io.IOException;

import java.io.InputStream;

import com.github.dockerjava.api.DockerClient;

import com.github.dockerjava.api.command.*;

import com.github.dockerjava.api.model.AuthConfig;

import com.github.dockerjava.core.command.*;

import com.github.dockerjava.jaxrs.DockerCmdExecFactoryImpl;

import com.google.common.base.Preconditions;

/**

* @author Konstantin Pelykh (kpelykh@gmail.com)

*

* @see https://github.com/docker/docker/blob/master/api/client/commands.go

*/

public class DockerClientImpl implements Closeable, DockerClient {

private final DockerClientConfig dockerClientConfig;

private DockerCmdExecFactory dockerCmdExecFactory;

private DockerClientImpl() {

this(DockerClientConfig.createDefaultConfigBuilder().build());

}

private DockerClientImpl(String serverUrl) {

this(configWithServerUrl(serverUrl));

}

private DockerClientImpl(DockerClientConfig dockerClientConfig) {

Preconditions.checkNotNull(dockerClientConfig, "config was not specified");

this.dockerClientConfig = dockerClientConfig;

}

private static DockerClientConfig configWithServerUrl(String serverUrl) {

return DockerClientConfig.createDefaultConfigBuilder()

.withUri(serverUrl)

.build();

}

public static DockerClientImpl getInstance() {

return new DockerClientImpl();

}

public static DockerClientImpl getInstance(DockerClientConfig dockerClientConfig) {

return new DockerClientImpl(dockerClientConfig);

}

public static DockerClientImpl getInstance(String serverUrl) {

return new DockerClientImpl(serverUrl);

}

public DockerClientImpl withDockerCmdExecFactory(

DockerCmdExecFactory dockerCmdExecFactory) {

Preconditions.checkNotNull(dockerCmdExecFactory, "dockerCmdExecFactory was not specified");

this.dockerCmdExecFactory = dockerCmdExecFactory;

this.dockerCmdExecFactory.init(dockerClientConfig);

return this;

}

private DockerCmdExecFactory getDockerCmdExecFactory() {

Preconditions.checkNotNull(dockerCmdExecFactory, "dockerCmdExecFactory was not specified");

return dockerCmdExecFactory;

}

public AuthConfig authConfig() {

checkNotNull(dockerClientConfig.getUsername(), "Configured username is null.");

checkNotNull(dockerClientConfig.getPassword(), "Configured password is null.");

checkNotNull(dockerClientConfig.getEmail(), "Configured email is null.");

AuthConfig authConfig = new AuthConfig();

authConfig.setUsername(dockerClientConfig.getUsername());

authConfig.setPassword(dockerClientConfig.getPassword());

authConfig.setEmail(dockerClientConfig.getEmail());

// TODO Make the registry address configurable

return authConfig;

}

/**

* * MISC API *

*/

/**

* Authenticate with the server, useful for checking authentication.

*/

@Override

public AuthCmd authCmd() {

return new AuthCmdImpl(getDockerCmdExecFactory().createAuthCmdExec(), authConfig());

}

@Override

public InfoCmd infoCmd() {

return new InfoCmdImpl(getDockerCmdExecFactory().createInfoCmdExec());

}

@Override

public PingCmd pingCmd() {

return new PingCmdImpl(getDockerCmdExecFactory().createPingCmdExec());

}

@Override

public VersionCmd versionCmd() {

return new VersionCmdImpl(getDockerCmdExecFactory().createVersionCmdExec());

}

/**

* * IMAGE API *

*/

@Override

public PullImageCmd pullImageCmd(String repository) {

return new PullImageCmdImpl(getDockerCmdExecFactory().createPullImageCmdExec(), repository);

}

@Override

public PushImageCmd pushImageCmd(String name) {

return new PushImageCmdImpl(getDockerCmdExecFactory().createPushImageCmdExec(), name);

}

@Override

public CreateImageCmd createImageCmd(String repository, InputStream imageStream) {

return new CreateImageCmdImpl(getDockerCmdExecFactory().createCreateImageCmdExec(), repository, imageStream);

}

@Override

public SearchImagesCmd searchImagesCmd(String term) {

return new SearchImagesCmdImpl(getDockerCmdExecFactory().createSearchImagesCmdExec(), term);

}

@Override

public RemoveImageCmd removeImageCmd(String imageId) {

return new RemoveImageCmdImpl(getDockerCmdExecFactory().createRemoveImageCmdExec(), imageId);

}

@Override

public ListImagesCmd listImagesCmd() {

return new ListImagesCmdImpl(getDockerCmdExecFactory().createListImagesCmdExec());

}

@Override

public InspectImageCmd inspectImageCmd(String imageId) {

return new InspectImageCmdImpl(getDockerCmdExecFactory().createInspectImageCmdExec(), imageId);

}

/**

* * CONTAINER API *

*/

@Override

public ListContainersCmd listContainersCmd() {

return new ListContainersCmdImpl(getDockerCmdExecFactory().createListContainersCmdExec());

}

@Override

public CreateContainerCmd createContainerCmd(String image) {

return new CreateContainerCmdImpl(getDockerCmdExecFactory().createCreateContainerCmdExec(), image);

}

@Override

public StartContainerCmd startContainerCmd(String containerId) {

return new StartContainerCmdImpl(getDockerCmdExecFactory().createStartContainerCmdExec(), containerId);

}

@Override

public InspectContainerCmd inspectContainerCmd(String containerId) {

return new InspectContainerCmdImpl(getDockerCmdExecFactory().createInspectContainerCmdExec(), containerId);

}

@Override

public RemoveContainerCmd removeContainerCmd(String containerId) {

return new RemoveContainerCmdImpl(getDockerCmdExecFactory().createRemoveContainerCmdExec(), containerId);

}

@Override

public WaitContainerCmd waitContainerCmd(String containerId) {

return new WaitContainerCmdImpl(getDockerCmdExecFactory().createWaitContainerCmdExec(), containerId);

}

@Override

public AttachContainerCmd attachContainerCmd(String containerId) {

return new AttachContainerCmdImpl(getDockerCmdExecFactory().createAttachContainerCmdExec(), containerId);

}

@Override

public LogContainerCmd logContainerCmd(String containerId) {

return new LogContainerCmdImpl(getDockerCmdExecFactory().createLogContainerCmdExec(), containerId);

}

@Override

public CopyFileFromContainerCmd copyFileFromContainerCmd(String containerId, String resource) {

return new CopyFileFromContainerCmdImpl(getDockerCmdExecFactory().createCopyFileFromContainerCmdExec(), containerId, resource);

}

@Override

public ContainerDiffCmd containerDiffCmd(String containerId) {

return new ContainerDiffCmdImpl(getDockerCmdExecFactory().createContainerDiffCmdExec(), containerId);

}

@Override

public StopContainerCmd stopContainerCmd(String containerId) {

return new StopContainerCmdImpl(getDockerCmdExecFactory().createStopContainerCmdExec(), containerId);

}

@Override

public KillContainerCmd killContainerCmd(String containerId) {

return new KillContainerCmdImpl(getDockerCmdExecFactory().createKillContainerCmdExec(), containerId);

}

@Override

public RestartContainerCmd restartContainerCmd(String containerId) {

return new RestartContainerCmdImpl(getDockerCmdExecFactory().createRestartContainerCmdExec(), containerId);

}

@Override

public CommitCmd commitCmd(String containerId) {

return new CommitCmdImpl(getDockerCmdExecFactory().createCommitCmdExec(), containerId);

}

@Override

public BuildImageCmd buildImageCmd(File dockerFolder) {

return new BuildImageCmdImpl(getDockerCmdExecFactory().createBuildImageCmdExec(), dockerFolder);

}

@Override

public BuildImageCmd buildImageCmd(InputStream tarInputStream) {

return new BuildImageCmdImpl(getDockerCmdExecFactory().createBuildImageCmdExec(), tarInputStream);

}

@Override

public TopContainerCmd topContainerCmd(String containerId) {

return new TopContainerCmdImpl(getDockerCmdExecFactory().createTopContainerCmdExec(), containerId);

}

@Override

public TagImageCmd tagImageCmd(String imageId, String repository, String tag) {

return new TagImageCmdImpl(getDockerCmdExecFactory().createTagImageCmdExec(), imageId, repository, tag);

}

@Override

public PauseContainerCmd pauseContainerCmd(String containerId) {

return new PauseContainerCmdImpl(getDockerCmdExecFactory().createPauseContainerCmdExec(), containerId);

}

@Override

public UnpauseContainerCmd unpauseContainerCmd(String containerId) {

return new UnpauseContainerCmdImpl(getDockerCmdExecFactory().createUnpauseContainerCmdExec(), containerId);

}

@Override

public void close() throws IOException {

getDockerCmdExecFactory().close();

}

}