org.darkware.wpman.WPManagerApplication.java Source code

Introduction

Here is the source code for org.darkware.wpman.WPManagerApplication.java

Source

/*******************************************************************************
 * Copyright (c) 2016. darkware.org and contributors
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 ******************************************************************************/

package org.darkware.wpman;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.dataformat.yaml.YAMLFactory;
import io.dropwizard.Application;
import io.dropwizard.setup.Bootstrap;
import io.dropwizard.setup.Environment;
import org.darkware.wpman.config.ReloadableWordpressConfig;
import org.darkware.wpman.config.WordpressConfig;
import org.darkware.wpman.rest.*;
import org.darkware.wpman.rest.health.NoopHealthCheck;
import org.darkware.wpman.util.JSONHelper;
import org.darkware.wpman.util.serialization.*;
import org.darkware.wpman.wpcli.WPCLI;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.nio.file.Files;

/**
 * The {@code WPManagerApplication} class is the primary application driver for the DropWizard framework.
 *
 * @author jeff
 * @since 2016-03-26
 */
public class WPManagerApplication extends Application<WPManagerConfiguration> {
    /** A public {@code Logger} facility provided as a default logging destination. */
    public static Logger log = LoggerFactory.getLogger("Startup");

    /**
     * The primary entry point for DropWizard execution.
     *
     * @param args The command line arguments.
     * @throws Exception If the application encounters an error.
     */
    public static void main(String[] args) throws Exception {
        new WPManagerApplication().run(args);
    }

    @Override
    public String getName() {
        return "WPManager";
    }

    @Override
    public void initialize(Bootstrap<WPManagerConfiguration> bootstrap) {
        // Register serialization helpers
        this.registerMappingModules(bootstrap.getObjectMapper());

        ContextManager.local().registerInstance(bootstrap.getObjectMapper());

        JSONHelper.use(bootstrap.getObjectMapper());
    }

    private void registerMappingModules(final ObjectMapper mapper) {
        mapper.registerModule(new PathModule());
        mapper.registerModule(new VersionModule());
        mapper.registerModule(new PluginStatusModule());
        mapper.registerModule(new WPDateModule());
        mapper.registerModule(new PermissiveBooleanModule());
        mapper.registerModule(new WPObjectTypeModule());
        mapper.registerModule(new WPActionModule());
        mapper.registerModule(new TimeWindowModule());
    }

    @Override
    public void run(WPManagerConfiguration configuration, Environment environment) {
        WPManagerApplication.log.info("WP-CLI is at: " + configuration.getWpcli().getBinaryPath());

        // Make a new object mapper with YAML support.
        //   The currently registered ContextManager object isn't set up for Yaml support.
        ObjectMapper om = new ObjectMapper(new YAMLFactory());
        this.registerMappingModules(om);

        WordpressConfig config = new ReloadableWordpressConfig(configuration.getPolicyFile(), om);

        // Create the all-important manager object.
        WPManager manager = new WPManager(config);

        environment.jersey().register(new UtilityResource());
        environment.jersey().register(new ConfigResource(manager));
        environment.jersey().register(new PluginResource(manager));
        environment.jersey().register(new ThemeResource(manager));
        environment.jersey().register(new BlogResource(manager));
        environment.jersey().register(new CronResource(manager));
        environment.jersey().register(new ActionResource(manager));
        environment.jersey().register(new CoreResource(manager));

        final NoopHealthCheck healthCheck = new NoopHealthCheck();
        environment.healthChecks().register("noop", healthCheck);

        // Initialize a WPManager
        WPCLI.setPath(configuration.getWpcli().getBinaryPath());
        if (Files.notExists(configuration.getWpcli().getBinaryPath()))
            WPCLI.update();
        manager.start();
    }
}