BuildCommand.java

/*
 * Copyright 2011 b1.org
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.b1.pack.cli;

import com.google.common.base.Preconditions;
import org.b1.pack.api.builder.BuilderCommand;
import org.b1.pack.api.builder.BuilderPack;
import org.b1.pack.api.builder.BuilderVolume;
import org.b1.pack.api.builder.PackBuilder;
import org.b1.pack.api.volume.VolumeAllocator;
import org.b1.pack.api.volume.VolumeService;

import java.io.File;
import java.io.IOException;
import java.util.List;
import java.util.Set;

public class BuildCommand implements PackCommand {

    @Override
    public void execute(final ArgSet argSet) throws IOException {
        System.out.println("Starting");
        Preconditions.checkArgument(argSet.getCompression() == null, "Compression not supported in this mode");
        File outputFolder = FileTools.getOutputFolder(argSet);
        final Set<FsObject> fsObjects = FileTools.getFsObjects(argSet.getFileNames());
        PackBuilder builder = PackBuilder.getInstance(argSet.getTypeFormat());
        FsBuilderProvider provider = new FsBuilderProvider(argSet.getMaxVolumeSize());
        List<BuilderVolume> volumes = builder.build(provider, new BuilderCommand() {
            @Override
            public void execute(BuilderPack pack) {
                for (FsObject fsObject : fsObjects) {
                    File file = fsObject.getFile();
                    if (file.isFile()) {
                        pack.addFile(new FsBuilderFile(fsObject));
                    } else if (file.isDirectory()) {
                        pack.addFolder(new FsBuilderFolder(fsObject));
                    } else {
                        throw new IllegalArgumentException("Not found: " + file);
                    }
                }
            }
        });
        VolumeAllocator volumeAllocator = VolumeService.getInstance(argSet.getTypeFormat())
                .createVolumeAllocator(new FsVolumeAllocatorProvider(argSet.getPackName(), argSet.getMaxVolumeSize() != null ? volumes.size() : 0));
        for (int i = 0, volumesSize = volumes.size(); i < volumesSize; i++) {
            String name = volumeAllocator.getVolumeName(i + 1);
            buildVolume(outputFolder == null ? new File(name) : new File(outputFolder, name), volumes.get(i));
        }
        System.out.println();
        System.out.println("Done");
    }

    private void buildVolume(File file, BuilderVolume volume) throws IOException {
        System.out.println();
        System.out.println("Creating volume " + file);
        System.out.println();
        Preconditions.checkState(!file.exists(), "File already exists: %s", file);
        FileTools.saveToFile(volume, file);
    }
}