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

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mark-maven-plugin Show documentation
A minimalist static site generator.
There is a newer version: 0.9.1
/*
 * Copyright © 2019-2022 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.jse.JSEServer;
import com.metreeca.mark.*;
import com.metreeca.rest.Response;
import com.metreeca.rest.services.Logger;

import org.apache.maven.plugin.logging.Log;

import java.awt.*;
import java.io.*;
import java.net.InetSocketAddress;
import java.net.URI;
import java.nio.charset.Charset;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Optional;
import java.util.concurrent.BlockingDeque;
import java.util.concurrent.LinkedBlockingDeque;
import java.util.function.*;
import java.util.stream.Stream;

import static com.metreeca.rest.Response.OK;
import static com.metreeca.rest.Wrapper.postprocessor;
import static com.metreeca.rest.formats.OutputFormat.output;
import static com.metreeca.rest.formats.TextFormat.text;
import static com.metreeca.rest.handlers.Publisher.publisher;
import static com.metreeca.rest.handlers.Router.router;
import static com.metreeca.rest.services.Logger.logger;
import static com.metreeca.rest.wrappers.Server.server;

import static java.lang.String.format;
import static java.nio.charset.StandardCharsets.UTF_8;

/**
 * Site serving task.
 *
 * Generates a processed version of the {@linkplain Opts#source() source} site folder in the
 * {@linkplain Opts#target() target} site folder and serves it on a development grade server for testing purposes,
 * watching the {@linkplain Opts#source() source} site folder for further changes to by synchronized.
 */
public final class Serve implements Task {

	private final InetSocketAddress address=new InetSocketAddress("localhost", 2020);

	private final BlockingDeque updates=new LinkedBlockingDeque<>();


	////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

	@Override public void exec(final Mark mark) {
		serve(mark);
		watch(mark);
	}


	////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

	private void serve(final Opts mark) {

		final Thread daemon=new Thread(() -> {
			try {

				new JSEServer()

						.address(address)

						.delegate(context -> context

								.set(logger(), () -> new MavenLogger(mark.logger()))

								.get(() -> server().wrap(router()

										.path("/~", router()
												.get(request -> request.reply(response -> {
													try {

														return response.status(OK).body(text(), updates.take());

													} catch ( final InterruptedException e ) {

														return response.status(OK);

													}
												}))
										)

										.path("/*", router()
												.get(publisher(mark.target())
														.with(postprocessor(this::rewrite))
												)
										)

								))
						)

						.start();

				if ( Desktop.isDesktopSupported() && Desktop.getDesktop().isSupported(Desktop.Action.BROWSE) ) {
					Desktop.getDesktop().browse(URI.create(format(
							"http://%s:%d/", address.getHostString(), address.getPort()
					)));
				}


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

		});

		daemon.setDaemon(true);
		daemon.start();
	}

	private void watch(final Mark mark) {

		final Path root=Paths.get("/");

		final Thread daemon=new Thread(() -> mark.watch((kind, target) -> {

			final String path=root.resolve(mark.target().relativize(target)).toString();

			Stream.of("", ".html", "index.html").forEach(suffix -> {
				if ( path.endsWith(suffix) ) { updates.offer(path.substring(0, path.length()-suffix.length())); }
			});

		}));

		daemon.setDaemon(true);
		daemon.start();
	}


	////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

	private static final String Live=Optional

			.ofNullable(Serve.class.getResource(Serve.class.getSimpleName()+".js"))

			.map(url -> {

				try ( InputStream input=url.openStream() ) {

					return new String(input.readAllBytes(), UTF_8);

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

			})

			.map(script -> format("", script))

			.orElse("");


	private Response rewrite(final Response response) {
		if ( response.header("Content-Type").filter(mime -> mime.startsWith("text/html")).isPresent() ) {

			return response.body(output()).fold(e -> response, target -> {

				final ByteArrayOutputStream buffer=new ByteArrayOutputStream(1000);

				target.accept(buffer);

				final Charset charset=Charset.forName(response.charset());

				final byte[] body=buffer.toString(charset)
						.replace("", Live+"")
						.getBytes(charset);

				return response
						.header("Content-Length", String.valueOf(body.length))
						.body(output(), output -> {
							try {
								output.write(body);
							} catch ( final IOException e ) {
								throw new UncheckedIOException(e);
							}
						});

			});

		} else {

			return response;

		}
	}


	////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

	private static final class MavenLogger extends Logger {

		private final Log logger;


		private MavenLogger(final Log logger) {
			this.logger=logger;
		}


		@Override public Logger entry(
				final Level level, final Object source,
				final Supplier message, final Throwable cause
		) {

			if ( cause == null ) {

				final Consumer sink=(level == Level.error) ? logger::error
						: (level == Level.warning) ? logger::warn
						: (level == Level.info) ? logger::info
						: logger::debug;

				sink.accept(message.get());


			} else {

				final BiConsumer sink=(level == Level.error) ? logger::error
						: (level == Level.warning) ? logger::warn
						: (level == Level.info) ? logger::info
						: logger::debug;

				sink.accept(message.get(), cause);

			}

			return this;
		}
	}

}