/*
 * Copyright (c) 2007-2009, Osmorc Development Team
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without modification,
 * are permitted provided that the following conditions are met:
 *     * Redistributions of source code must retain the above copyright notice, this list
 *       of conditions and the following disclaimer.
 *     * Redistributions in binary form must reproduce the above copyright notice, this
 *       list of conditions and the following disclaimer in the documentation and/or other
 *       materials provided with the distribution.
 *     * Neither the name of 'Osmorc Development Team' nor the names of its contributors may be
 *       used to endorse or promote products derived from this software without specific
 *       prior written permission.
 * 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 org.osmorc.manifest.lang.psi.impl;

import com.intellij.lang.ASTNode;
import com.intellij.psi.PsiElement;
import com.intellij.psi.stubs.IStubElementType;
import com.intellij.util.IncorrectOperationException;
import org.jetbrains.annotations.NotNull;
import org.osmorc.manifest.lang.ManifestTokenType;
import org.osmorc.manifest.lang.psi.Header;
import org.osmorc.manifest.lang.psi.ManifestToken;
import org.osmorc.manifest.lang.psi.stub.HeaderStub;

/**
 * @author Robert F. Beeger (robert@beeger.net)
 */
public class HeaderImpl extends ManifestElementBase<HeaderStub> implements Header {
    public HeaderImpl(HeaderStub stub, @NotNull IStubElementType nodeType) {
        super(stub, nodeType);
    }

    public HeaderImpl(@NotNull ASTNode node) {
        super(node);
    }

    public String getName() {
        String result;
        HeaderStub stub = getStub();
        if (stub != null) {
            result = stub.getName();    
        } else {
            ManifestToken nameToken = getNameToken();
            result = nameToken != null ? nameToken.getText() : null;
        }

        return result != null ? result : "";
    }

    public PsiElement setName(@NotNull String name) throws IncorrectOperationException {
        ManifestToken nameToken = getNameToken();
        if (nameToken != null) {
            nameToken.replaceToken(name);
        }
        return this;
    }

    public ManifestToken getNameToken() {
        return (ManifestToken) getNode().findChildByType(ManifestTokenType.HEADER_NAME);
    }

    public ManifestToken getColonToken() {
        return (ManifestToken) getNode().findChildByType(ManifestTokenType.COLON);
    }
}