<?php

/**
 * @file
 * Custom tokens for Metatag:hreflang.
 */

/**
 * Implements hook_token_info().
 */
function metatag_hreflang_token_info() {
  // This only makes sense if either the Translation or Entity Translation
  // modules are enabled.
  if (!(module_exists('translation') || module_exists('entity_translation'))) {
    return;
  }

  // Don't do anything if the patch was applied to Entity Translation to add
  // these.
  // @see https://www.drupal.org/node/2603056
  if (module_exists('entity_translation') && module_load_include('tokens.inc', 'entity_translation')) {
    return;
  }

  $info = array();

  $languages = language_list('enabled');

  // Verify there at least one language is available.
  if (!empty($languages[1]) && is_array($languages[1]) && count($languages[1])) {
    if (module_exists('node')) {
      foreach ($languages[1] as $langcode => $language) {
        $info['tokens']['node']['url-' . $langcode] = array(
          'name' => t('URL (@lang translation)', array('@lang' => $language->name)),
          'description' => t('The URL for the %lang translation of the node, if available.', array('%lang' => $language->name)),
        );
      }
    }
  }

  $info['tokens']['node']['url-original'] = array(
    'name' => t('URL (original language)'),
    'description' => t("The URL for the node that is the original source for the current node's translation."),
  );

  return $info;
}

/**
 * Implements hook_tokens().
 */
function metatag_hreflang_tokens($type, $tokens, array $data = array(), array $options = array()) {
  $replacements = array();

  $sanitize = !empty($options['sanitize']);

  // Node tokens.
  if ($type == 'node' && !empty($data['node'])) {
    // Shortcuts.
    $node = $data['node'];

    $languages = language_list('enabled');
    if (!empty($languages[1]) && is_array($languages[1]) && count($languages[1]) > 1) {
      foreach ($tokens as $name => $original) {
        // The original entity's URL.
        if ($name == 'url-original') {
          // Basic options regardless of the translation mechanism used.
          $url_options = $options;
          $url_options['absolute'] = TRUE;

          // Core's content translation.
          if (!empty($node->tnid)) {
            $node_original = node_load($node->tnid);
            // Only deal with published nodes.
            if ($node_original->status) {
              $url_options['language'] = $languages[1][$node_original->language];
              $replacements[$original] = url('node/' . $node_original->nid, $url_options);
            }
          }
          // Entity Translation stores the translation data differently to core.
          elseif (isset($node->translations, $node->translations->original, $languages[1][$node->translations->original])) {
            $url_options['language'] = $languages[1][$node->translations->original];
            $replacements[$original] = url('node/' . $node->nid, $url_options);
          }
        }

        // Separate URLs for each translation.
        // Core's content translation.
        if (!empty($node->tnid)) {
          $translations = translation_node_get_translations($node->tnid);
          foreach ($translations as $langcode => $translation) {
            // Only deal with published nodes.
            if ($translation->status && $name == 'url-' . $langcode) {
              $url_options = $options;
              $url_options['absolute'] = TRUE;
              $url_options['language'] = $languages[1][$langcode];
              $replacements[$original] = url('node/' . $translation->nid, $url_options);
            }
          }
        }
        // Entity Translation.
        elseif (isset($node->translations, $node->translations->data)) {
          foreach ($node->translations->data as $langcode => $translation) {
            // Only deal with published nodes.
            if ($translation['status'] && $name == 'url-' . $langcode) {
              $url_options = $options;
              $url_options['absolute'] = TRUE;
              $url_options['language'] = $languages[1][$langcode];
              $replacements[$original] = url('node/' . $node->nid, $url_options);
            }
          }
        }
      }
    }
  }

  return $replacements;
}