<?php

namespace CoffeeScript;

class yy_While extends yy_Base
{
  public $children = array('condition', 'guard', 'body');

  public $returns = FALSE;

  function constructor($condition = NULL, $options = NULL)
  {
    $this->condition = (isset($options['invert']) && $options['invert']) ? 
      $condition->invert() : $condition;

    $this->guard = isset($options['guard']) ? $options['guard'] : NULL;

    return $this;
  }

  function add_body($body)
  {
    $this->body = $body;
    return $this;
  }

  function compile_node($options)
  {
    $options['indent'] .= TAB;
    $set = '';
    $body = $this->body;

    if ($body->is_empty())
    {
      $body = '';
    }
    else
    {
      if ($this->returns)
      {
        $body->make_return($rvar = $options['scope']->free_variable('results'));
        $set = "{$this->tab}{$rvar} = [];\n";
      }

      if ($this->guard)
      {
        if ($body->expressions)
        {
          array_unshift($body->expressions, yy('If', yy('Parens', $this->guard)->invert(), yy('Literal', 'continue')));
        }
        else
        {
          $body = yy_Block::wrap(array(yy('If', $this->guard, $body)));
        }
      }

      $body = "\n".$body->compile($options, LEVEL_TOP)."\n{$this->tab}";
    }

    $code = $set.$this->tab.'while ('.$this->condition->compile($options, LEVEL_PAREN).") {{$body}}";

    if ($this->returns)
    {
      $code .= "\n{$this->tab}return {$rvar};";
    }

    return $code;
  }

  function is_statement()
  {
    return TRUE;
  }

  function jumps()
  {
    $expressions = isset($this->body->expressions) ? $this->body->expressions : array();

    if ( ! count($expressions))
    {
      return FALSE;
    }

    foreach ($expressions as $node)
    {
      if ($node->jumps(array('loop' => TRUE)))
      {
        return $node;
      }
    }

    return FALSE;
  }

  function make_return($res = NULL)
  {
    if ($res)
    {
      return parent::make_return($res);
    }
    else
    {
      $this->returns = ! $this->jumps(array('loop' => TRUE));
    }

    return $this;
  }
}

?>