Club Drupal

 *

 * The tokens we provide on generic entities require token module.

  // Add tokens for the flag count available at the entity level.

  // These require token module because we need its helper data and functions

  // to deal with token types that are not the same as the entity types they are

  // for (in particular, terms and vocabularies).

  if (module_exists('token')) {

    $entity_info = entity_get_info();

    foreach (flag_get_types() as $flag_type) {

      // The flag type is the entity type, but this is not necessarily the same

      // as the entity's token type.

      $token_type = $entity_info[$flag_type]['token type'];

      $flags = flag_get_flags($flag_type);

      foreach ($flags as $flag) {

        $tokens[$token_type]['flag-' . str_replace('_', '-', $flag->name) . '-count'] = array(

          'name' => t('@flag flag count', array('@flag' => $flag->get_title())),

          'description' => t('Total flag count for flag @flag', array('@flag' => $flag->get_title())),

          'flag-type' => $flag_type,

        );

        $tokens[$token_type]['flag-' . str_replace('_', '-', $flag->name) . '-link'] = array(

          'name' => t('@flag flag link', array('@flag' => $flag->get_title())),

          'description' => t('Flag/unflag link for @flag', array('@flag' => $flag->get_title())),

          'flag-type' => $flag_type,

        );

      }

  // We only provide tokens on entity types if we have token module's helper

  // methods available.

  if (isset($data[$type]) && module_exists('token')) {

    $entity_type = token_get_entity_mapping('token', $type);

    if ($entity_type && in_array($entity_type, flag_get_types())) {

      $flags = flag_get_flags($entity_type);

      $object = $data[$type];

      foreach ($flags as $flag) {

        foreach ($tokens as $name => $original) {

          $flag_count_token = 'flag-' . str_replace('_', '-', $flag->name) . '-count';

          $flag_link_token = 'flag-' . str_replace('_', '-', $flag->name) . '-link';

          if ($name == $flag_count_token) {

            $replacements[$original] = $flag->get_count($flag->get_entity_id($object));

          }

          elseif ($name == $flag_link_token) {

            $replacements[$original] = flag_create_link($flag->name, $flag->get_entity_id($object));

          }