trunk/components/camel-scala/src/main/scala/org/apache/camel/scala/dsl/SAbstractDefinition.scala

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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 org.apache.camel
package scala
package dsl

import org.apache.camel.model.DataFormatDefinition
import org.apache.camel.builder.PredicateBuilder.toPredicate
import org.apache.camel.{Predicate, Exchange}

import org.apache.camel.model._
import org.apache.camel.spi.Policy


import org.apache.camel.processor.aggregate.AggregationStrategy

import org.apache.camel.scala.dsl.builder.RouteBuilder
import reflect.Manifest
import java.lang.String
import java.util.Comparator

abstract class SAbstractDefinition[P <: ProcessorDefinition[_]] extends DSL with Wrapper[P] with Block {

  val target : P
  val unwrap = target
  implicit val builder: RouteBuilder
  implicit def expressionBuilder(expression: Exchange => Any) = new ScalaExpression(expression)
  implicit def predicateBuilder(predicate: Exchange => Any) = new ScalaPredicate(predicate)
  
  def -->(uris: String*) = to(uris:_*)
  def to(uris: String*) = {
    uris.length match {
      case 1 => target.to(uris(0))
      case _ => {
        val multi = multicast
        uris.foreach(multi.to(_))
      }
    }
    this
  }
  
  def when(filter: Exchange => Any) : DSL with Block = SChoiceDefinition(target.choice).when(filter)
    
  def as[Target](toType: Class[Target]) = wrap(target.convertBodyTo(toType))

  def attempt : STryDefinition = STryDefinition(target.doTry)
  
  def split(expression: Exchange => Any) = SSplitDefinition(target.split(expression))
    
  def apply(block: => Unit) = {
    builder.build(this, block)
    this
  }

  def bean(bean: Any) = bean match {
    case cls: Class[_] => wrap(target.bean(cls))
    case ref: String => wrap(target.beanRef(ref))
    case obj: Any => wrap(target.bean(obj))
  }
  
  def choice = SChoiceDefinition(target.choice)
  
  def enrich(uri: String, strategy: AggregationStrategy) = wrap(target.enrich(uri, strategy))

  def filter(predicate: Exchange => Any) = SFilterDefinition(target.filter(predicateBuilder(predicate)))
    
  def otherwise : SChoiceDefinition = 
    throw new Exception("otherwise is only supported in a choice block or after a when statement")

  def handle[E](block: => Unit)(implicit manifest: Manifest[E]) = SOnExceptionDefinition(target.onException(manifest.erasure)).apply(block)
  
  def idempotentconsumer(expression: Exchange => Any) = SIdempotentConsumerDefinition(target.idempotentConsumer(expression, null))
  
  def inOnly = wrap(target.inOnly)
  def inOut = wrap(target.inOut)

  def log(message: String) = wrap(target.log(message))
  def log(level: LoggingLevel, message: String) = wrap(target.log(level, message))
  def log(level: LoggingLevel, logName: String, message: String) = wrap(target.log(level, logName, message))

  def loop(expression: Exchange => Any) = SLoopDefinition(target.loop(expression))
  
  def marshal(format: DataFormatDefinition) = wrap(target.marshal(format))
  
  def multicast = SMulticastDefinition(target.multicast)
  
  def process(function: Exchange => Unit) = wrap(target.process(new ScalaProcessor(function)))
  def process(processor: Processor) = wrap(target.process(processor))
 
  def throttle(frequency: Frequency) = SThrottleDefinition(target.throttle(frequency.count).timePeriodMillis(frequency.period.milliseconds))
  
  def loadbalance = SLoadBalanceDefinition(target.loadBalance)
  
  def delay(period: Period) = SDelayDefinition(target.delay(period.milliseconds))
  
  def onCompletion : SOnCompletionDefinition = {
    var completion = SOnCompletionDefinition(target.onCompletion)
    // let's end the block in the Java DSL, we have a better way of handling blocks here
    completion.target.end
    completion
  }
  def onCompletion(predicate: Exchange => Boolean) = onCompletion().when(predicate).asInstanceOf[SOnCompletionDefinition]
  def onCompletion(config: Config[SOnCompletionDefinition]) = {
    val completion = onCompletion().asInstanceOf[SOnCompletionDefinition]
    config.configure(completion)
    completion
  }

  def pipeline = SPipelineDefinition(target.pipeline)
  
  def policy(policy: Policy) = wrap(target.policy(policy))

  def pollEnrich(uri: String, strategy: AggregationStrategy = null, timeout: Long = 0) = 
    wrap(target.pollEnrich(uri, timeout, strategy))

  def recipients(expression: Exchange => Any) = wrap(target.recipientList(expression))
  
  def resequence(expression: Exchange => Any) = SResequenceDefinition(target.resequence(expression))
  
  def rollback = wrap(target.rollback)

  def routingSlip(header: String) = wrap(target.routingSlip(header))
  def routingSlip(header: String, separator: String) = wrap(target.routingSlip(header, separator))
  def routingSlip(expression: Exchange => Any) = wrap(target.routingSlip(expression))

  def sort[T](expression: (Exchange) => Any, comparator: Comparator[T] = null) = wrap(target.sort(expression, comparator))

  def dynamicRouter(expression: Exchange => Any) = wrap(target.dynamicRouter(expression))

  def setbody(expression: Exchange => Any) = wrap(target.setBody(expression))

  def setfaultbody(expression: Exchange => Any) = wrap(target.setFaultBody(expression))
  
  def setheader(name: String, expression: Exchange => Any) = wrap(target.setHeader(name, expression))

  def stop = wrap(target.stop)

  def threads = SThreadsDefinition(target.threads)

  def throwException(exception: Exception) = wrap(target.throwException(exception))

  def transacted = wrap(target.transacted)
  def transacted(ref: String) = wrap(target.transacted(ref))

  def transform(expression: Exchange => Any) = wrap(target.transform(expression))
  
  def unmarshal(format: DataFormatDefinition) = wrap(target.unmarshal(format))

  def validate(expression: Exchange => Any) = wrap(target.validate(predicateBuilder(expression)))

  def wiretap(uri: String) = wrap(target.wireTap(uri))
  def wiretap(uri: String, expression: Exchange => Any) = wrap(target.wireTap(uri, expression))
  
  def aggregate(expression: Exchange => Any, strategy: AggregationStrategy) = SAggregateDefinition(target.aggregate(expression, strategy))

  /**
   * Helper method to return this Scala type instead of creating another wrapper type for the processor
   */
  def wrap(block : => Unit) : SAbstractDefinition[_]  = {
     block
     this
  }
}