/*
  *
  *  *  Copyright 2014 Orient Technologies LTD (info(at)orientechnologies.com)
  *  *
  *  *  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.
  *  *
  *  * For more information: http://www.orientechnologies.com
  *
  */
package com.orientechnologies.orient.core.type;

import java.io.Serializable;

import com.orientechnologies.orient.core.annotation.OAfterDeserialization;
import com.orientechnologies.orient.core.annotation.ODocumentInstance;
import com.orientechnologies.orient.core.id.ORID;
import com.orientechnologies.orient.core.record.impl.ODocument;

/**
 * Base abstract class to wrap a document.
 * 
 * @see ODocumentWrapperNoClass
 * @author Luca Garulli (l.garulli--at--orientechnologies.com)
 * 
 */
@SuppressWarnings("unchecked")
public class ODocumentWrapper implements Serializable {
  @ODocumentInstance
  protected ODocument document;

  public ODocumentWrapper() {
  }

  public ODocumentWrapper(final ORID iRID) {
    this(new ODocument(iRID));
  }

  public ODocumentWrapper(final String iClassName) {
    this(new ODocument(iClassName));
  }

  public ODocumentWrapper(final ODocument iDocument) {
    document = iDocument;
  }

  @OAfterDeserialization
  public void fromStream(final ODocument iDocument) {
    document = iDocument;
  }

  public ODocument toStream() {
    return document;
  }

  public <RET extends ODocumentWrapper> RET load() {
    document = (ODocument) document.load();
    return (RET) this;
  }

  public <RET extends ODocumentWrapper> RET load(final String iFetchPlan) {
    document = document.load(iFetchPlan);
    return (RET) this;
  }

  public <RET extends ODocumentWrapper> RET load(final String iFetchPlan, final boolean iIgnoreCache) {
    document = document.load(iFetchPlan, iIgnoreCache);
    return (RET) this;
  }

  public <RET extends ODocumentWrapper> RET load(final String iFetchPlan, final boolean iIgnoreCache, final boolean loadTombstone) {
    document = document.load(iFetchPlan, iIgnoreCache, loadTombstone);
    return (RET) this;
  }

  public <RET extends ODocumentWrapper> RET reload() {
    document.reload();
    return (RET) this;
  }

  public <RET extends ODocumentWrapper> RET reload(final String iFetchPlan) {
    document.reload(iFetchPlan, true);
    return (RET) this;
  }

  public <RET extends ODocumentWrapper> RET reload(final String iFetchPlan, final boolean iIgnoreCache) {
    document.reload(iFetchPlan, iIgnoreCache);
    return (RET) this;
  }

  public <RET extends ODocumentWrapper> RET save() {
    document.save();
    return (RET) this;
  }

  public <RET extends ODocumentWrapper> RET save(final String iClusterName) {
    document.save(iClusterName);
    return (RET) this;
  }

  public ODocument getDocument() {
    return document;
  }

  @Override
  public int hashCode() {
    final int prime = 31;
    int result = 1;
    result = prime * result + ((document == null) ? 0 : document.hashCode());
    return result;
  }

  @Override
  public boolean equals(Object obj) {
    if (this == obj)
      return true;
    if (obj == null)
      return false;
    if (getClass() != obj.getClass())
      return false;
    final ODocumentWrapper other = (ODocumentWrapper) obj;
    if (document == null) {
      if (other.document != null)
        return false;
    } else if (!document.equals(other.document))
      return false;
    return true;
  }

  @Override
  public String toString() {
    return document != null ? document.toString() : "?";
  }
}