/*
 * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
 * The software in this package is published under the terms of the CPAL v1.0
 * license, a copy of which has been included with this distribution in the
 * LICENSE.txt file.
 */
package org.mule.runtime.core.transformer.simple;

import org.mule.runtime.api.message.Message;
import org.mule.runtime.api.metadata.DataType;
import org.mule.runtime.core.api.transformer.DiscoverableTransformer;
import org.mule.runtime.core.api.transformer.TransformerException;
import org.mule.runtime.core.transformer.AbstractTransformer;

import java.io.Serializable;
import java.nio.charset.Charset;

/**
 * <code>SerializableToByteArray</code> converts a serializable object or a String to a byte array. If <code>Message</code> is
 * configured as a source type on this transformer by calling {@link #setAcceptMuleMessage(boolean)} then the {@link Message} will
 * be serialised. This is useful for transports such as TCP where the message headers would normally be lost.
 */
public class SerializableToByteArray extends AbstractTransformer implements DiscoverableTransformer {

  /**
   * Give core transformers a slightly higher priority
   */
  private int priorityWeighting = DiscoverableTransformer.DEFAULT_PRIORITY_WEIGHTING + 1;

  public SerializableToByteArray() {
    this.registerSourceType(DataType.fromType(Serializable.class));
    this.setReturnDataType(DataType.BYTE_ARRAY);
  }

  public boolean isAcceptMuleMessage() {
    return this.isSourceDataTypeSupported(DataType.MULE_MESSAGE, true);
  }

  public void setAcceptMuleMessage(boolean value) {
    if (value) {
      this.registerSourceType(DataType.MULE_MESSAGE);
    } else {
      this.unregisterSourceType(DataType.MULE_MESSAGE);
    }
  }

  @Override
  public Object doTransform(Object src, Charset outputEncoding) throws TransformerException {
    /*
     * If the Message source type has been registered then we can assume that the whole message is to be serialised, not just the
     * payload. This can be useful for protocols such as tcp where the protocol does not support headers and the whole message
     * needs to be serialized.
     */

    try {
      return muleContext.getObjectSerializer().getExternalProtocol().serialize(src);
    } catch (Exception e) {
      throw new TransformerException(this, e);
    }
  }

  @Override
  public int getPriorityWeighting() {
    return priorityWeighting;
  }

  @Override
  public void setPriorityWeighting(int priorityWeighting) {
    this.priorityWeighting = priorityWeighting;
  }
}