<?php

/**
 * @group markup
 */
abstract class PhabricatorRemarkupRuleObjectHandle
  extends PhutilRemarkupRule {

  const KEY_RULE_HANDLE = 'rule.handle';

  abstract protected function getObjectNamePrefix();
  abstract protected function loadObjectPHID($id);

  public function apply($text) {
    $prefix = $this->getObjectNamePrefix();
    return $this->replaceHTML(
      "@\B{{$prefix}(\d+)}\B@",
      array($this, 'markupObjectHandle'),
      $text);
  }

  protected function markupObjectHandle($matches) {
    // TODO: These are single gets but should be okay for now, they're behind
    // the cache.
    $phid = $this->loadObjectPHID($matches[1]);
    if (!$phid) {
      return $matches[0];
    }

    $engine = $this->getEngine();
    $token = $engine->storeText('');

    $metadata_key = self::KEY_RULE_HANDLE;
    $metadata = $engine->getTextMetadata($metadata_key, array());
    if (empty($metadata[$phid])) {
      $metadata[$phid] = array();
    }
    $metadata[$phid][] = $token;
    $engine->setTextMetadata($metadata_key, $metadata);

    return $token;
  }

  public function didMarkupText() {
    $engine = $this->getEngine();

    $metadata_key = self::KEY_RULE_HANDLE;
    $metadata = $engine->getTextMetadata($metadata_key, array());
    if (empty($metadata)) {
      return;
    }

    $handles = id(new PhabricatorObjectHandleData(array_keys($metadata)))
      ->loadHandles();

    foreach ($metadata as $phid => $tokens) {
      $link = $handles[$phid]->renderLink();
      foreach ($tokens as $token) {
        $engine->overwriteStoredText($token, $link);
      }
    }

    $engine->setTextMetadata($metadata_key, array());
  }

}