phorge-phorge/src/applications/diffusion/controller/file/DiffusionBrowseFileController.php

<?php

/*
 * Copyright 2011 Facebook, Inc.
 *
 * Licensed 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.
 */

class DiffusionBrowseFileController extends DiffusionController {

  public function processRequest() {

    // Build the view selection form.
    $select_map = array(
      'highlighted' => 'View as Highlighted Text',
      'blame' => 'View as Highlighted Text with Blame',
      'plain' => 'View as Plain Text',
      'plainblame' => 'View as Plain Text with Blame',
    );

    $request = $this->getRequest();

    $selected = $request->getStr('view');
    $select = '<select name="view">';
    foreach ($select_map as $k => $v) {
      $option = phutil_render_tag(
        'option',
        array(
          'value' => $k,
          'selected' => ($k == $selected) ? 'selected' : null,
        ),
        phutil_escape_html($v));

      $select .= $option;
    }
    $select .= '</select>';

    $view_select_panel = new AphrontPanelView();
    $view_select_form = phutil_render_tag(
      'form',
      array(
        'action' => $request->getRequestURI(),
        'method' => 'get',
        'style'  => 'display: inline',
      ),
      $select.
      '<button>view</button>');
    $view_select_panel->appendChild($view_select_form);

    // Build the content of the file.
    $corpus = $this->buildCorpus($selected);

    // Render the page.
    $content = array();
    $content[] = $this->buildCrumbs(
      array(
        'branch' => true,
        'path'   => true,
        'view'   => 'browse',
      ));
    $content[] = $view_select_panel;
    $content[] = $corpus;

    $nav = $this->buildSideNav('browse', true);
    $nav->appendChild($content);

    return $this->buildStandardPageResponse(
      $nav,
      array(
        'title' => 'Browse',
      ));
  }


  public static function renderRevision(
    DiffusionRequest $drequest,
    $revision) {

    $callsign = $drequest->getCallsign();

    $name = 'r'.$callsign.$revision;
    return phutil_render_tag(
      'a',
      array(
        'href' => '/'.$name,
      ),
      $name
    );
  }


  private function buildCorpus($selected) {
    $blame = ($selected == 'blame' || $selected == 'plainblame');

    $file_query = DiffusionFileContentQuery::newFromDiffusionRequest(
      $this->diffusionRequest);
    $file_query->setNeedsBlame($blame);

    // TODO: image
    // TODO: blame of blame.
    switch ($selected) {
      case 'plain':
      case 'plainblame':
        $style =
          "margin: 1em 2em; width: 90%; height: 80em; font-family: monospace";
        $corpus = phutil_render_tag(
          'textarea',
          array(
            'style' => $style,
          ),
          phutil_escape_html($file_query->getRawData()));
          break;

      case 'highlighted':
      case 'blame':
      default:
        require_celerity_resource('syntax-highlighting-css');
        require_celerity_resource('diffusion-source-css');

        list($data, $blamedata, $revs) = $file_query->getTokenizedData();

        $drequest = $this->getDiffusionRequest();
        $path = $drequest->getPath();
        $highlightEngine = new PhutilDefaultSyntaxHighlighterEngine();
        $data = $highlightEngine->highlightSource($path, $data);

        $data = explode("\n", rtrim($data));

        $rows = $this->buildDisplayRows($data, $blame, $blamedata, $drequest,
          $revs);

        $corpus_table = phutil_render_tag(
          'table',
          array(
            'class' => "diffusion-source remarkup-code",
          ),
          implode("\n", $rows));
        $corpus = phutil_render_tag(
          'div',
          array(
            'style' => 'padding: 0pt 2em;',
          ),
          $corpus_table);

        break;
    }

    return $corpus;
  }


  private static function buildDisplayRows($data, $blame, $blamedata, $drequest,
    $revs) {
    $last = null;
    $color = null;
    $rows = array();
    $n = 1;
    foreach ($data as $k => $line) {
      if ($blame) {
        if ($last == $blamedata[$k][0]) {
          $blameinfo =
            '<th style="background: '.$color.'; width: 9em;"></th>'.
            '<th style="background: '.$color.'"></th>';
        } else {
          switch ($drequest->getRepository()->getVersionControlSystem()) {
            case PhabricatorRepositoryType::REPOSITORY_TYPE_GIT:
              // TODO: better color for git.
              $color = '#dddddd';
              break;
            case PhabricatorRepositoryType::REPOSITORY_TYPE_SVN:
              $color = sprintf(
               '#%02xee%02x',
               $revs[$blamedata[$k][0]],
               $revs[$blamedata[$k][0]]);
              break;
            default:
              throw new Exception('repository type not supported');
          }
          $revision_link = self::renderRevision(
           $drequest,
           $blamedata[$k][0]);

          $author_link = $blamedata[$k][1];
          $blameinfo =
            '<th style="background: '.$color.
              '; width: 9em;">'.$revision_link.'</th>'.
            '<th style="background: '.$color.
              '; font-weight: normal; color: #333;">'.$author_link.'</th>';
          $last = $blamedata[$k][0];
        }
      } else {
        $blameinfo = null;
      }

      if ($n == $drequest->getLine()) {
        $tr = '<tr style="background: #ffff00;">';
        $targ = '<a id="scroll_target"></a>';
        Javelin::initBehavior('diffusion-jump-to',
          array('target' => 'scroll_target'));
      } else {
        $tr = '<tr>';
        $targ = null;
      }

      $uri_path = $drequest->getUriPath();
      $uri_rev  = $drequest->getCommit();

      $l = phutil_render_tag(
        'a',
        array(
          'href' => $uri_path.';'.$uri_rev.'$'.$n,
        ),
        $n);

      $rows[] = $tr.$blameinfo.'<th>'.$l.'</th><td>'.$targ.$line.'</td></tr>';
      ++$n;
    }

    return $rows;
  }
}