web/gui/src/main/java/org/onosproject/ui/impl/ApplicationViewMessageHandler.java

/*
 * Copyright 2015 Open Networking Laboratory
 *
 * 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.onosproject.ui.impl;

import com.fasterxml.jackson.databind.node.ObjectNode;
import com.google.common.collect.ImmutableSet;
import org.onosproject.app.ApplicationAdminService;
import org.onosproject.app.ApplicationService;
import org.onosproject.app.ApplicationState;
import org.onosproject.core.Application;
import org.onosproject.core.ApplicationId;
import org.onosproject.ui.UiMessageHandler;
import org.onosproject.ui.table.AbstractTableRow;
import org.onosproject.ui.table.RowComparator;
import org.onosproject.ui.table.TableRow;
import org.onosproject.ui.table.TableUtils;

import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

import static org.onosproject.app.ApplicationState.ACTIVE;

/**
 * Message handler for application view related messages.
 */
public class ApplicationViewMessageHandler extends UiMessageHandler {

    /**
     * Creates a new message handler for the application messages.
     */
    protected ApplicationViewMessageHandler() {
        super(ImmutableSet.of("appDataRequest", "appManagementRequest"));
    }

    @Override
    public void process(ObjectNode message) {
        String type = eventType(message);
        if (type.equals("appDataRequest")) {
            sendAppList(message);
        } else if (type.equals("appManagementRequest")) {
            processManagementCommand(message);
        }
    }

    private void sendAppList(ObjectNode message) {
        ObjectNode payload = payload(message);
        RowComparator rc = TableUtils.createRowComparator(payload);

        ApplicationService service = get(ApplicationService.class);
        TableRow[] rows = generateTableRows(service);
        Arrays.sort(rows, rc);
        ObjectNode rootNode = mapper.createObjectNode();
        rootNode.set("apps", TableUtils.generateArrayNode(rows));

        connection().sendMessage("appDataResponse", 0, rootNode);
    }

    private void processManagementCommand(ObjectNode message) {
        ObjectNode payload = payload(message);
        String action = string(payload, "action");
        String name = string(payload, "name");
        if (action != null && name != null) {
            ApplicationAdminService service = get(ApplicationAdminService.class);
            ApplicationId appId = service.getId(name);
            if (action.equals("activate")) {
                service.activate(appId);
            } else if (action.equals("deactivate")) {
                service.deactivate(appId);
            } else if (action.equals("uninstall")) {
                service.uninstall(appId);
            }
            sendAppList(message);
        }
    }

    private TableRow[] generateTableRows(ApplicationService service) {
        List<TableRow> list = service.getApplications().stream()
                .map(application -> new ApplicationTableRow(service, application))
                .collect(Collectors.toList());
        return list.toArray(new TableRow[list.size()]);
    }

    /**
     * TableRow implementation for
     * {@link org.onosproject.core.Application applications}.
     */
    private static class ApplicationTableRow extends AbstractTableRow {

        private static final String STATE = "state";
        private static final String STATE_IID = "_iconid_state";
        private static final String ID = "id";
        private static final String VERSION = "version";
        private static final String ORIGIN = "origin";
        private static final String DESC = "desc";

        private static final String[] COL_IDS = {
                STATE, STATE_IID, ID, VERSION, ORIGIN, DESC
        };

        private static final String ICON_ID_ACTIVE = "active";
        private static final String ICON_ID_INACTIVE = "appInactive";


        public ApplicationTableRow(ApplicationService service, Application app) {
            ApplicationState state = service.getState(app.id());
            String iconId = state == ACTIVE ? ICON_ID_ACTIVE : ICON_ID_INACTIVE;

            add(STATE, state);
            add(STATE_IID, iconId);
            add(ID, app.id().name());
            add(VERSION, app.version());
            add(ORIGIN, app.origin());
            add(DESC, app.description());
        }

        @Override
        protected String[] columnIds() {
            return COL_IDS;
        }
    }

}