/*
 * Copyright 2017 NAVER Corp.
 *
 * 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.navercorp.pinpoint.web.vo.timeline.inspector;

import com.navercorp.pinpoint.common.server.util.AgentEventType;
import com.navercorp.pinpoint.common.server.util.AgentEventTypeCategory;
import com.navercorp.pinpoint.common.server.util.AgentLifeCycleState;
import com.navercorp.pinpoint.web.vo.AgentEvent;

import java.util.Set;

/**
 * @author HyunGil Jeong
 */
public enum AgentState {
    RUNNING,
    SHUTDOWN,
    UNKNOWN;

    private static final Set<AgentEventType> VALID_EVENT_TYPES = AgentEventType.getTypesByCatgory(AgentEventTypeCategory.AGENT_LIFECYCLE);

    public static AgentState fromAgentLifeCycleState(AgentLifeCycleState state) {
        switch (state) {
            case RUNNING:
                return RUNNING;
            case SHUTDOWN:
            case UNEXPECTED_SHUTDOWN:
                return SHUTDOWN;
            case DISCONNECTED:
            case UNKNOWN:
            default:
                return UNKNOWN;
        }
    }

    public static AgentState fromAgentEvent(AgentEvent agentEvent) {
        AgentEventType eventType = AgentEventType.getTypeByCode(agentEvent.getEventTypeCode());
        if (eventType != null && VALID_EVENT_TYPES.contains(eventType)) {
            switch (eventType) {
                case AGENT_CONNECTED:
                case AGENT_PING:
                    return RUNNING;
                case AGENT_SHUTDOWN:
                case AGENT_UNEXPECTED_SHUTDOWN:
                    return SHUTDOWN;
                case AGENT_CLOSED_BY_SERVER:
                case AGENT_UNEXPECTED_CLOSE_BY_SERVER:
                default:
                    return UNKNOWN;
            }
        } else {
            throw new IllegalArgumentException("agentEvent must be a life cycle event");
        }
    }
}