com.metreeca.mark.tasks.Clean Maven / Gradle / Ivy

Go to download
/*
 * Copyright © 2019-2023 Metreeca srl
 *
 * 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 com.metreeca.mark.tasks;

import com.metreeca.mark.*;

import java.io.IOException;
import java.io.UncheckedIOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.stream.Stream;

import static java.util.Comparator.reverseOrder;

/**
 * Site cleaning task.
 *
 * Removes the {@linkplain Opts#target() target} site folder.
 */
public final class Clean implements Task {

    @Override public void exec(final Mark mark) {

        final Path source=mark.source();
        final Path target=mark.target();

        if ( source.equals(target) ) { // in-place generation › remove volatile files

            try ( final Stream sources=Files.walk(mark.source()) ) {

                // delete generated files // !!! actually generated?

                mark.scan(sources).forEach(file -> delete(target.resolve(file.path())));

                // delete bundled assets // !!! actually bundled?

                mark.assets().forEach((path, resource) -> delete(target.resolve(path)));

            } catch ( final IOException e ) {
                throw new UncheckedIOException(e);
            }

        } else if ( Files.exists(target) ) { // delete target folder

            try ( final Stream walk=Files.walk(target) ) {

                walk.sorted(reverseOrder()).forEachOrdered(this::delete);

            } catch ( final IOException e ) {
                throw new UncheckedIOException(e);
            }

        }
    }


    private void delete(final Path path) {
        try {

            if ( Files.exists(path) ) { Files.delete(path); }

        } catch ( final IOException e ) {
            throw new UncheckedIOException(e);
        }
    }

}