autoassignrole.module in Auto Assign Role 6

The main autoassignrole.module file

Designate a role to assign all new users to in addition to providing a mechanism for new users to select a role or role for their account.

File

autoassignrole.module

<?php

/**
 * @file
 *
 * The main autoassignrole.module file
 *
 * Designate a role to assign all new users to in addition to providing a
 * mechanism for new users to select a role or role for their account.
 */

/**
 * Implementation of hook_menu().
 *
 * @return array
 */
function autoassignrole_menu() {
  $items = array();
  $items['admin/user/autoassignrole/autocomplete/node'] = array(
    'title' => 'Node autocomplete',
    'page callback' => 'autoassignrole_autocomplete_node',
    'access callback' => 'user_access',
    'access arguments' => array(
      'administer autoassignrole',
    ),
    'type' => MENU_CALLBACK,
    'file' => 'autoassignrole-admin.inc',
  );
  $items['admin/user/autoassignrole'] = array(
    'title' => t('Auto assign role'),
    'description' => t('Designate a role to assign all new users to.'),
    'page callback' => 'drupal_get_form',
    'page arguments' => array(
      'autoassignrole_admin_form',
    ),
    'access arguments' => array(
      'administer autoassignrole',
    ),
    'type' => MENU_NORMAL_ITEM,
    'file' => 'autoassignrole-admin.inc',
  );

  // path based role assignments that are listed as a menu item
  $result = db_query("SELECT rid, path, display, title, weight, menu FROM {autoassignrole_page}");
  while ($r = db_fetch_object($result)) {
    switch ($r->display) {
      case 0:
        $items[$r->path] = array(
          'title' => check_plain($r->title),
          'page arguments' => array(
            $r->rid,
          ),
          'page callback' => 'autoassignrole_path',
          'access callback' => '_autoassignrole_path_access',
          'file' => 'autoassignrole-path.inc',
          'weight' => $r->weight,
          'type' => MENU_NORMAL_ITEM,
          'menu_name' => $r->menu,
        );
        $items[$r->path . '/register'] = array(
          'title' => 'Create new account',
          'page arguments' => array(
            $r->rid,
          ),
          'page callback' => 'autoassignrole_path',
          'access callback' => '_autoassignrole_path_access',
          'file' => 'autoassignrole-path.inc',
          'type' => MENU_DEFAULT_LOCAL_TASK,
        );
        $items[$r->path . '/login'] = array(
          'title' => 'Log in',
          'access callback' => 'user_is_anonymous',
          'type' => MENU_LOCAL_TASK,
          'page callback' => '_autoassignrole_user_login',
          'file' => 'autoassignrole-path.inc',
        );
        $items[$r->path . '/password'] = array(
          'title' => 'Request new password',
          'access callback' => 'user_is_anonymous',
          'type' => MENU_LOCAL_TASK,
          'page callback' => '_autoassignrole_user_password',
          'file' => 'autoassignrole-path.inc',
        );
        break;

      // tabs on user registration pages
      case 1:
        $items['user/' . $r->path] = array(
          'title' => check_plain($r->title),
          'page arguments' => array(
            $r->rid,
          ),
          'page callback' => 'autoassignrole_path',
          'access callback' => '_autoassignrole_path_access',
          'file' => 'autoassignrole-path.inc',
          'type' => MENU_LOCAL_TASK,
          'weight' => $r->weight,
        );
        break;

      // no menu items
      case 2:
        $items[$r->path] = array(
          'title' => check_plain($r->title),
          'page arguments' => array(
            $r->rid,
          ),
          'page callback' => 'autoassignrole_path',
          'access callback' => '_autoassignrole_path_access',
          'file' => 'autoassignrole-path.inc',
          'weight' => $r->weight,
          'type' => MENU_CALLBACK,
        );
        $items[$r->path . '/register'] = array(
          'title' => 'Create new account',
          'page arguments' => array(
            $r->rid,
          ),
          'page callback' => 'autoassignrole_path',
          'access callback' => '_autoassignrole_path_access',
          'file' => 'autoassignrole-path.inc',
          'type' => MENU_DEFAULT_LOCAL_TASK,
        );
        $items[$r->path . '/login'] = array(
          'title' => 'Log in',
          'access callback' => 'user_is_anonymous',
          'type' => MENU_LOCAL_TASK,
          'page callback' => '_autoassignrole_user_login',
          'file' => 'autoassignrole-path.inc',
        );
        $items[$r->path . '/password'] = array(
          'title' => 'Request new password',
          'access callback' => 'user_is_anonymous',
          'type' => MENU_LOCAL_TASK,
          'page callback' => '_autoassignrole_user_password',
          'file' => 'autoassignrole-path.inc',
        );
        break;
    }
  }
  return $items;
}
function _autoassignrole_path_access() {
  global $user;
  if (variable_get('user_register', 1) && $user->uid == 0) {
    return TRUE;
  }
  if (arg(0) == 'admin' && arg(2) == 'menu-customize' && user_access('administer menu', $user)) {
    return TRUE;
  }
  return FALSE;
}

/**
 * Implementation of hook_perm().
 * @return array
 */
function autoassignrole_perm() {
  return array(
    'administer autoassignrole',
  );
}

/**
 * Implementation of hook_user().
 */
function autoassignrole_user($op, &$edit, &$account, $category = NULL) {
  switch ($op) {
    case 'insert':

      // If this is an administrator creating the account only use auto_assign if
      // allowed by auto_admin_active
      if (arg(0) == 'admin' && _autoassignrole_get_settings('auto_admin_active') == 0) {
        return;
      }
      $rolenames = user_roles();
      if (_autoassignrole_get_settings('user_active') == 1) {
        $user_roles = _autoassignrole_get_settings('user_roles');
        if (is_array($edit['user_roles'])) {
          foreach ($edit['user_roles'] as $k => $v) {
            if ($v != 0 && in_array($k, $user_roles, TRUE)) {
              $edit['roles'][$k] = $v;
            }
          }
        }
        elseif ($edit['user_roles'] != '') {
          $edit['roles'][$edit['user_roles']] = $edit['user_roles'];
        }
      }
      if ($rid = autoassignrole_get_active_path_rid()) {
        $edit['roles'][$rid] = $rolenames[$rid];
      }
      if (_autoassignrole_get_settings('auto_active') == 1) {
        $auto_roles = _autoassignrole_get_settings('auto_roles');
        foreach ($auto_roles as $k => $v) {
          $edit['roles'][$k] = $rolenames[$k];
        }
      }
      break;
    case 'register':
      if (_autoassignrole_get_settings('user_active') == 1) {
        $roles = user_roles(TRUE);
        $user_roles = _autoassignrole_get_settings('user_roles');
        $path_roles = db_query("SELECT rid from {autoassignrole_page}");
        while ($path_role = db_fetch_object($path_roles)) {
          unset($roles[$path_role->rid]);
        }
        foreach ($roles as $k => $r) {
          if (!in_array($k, $user_roles, TRUE)) {
            unset($roles[$k]);
          }
        }
        if (count($roles)) {
          $form['autoassignrole_user'] = array(
            '#type' => 'fieldset',
            '#title' => _autoassignrole_get_settings('user_fieldset_title'),
            '#collapsible' => FALSE,
            '#collapsed' => FALSE,
          );
          if (_autoassignrole_get_settings('user_sort') == "SORT_ASC") {
            uasort($roles, '_autoassignrole_array_asc');
          }
          else {
            uasort($roles, '_autoassignrole_array_desc');
          }
          if (!$edit || !array_key_exists('user_roles', $edit)) {
            $edit['user_roles'] = array();
          }
          if (_autoassignrole_get_settings('user_selection') == 0) {
            $form['autoassignrole_user']['user_roles'] = array(
              '#type' => 'radios',
              '#title' => t(_autoassignrole_get_settings('user_title')),
              '#options' => $roles,
              '#description' => t(_autoassignrole_get_settings('user_description')),
            );
          }
          if (_autoassignrole_get_settings('user_selection') == 1) {
            $form['autoassignrole_user']['user_roles'] = array(
              '#type' => 'select',
              '#title' => t(_autoassignrole_get_settings('user_title')),
              '#default_value' => '',
              '#description' => t(_autoassignrole_get_settings('user_description')),
            );
            if (_autoassignrole_get_settings('user_multiple')) {
              $form['autoassignrole_user']['user_roles']['#multiple'] = TRUE;
              unset($form['autoassignrole_user']['user_roles']['#default_value']);
            }
            else {
              $roles[''] = '';
            }
            $form['autoassignrole_user']['user_roles']['#options'] = $roles;
          }
          if (_autoassignrole_get_settings('user_selection') == 2) {
            $form['autoassignrole_user']['user_roles'] = array(
              '#type' => 'checkboxes',
              '#title' => t(_autoassignrole_get_settings('user_title')),
              '#default_value' => $edit['user_roles'],
              '#options' => $roles,
              '#description' => _autoassignrole_get_settings('user_description'),
            );
          }
          if (_autoassignrole_get_settings('user_required')) {
            $form['autoassignrole_user']['user_roles']['#required'] = TRUE;
          }
          return $form;
        }
      }
      break;
  }
}

/**
 * API function that returns an array of AAR's settings.
 *
 */
function _autoassignrole_get_settings($value) {

  // if we are dealing with a variable in the format of path_field_rid get
  // variables from {autoassignrole_page}
  if (preg_match('/^path_(\\D[^_]*)_(\\d*)/i', $value, $matches)) {
    $result = db_fetch_array(db_query("SELECT rid, display, path, title, description, format, weight FROM {autoassignrole_page} WHERE rid = %d", $matches[2]));
    if (count($result) > 1 && $matches[1] == 'active') {
      return 1;
    }
    elseif (count($result) == 1 && $matches[1] == 'active') {
      return 0;
    }
    else {
      if (isset($result[$matches[1]])) {
        return $result[$matches[1]];
      }
      else {
        return FALSE;
      }
    }
  }
  if (preg_match('/^path_(\\d*)/i', $value, $matches)) {
    $result = db_fetch_array(db_query("SELECT path FROM {autoassignrole_page} WHERE rid = %d", $matches[1]));
    if (isset($result['path'])) {
      return $result['path'];
    }
    else {
      return FALSE;
    }
  }

  // if we are not dealing with {autoassignrole_page} then get variables from
  // {autoassignrole}
  $settings = array();
  $result = db_query("SELECT arid, value from {autoassignrole}");
  while ($s = db_fetch_object($result)) {
    $settings[$s->arid] = $s->value;
  }

  // break apart auto_roles[x] and user_roles[x]
  if (preg_match('/^(\\D[^_]*_\\D[^_]*)\\[(\\d*)]/i', $value, $matches)) {
    $values = unserialize($settings[$matches[1]]);
    return $values[$matches[2]];
  }
  switch ($value) {
    case 'auto_roles':
    case 'user_roles':

      // return all instances as an array
      $roles = $settings[$value];
      $roles = unserialize($roles);
      if (!is_array($roles)) {
        $roles = array();
      }
      foreach ($roles as $k => $r) {
        if ($r == 0) {
          unset($roles[$k]);
        }
      }
      return $roles;
      break;
    default:
      if (isset($settings[$value])) {
        return $settings[$value];
      }
      else {
        return FALSE;
      }
      break;
  }
}
function _autoassignrole_path($rid, $value) {
  static $pages = array();
  if (count($pages) == 0) {
    $result = db_query("SELECT rid, display, path, title, description from {autoassignrole_page}");
    while ($s = db_fetch_object($result)) {
      $pages[$s->rid] = array(
        'path' => $s->path,
        'display' => $s->display,
        'title' => $s->title,
        'description' => $s->description,
      );
    }
    return _autoassignrole_get_settings($rid, $value);
  }
  else {
    if (isset($pages[$rid][$value])) {
      $return = $pages[$rid][$value];
    }
    else {
      $return = '';
    }
    return $return;
  }
}
function _autoassignrole_user_input($args) {
  switch ($args) {
    case 'type':
      if (_autoassignrole_get_settings('user_multiple') == 0) {
        $type = 'radios';
      }
      else {
        $type = 'checkboxes';
      }
      return $type;
      break;
    case 'required':
      if (_autoassignrole_get_settings('user_required') == 0) {
        $required = FALSE;
      }
      else {
        $required = TRUE;
      }
      return $required;
      break;
  }
}
function _autoassignrole_array_intersect_key($isec, $keys) {
  $argc = func_num_args();
  if ($argc > 2) {
    for ($i = 1; !empty($isec) && $i < $argc; $i++) {
      $arr = func_get_arg($i);
      foreach (array_keys($isec) as $key) {
        if (!isset($arr[$key])) {
          unset($isec[$key]);
        }
      }
    }
    return $isec;
  }
  else {
    $res = array();
    foreach (array_keys($isec) as $key) {
      if (isset($keys[$key])) {
        $res[$key] = t($isec[$key]);
      }
    }
    return $res;
  }
}

/**
 * method to sort array in a descending fashion while preserving keys
 * @param string $a a string to compare
 * @param string $b a string to compare
 * @return int
 */
function _autoassignrole_array_desc($a, $b) {
  if ($a == $b) {
    return 0;
  }
  return $a > $b ? -1 : 1;
}

/**
 * method to sort array in an ascending fashion while preserving keys
 * @param string $a a string to compare
 * @param string $b a string to compare
 * @return int
 */
function _autoassignrole_array_asc($a, $b) {
  if ($a == $b) {
    return 0;
  }
  return $a < $b ? -1 : 1;
}

/**
 * Implementation of hook_form_alter().
 *
 * Integrate with content profile's registration integration
 */
function autoassignrole_form_alter(&$form, $form_state, $form_id) {
  if ($form_id == 'content_profile_admin_settings') {
    $type = $form_state['type'];
    $result = db_query("SELECT ar.path, ar.rid, r.name FROM {autoassignrole_page} ar INNER JOIN {role} r ON ar.rid = r.rid");
    $options = array();
    while ($r = db_fetch_object($result)) {
      $options[$r->rid] = filter_xss_admin("({$r->name}) {$r->path}");
    }
    $form['registration']['autoassignrole_use'] = array(
      '#type' => 'checkboxes',
      '#title' => t('Use on Auto Assign Role paths'),
      '#default_value' => content_profile_get_settings($type, 'autoassignrole_use'),
      '#options' => $options,
      '#description' => t('The Auto Assign Role module gives you the ability to assign paths a user can register from for a role.  After associating a <a href="@aar">path with a role</a> your selection can  associate this content type with a path.', array(
        '@aar' => url('admin/user/autoassignrole'),
      )),
      '#disabled' => count($options) == 0,
      '#weight' => 3,
    );
    array_unshift($form['#submit'], 'autoassignrole_content_profile_admin_form_submit');
  }
  elseif ($form_id == 'user_register' && module_exists('content_profile_registration')) {
    if (implode('/', arg()) != 'user/register' && ($rid = autoassignrole_get_active_path_rid())) {

      // Set the content types which should be shown by the content_profile_registration module
      $form['#content_profile_registration_use_types'] = array();
      foreach (content_profile_get_types('names') as $type => $name) {
        $cp_settings = content_profile_get_settings($type, 'autoassignrole_use');
        if (is_array($cp_settings) && in_array($rid, $cp_settings)) {
          $form['#content_profile_registration_use_types'][$type] = $name;
        }
      }
    }
  }
}
function autoassignrole_content_profile_admin_form_submit($form, &$form_state) {
  $form_state['values']['autoassignrole_use'] = array_keys(array_filter($form_state['values']['autoassignrole_use']));
}

/**
 * Implementation of hook_content_profile_settings().
 */
function autoassignrole_content_profile_settings() {
  static $autoassignrole;
  if (empty($autoassignrole)) {
    $autoassignrole['autoassignrole_use'] = array_keys(user_roles(TRUE));
  }
  return $autoassignrole;
}

/**
 * Returns the role id of the currently active AAR-path.
 *
 * @return
 *  array An array of roles
 */
function autoassignrole_get_active_path_rid() {
  $item = menu_get_item();
  if ($item['page_callback'] == 'autoassignrole_path') {
    return $item['page_arguments'][0];
  }
  return FALSE;
}

/**
 * Get all assignable roles that AAR can possibly apply.
 *
 * @return
 *  array An array of roles
 */
function autoassignrole_get_all_roles() {
  $aar_roles = array();

  // Select all auto and user selectable roles
  $sql = "SELECT value FROM {autoassignrole} WHERE arid = 'auto_roles' OR arid = 'user_roles'";
  $result = db_fetch_object(db_query($sql));
  $aar_roles = unserialize($result->value);

  // unset anything that isn't active
  foreach ($aar_roles as $key => $value) {
    if ($value == 0) {
      unset($aar_roles[$key]);
    }
  }

  // Select all path assignable roles
  $sql = "SELECT rid FROM {autoassignrole_page}";
  $result = db_query($sql);
  while ($row = db_fetch_object($result)) {
    $aar_roles[$row->rid] = $row->rid;
  }
  return _autoassignrole_clean_roles($aar_roles);
}

/**
 * Get all assignable roles that AAR will apply automatically.
 *
 * @return
 *  array An array of roles
 */
function autoassignrole_get_auto_roles() {
  $aar_roles = array();

  // Select all auto roles
  $sql = "SELECT value FROM {autoassignrole} WHERE arid = 'auto_roles'";
  $result = db_fetch_object(db_query($sql));
  $aar_roles = unserialize($result->value);

  // unset anything that isn't active
  foreach ($aar_roles as $key => $value) {
    if ($value == 0) {
      unset($aar_roles[$key]);
    }
  }
  return _autoassignrole_clean_roles($aar_roles);
}

/**
 * Get all user selectable roles
 *
 * @return
 *  array An array of roles
 */
function autoassignrole_get_user_selectable_roles() {
  $aar_roles = array();

  // Select all user selectable roles
  $sql = "SELECT value FROM {autoassignrole} WHERE arid = 'user_roles'";
  $result = db_fetch_object(db_query($sql));
  $aar_roles = unserialize($result->value);
  return _autoassignrole_clean_roles($aar_roles);
}

/**
 * Get all path assignable roles
 *
 * @param $path
 *  array Optional array of paths to restrict selection to when using an $op of path
 * @return
 *  array An array of roles
 */
function autoassignrole_get_path_roles($path = array()) {
  $aar_roles = array();

  // Select path based roles
  if (count($path) == 0) {
    $sql = "SELECT rid FROM {autoassignrole_page}";
    $result = db_query($sql);
  }
  else {
    foreach ($path as $key => $value) {
      $path[$key] = "'{$value}'";
    }
    $sql = "SELECT rid FROM {autoassignrole_page} WHERE path IN(" . implode(',', $path) . ")";
    $result = db_query($sql);
  }
  while ($row = db_fetch_object($result)) {
    $aar_roles[$row->rid] = $row->rid;
  }
  return _autoassignrole_clean_roles($aar_roles);
}
function _autoassignrole_clean_roles($aar_roles) {
  $roles = user_roles();

  // Use the results of user_roles() and unset anything not available from AAR
  foreach ($roles as $key => $role) {
    if (!array_key_exists($key, $aar_roles)) {
      unset($roles[$key]);
    }
  }
  return $roles;
}

/**
 * Get all roles that will be assigned based on the active path
 *
 * @param $path
 *  string An optional path to use as the active path
 * @return
 *  array An array of roles
 */
function autoassignrole_get_active_path_roles($path = NULL) {
  if (!is_array($path)) {
    $path = array(
      $path,
    );
  }
  if (count($path) == 0) {
    $menu_item = menu_get_item();
    $path[] = $menu_item['path'];
  }
  $path_roles = autoassignrole_get_path_roles($path);
  $auto_roles = autoassignrole_get_auto_roles();
  foreach ($auto_roles as $key => $value) {
    $path_roles[$key] = $value;
  }
  return _autoassignrole_clean_roles($path_roles);
}

/**
 * An API like call to return the roles a user has available or will be assigned
 *
 * @param $op
 *  string active, all, auto, path, user
 * @param $path
 *  array Optional array of paths to restrict selection to when using an $op of path
 * @return
 *  array An array of roles
 */
function autoassignrole_get_roles($op = 'all', $path = array()) {
  switch ($op) {
    case 'active':
      return autoassignrole_get_active_path_roles($path);
      break;
    case 'all':
      return autoassignrole_get_all_roles();
      break;
    case 'auto':
      return autoassignrole_get_auto_roles();
      break;
    case 'path':
      return autoassignrole_get_path_roles($path);
      break;
    case 'user':
      return autoassignrole_get_user_selectable_roles();
      break;
  }
}

/**
 * Implementation of hook_menu_alter()
 *
 * Redirect the default user register page to a path that has been designated as
 * the replacement registration path.
 */
function autoassignrole_menu_alter(&$items) {
  $row = db_fetch_object(db_query("SELECT rid, path, display, title, weight, menu, registration FROM {autoassignrole_page} WHERE registration = 1"));
  if (isset($row->registration)) {
    $path = $row->path;
  }
  else {
    $row = db_fetch_object(db_query("SELECT arid, value FROM {autoassignrole} WHERE arid = '%s'", 'node_user_register'));
    if (!empty($row->value)) {
      $path = 'node/' . $row->value;
    }
  }
  if (isset($path)) {
    $items['user/register']['page arguments'] = array(
      $path,
    );
    $items['user/register']['page callback'] = 'drupal_goto';
    $items['user/register']['type'] = MENU_LOCAL_TASK;
  }
}

Functions

Name	Description
autoassignrole_content_profile_admin_form_submit
autoassignrole_content_profile_settings	Implementation of hook_content_profile_settings().
autoassignrole_form_alter	Implementation of hook_form_alter().
autoassignrole_get_active_path_rid	Returns the role id of the currently active AAR-path.
autoassignrole_get_active_path_roles	Get all roles that will be assigned based on the active path
autoassignrole_get_all_roles	Get all assignable roles that AAR can possibly apply.
autoassignrole_get_auto_roles	Get all assignable roles that AAR will apply automatically.
autoassignrole_get_path_roles	Get all path assignable roles
autoassignrole_get_roles	An API like call to return the roles a user has available or will be assigned
autoassignrole_get_user_selectable_roles	Get all user selectable roles
autoassignrole_menu	Implementation of hook_menu().
autoassignrole_menu_alter	Implementation of hook_menu_alter()
autoassignrole_perm	Implementation of hook_perm().
autoassignrole_user	Implementation of hook_user().
_autoassignrole_array_asc	method to sort array in an ascending fashion while preserving keys
_autoassignrole_array_desc	method to sort array in a descending fashion while preserving keys
_autoassignrole_array_intersect_key
_autoassignrole_clean_roles
_autoassignrole_get_settings	API function that returns an array of AAR's settings.
_autoassignrole_path
_autoassignrole_path_access
_autoassignrole_user_input