<?php
namespace taskman;
use Exception;
use stdClass;

task('bhl_clean', function()
{
  bhl_clean();
});

task('bhl_clean_cache', function()
{
  bhl_clean_cache();
});

task('bhl_make_upm_package', function() {
  bhl_make_upm_package();
});

function bhl_proj_file()
{
  return get('BHL_PROJ_FILE');
}

function bhl_proj(?string $proj_file = null)
{
  global $GAME_ROOT;

  static $projs = [];

  $proj_file ??= bhl_proj_file();

  if(!isset($projs[$proj_file]))
  {
    $proj = json_decode(ensure_read($proj_file));
    if(!$proj)
      throw new Exception("Bad bhl project file: $proj_file");

    $proj->file_path = $proj_file; //NOTE: adding path to the file for convenience

    foreach($proj->src_dirs as $k => $v)
      $proj->src_dirs[$k] = _bhl_make_abs_path($proj_file, $v);

    foreach($proj->bindings_sources as $k => $v)
      $proj->bindings_sources[$k] = _bhl_make_abs_path($proj_file, $v);
    $proj->bindings_dll= _bhl_make_abs_path($proj_file, $proj->bindings_dll);

    if(isset($proj->postproc_sources))
    {
      foreach($proj->postproc_sources as $k => $v)
        $proj->postproc_sources[$k] = _bhl_make_abs_path($proj_file, $v);
    }

    if(isset($proj->postproc_dll))
      $proj->postproc_dll = _bhl_make_abs_path($proj_file, $proj->postproc_dll);

    $proj->result_file = _bhl_make_abs_path($proj_file, $proj->result_file);
    $proj->error_file = _bhl_make_abs_path($proj_file, $proj->error_file);
    $proj->tmp_dir = _bhl_make_abs_path($proj_file, $proj->tmp_dir);

    $projs[$proj_file] = $proj;
  }
  return $projs[$proj_file];
}

function _bhl_make_abs_path($proj_file, $path)
{
  if($path && $path[0] == '.')
    return dirname($proj_file) . '/' . $path;
  else
    return $path;
}

function bhl_result_file(stdClass $bhl_proj = null)
{
  $bhl_proj ??= bhl_proj();
  return $bhl_proj->result_file;
}

function bhl_dir()
{
  global $GAME_ROOT;
  return "$GAME_ROOT/composer/vendor/bit/bhl";
}

function bhl_shell($cmd, &$ret_var, &$ret_out)
{
  $prev_path = mono_try_override_path();

  try
  {
    shell_try(bhl_dir()."/bhl $cmd", $ret_var, $ret_out);
  }
  finally
  {
    mono_try_restore_path($prev_path);
  }
}

function bhl_shell_ensure($cmd)
{
  bhl_shell($cmd, $ret_var, $ret_out);
  if($ret_var !== 0)
    throw new Exception("Error executing shell cmd: $ret_var");
}

function bhl_scan_files(stdClass $bhl_proj = null)
{
  $bhl_proj ??= bhl_proj();
  return scan_files_rec($bhl_proj->src_dirs, array('.bhl'));
}

function bhl_run($debug = true, $force = false, $exit_on_err = true, stdClass $bhl_proj = null)
{
  global $GAME_ROOT;

  $bhl_proj ??= bhl_proj();
  $result_file = $bhl_proj->result_file;

  if($force || need_to_regen($result_file, bhl_scan_files()))
  {
    bhl_shell("compile -p " . $bhl_proj->file_path . " " .
      ($debug ? " -d" : "") . " " .
      ($force || !getor("BHL_USE_CACHE", true) ? " -C" : ""),
      $ret_var, $ret_out
    );

    if($ret_var != 0)
    {
      bhl_handle_error_result($ret_out, $bhl_proj->error_file, $exit_on_err);
      if(!$exit_on_err)
        return false;
    }
    else
      echo "BHL BUNDLE: total " . kb_len(filesize($result_file)) . "\n"; 
  }

  @touch($result_file);
}

function bhl_handle_error_result(array $ret_out, $err_file, $exit = true)
{
  if(!is_file($err_file))
  {
    stderr("Something went wrong: " . implode("\n", $ret_out));
    if($exit)
      exit(1);
  }

  $history = array();
  $err_lines = file($err_file);
  $err_count = 0;
  foreach($err_lines as $err)
  {
    $jerr = json_decode($err);
    if(!$jerr)
    {
      stderr("Something went wrong, bhl error is in invalid format: $err\n");
      continue;
    }

    $file = $jerr->file;
    //let's show only one error per file's line
    if(isset($history[$file.$jerr->line]))
      continue;
    $history[$file.$jerr->line] = true;
    ++$err_count;
    stderr("BHL ERROR (#$err_count): " . $jerr->error . " \nin '$file', line {$jerr->line}\n");
    if(file_exists($file))
      stderr(bhl_show_position($jerr->line, $jerr->column, file($file)) . "\n");
  }
  if($err_lines && $exit)
    exit(1);
}

function bhl_clean(stdClass $bhl_proj = null)
{
  bhl_clean_cache($bhl_proj);
  bhl_shell_ensure("clean");
}

function bhl_clean_cache(stdClass $bhl_proj = null)
{
  $bhl_proj ??= bhl_proj();
  ensure_rm($bhl_proj->tmp_dir);
}

function bhl_show_position($line, $row, array $lines)
{
  if($line > 0 && $line <= count($lines))
  {
    //handling tabs
    $hint = str_replace("\t", "    ", $lines[$line-1]);
    for($c=0;$c<$row;++$c)
    {
      if($lines[$line-1][$c] === "\t")
        $hint .= "----";
      else
        $hint .= "-";
    }
    $hint .= "^";
    return $hint;
  }
  else
    return "??? @($line:$row)";
}

function bhl_line_row_to_pos($file, $line, $row)
{
  $pos = 0;
  $lines = file($file);
  //something went wrong
  if($line <= 0 || $line > count($lines))
    return null;
  for($i=0;$i<($line-1);++$i)
    $pos += strlen($lines[$i]);
  $pos += $row;
  return $pos;
}

function bhl_map_module_to_file($module, stdClass $bhl_proj = null)
{
  $bhl_proj ??= bhl_proj();
  foreach($bhl_proj->src_dirs as $dir)
  {
    $tmp = $dir.'/'.$module.'.bhl';
    if(file_exists($tmp))
      return $tmp;
  }
  return null;
}

function bhl_validate_func_ref($module_file, $func, array $signature)
{
  if(sizeof($signature) == 0)
    throw new Exception("Signature is invalid");

  $module_src = file_get_contents($module_file);

  if(!$module_src)
    throw new Exception("Bad module file '{$module_file}'");

  $signature_pattern = '';
  $signature_pattern .= '~func\s+';
  if($signature[0] !== 'void')
  {
    $signature_pattern .= preg_quote($signature[0]);
    $signature_pattern .= '\s+';
  }
  $signature_pattern .= $func;
  $signature_pattern .= '\s*\(';
  array_shift($signature);
  foreach($signature as $type)
  {
    $signature_pattern .= '\s*';
    $signature_pattern .= preg_quote($type);
    $signature_pattern .= '\s*\S+';
    $signature_pattern .= '\s*,';
  }
  $signature_pattern = rtrim($signature_pattern, '\s*,');
  $signature_pattern .= '\s*\)';
  $signature_pattern .= '~';

  if(!preg_match($signature_pattern, $module_src))
    throw new Exception("Func '$func' signature '".implode(',', $signature)."' not found in module '$module_file'");
}

function bhl_upm_path()
{
  global $GAME_ROOT;
  $pkg_dir = get("UNITY_ASSETS_DIR") . '/../Packages/bhl';
  return $pkg_dir;
}

function bhl_clean_upm_package()
{
  ensure_rm(bhl_upm_path() . '/Runtime/code/');
}

function bhl_make_upm_package()
{
  global $GAME_ROOT;

  $pkg_dir = bhl_upm_path();
  ensure_mkdir($pkg_dir);

  $srcs = array();
  $lines = file(bhl_dir() . '/bhl.cs');
  $started = false;
  foreach($lines as $line)
  {
    if(!$started && strpos($line, 'static readonly string[] VM_SRC = new string[] {') !== false)
    {
      $started = true;
      continue;
    }
    else if($started && strpos($line, '};') !== false) 
      break;
    else if($started)
    {
      $mask = trim($line);
      $mask = str_replace('$"{BHL_ROOT}/', '', $mask);
      $mask = str_replace('",', '', $mask);
      $srcs = array_merge($srcs, glob(bhl_dir() . '/' . $mask));
    }
  }

  $trgs = array();
  $changed = false;
  foreach($srcs as $src)
  {
    $trg = $pkg_dir . '/Runtime/code/' . str_replace(bhl_dir(), '', $src);
    if(!is_file($trg) || file_get_contents($src) != file_get_contents($trg))
      $changed = true;
    $trgs[$src] = $trg;
  }

  if($changed)
  {
    bhl_clean_upm_package();
    foreach($trgs as $src => $trg)
      ensure_copy($src, $trg);
  }
}