/*
 * Copyright 2018, TeamDev. All rights reserved.
 *
 * Redistribution and use in source and/or binary forms, with or without
 * modification, must retain the above copyright notice and the following
 * disclaimer.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
package io.spine.server.entity.model;

import com.google.protobuf.Message;
import io.spine.server.entity.Entity;
import io.spine.type.MessageClass;

import static com.google.common.base.Preconditions.checkNotNull;

/**
 * A value object holding a class of an {@linkplain Entity#getState() entity state}.
 *
 * @author Alex Tymchenko
 */
public final class EntityStateClass extends MessageClass<Message> {

    private static final long serialVersionUID = 0L;

    private EntityStateClass(Class<? extends Message> value) {
        super(value);
    }

    public static EntityStateClass of(Entity entity) {
        checkNotNull(entity);
        Message state = entity.getState();

        checkNotNull(state);
        Class<? extends Message> stateClass = state.getClass();

        EntityStateClass result = new EntityStateClass(stateClass);
        return result;
    }

    public static EntityStateClass of(Message entityState) {
        checkNotNull(entityState);
        return new EntityStateClass(entityState.getClass());
    }

    public static EntityStateClass from(Class<? extends Message> value) {
        checkNotNull(value);
        return new EntityStateClass(value);
    }
}

@@ -1549,32 +1549,107 @@
 (value);%0A    %7D%0A%0A
+    /**%0A     * Obtains the class of the state of the given entity.%0A     */%0A
     public stati
@@ -1760,25 +1760,24 @@
 tate();%0A
-%0A
         
 checkNot
@@ -1768,28 +1768,25 @@
         
-checkNotNull
+return of
 (state);
@@ -1794,166 +1794,118 @@
     
+%7D%0A%0A
     
-Class%3C? extends Message%3E stateClass = state.getClass();%0A%0A        EntityStateClass result = new EntityStateClass(stateClass);%0A        return result;
+/**%0A     * Creates an instance of %7B@code EntityStateClass%7D from the class of the given message.
 %0A    
-%7D%0A
+ */
 %0A   
@@ -2004,37 +2004,38 @@
 %0A        return 
-new E
+from(e
 ntityStateClass(
@@ -2032,45 +2032,121 @@
 tate
+.get
 Class(
-entityState.getClass());
+));%0A    %7D%0A%0A    /**%0A     * Creates an instance of %7B@code EntityStateClass%7D from the given class.
 %0A    
-%7D%0A
+ */
 %0A   

Document public API.