<?php
// $Id: subdomain.module,v 1.4.2.2 2008-06-13 04:07:43 setvik Exp $

// token.module token prefix
define('SUBDOMAIN_TOKEN_PREFIX', '~');
define('SUBDOMAIN_TOKEN_PREFIX_LENGTH', 1);

// Subdomain rewrite type (OG group or Node author)
define('SUBDOMAIN_GROUP', 1);
define('SUBDOMAIN_USER', 2);
define('SUBDOMAIN_VOCAB', 3);

define('SUBDOMAIN_STYLE_NAME', 1);
define('SUBDOMAIN_STYLE_CUSTOM', 2);

// Used internally for custom subdomain rewrites (i.e. not used by token.module)
define('SUBDOMAIN_CUSTOM_TOKEN_NAME', '%name%');
define('SUBDOMAIN_CUSTOM_TOKEN_ID', '%id');

/**
 * Implementation of hook_menu()
 */
function subdomain_menu() {
  $items['admin/build/path/subdomain'] = array(
    'title' => t('Subdomain settings'),
    'description' => t('Configure automated subdomain path alias settings.'),
    'page callback' => 'drupal_get_form',
    'page arguments' => array('subdomain_admin_settings'),
    'access arguments' => array('administer site configuration'),
    'type' => MENU_LOCAL_TASK,
    'weight' => 11
  );
  return $items;
}

/**
 * Implementation of hook_nodeapi()
 */
function subdomain_nodeapi(&$node, $op, $teaser, $page) {
  switch ($op) {
    case 'validate':
      $type = node_get_types('type',$node);
      $type_name = strtolower($type->name);
      $type_title = strtolower($type->title_label);
      if (module_exists('og') && og_is_group_type($node->type)) {
        $node->title = trim($node->title);

        // Allow only letters, numbers, and spaces
        if (!preg_match('!^[a-zA-Z0-9 ]+$!', $node->title)) {
          form_set_error('title', "The $type_name's $type_title can contain only letters, numbers and spaces");
        }

        // Disallow reserved subdomain
        elseif (subdomain_reserved($node->title)) {
          form_set_error('title', "Sorry. '$node->title' is not available for use. Please enter a different $type_title for your $type_name.");
        }

        // Ensure group name is unique
        elseif (db_result(db_query("SELECT 1 FROM {node} WHERE nid <> %d AND type = '%s' AND title = '%s'", $node->nid, $node->type, $node->title))) {
          form_set_error('title', "There is already a $type_name with that $type_title. Please enter a different $type_name");
        }

        // Disallow changes to group name (to avoid SEO issues and path alias getting out of sync with group content aliases)
        if ($node->nid) {
          $currentgroupname = db_result(db_query("SELECT title FROM {node} WHERE nid = %d", $node->nid));
          if (strtolower($currentgroupname) != strtolower($node->title)) {
            form_set_error('title', "Sorry, $type_name {$type_title}s are permanent and may not be changed. You can change capitalization only.");
          }
        }
      }
      
      // Disallow node titles that start with the subdomain token prefix character (else they'll be inadvertantly rewritten as subdomains)
      if (substr(trim($node->title), 0, strlen(SUBDOMAIN_TOKEN_PREFIX)) == SUBDOMAIN_TOKEN_PREFIX) {
        form_set_error('title', "Sorry, {$type_title}s can't start with a ". SUBDOMAIN_TOKEN_PREFIX .".");
      }
      break;

    case 'load':
      if (module_exists('og') && og_is_group_type($node->type)) {
        $node->subdomain = subdomain_raw_to_subdomain($node->title);
      }
      break;
  }
}

/**
 * Implementation of hook_user
 */
function subdomain_user($op, &$edit, &$user_edit, $category = NULL) {
  if ($op == 'validate' && variable_get('subdomain_type', SUBDOMAIN_USER) == SUBDOMAIN_USER) {
    // for SEO, disallow username changes if subdomains are set to Node Authors
    if (is_numeric($user_edit->uid) && $user_edit->name != $edit['name']) {
      form_set_error('name', 'Sorry. Usernames are permanent and cannot be changed');
    }

    // Allow only letters and spaces in usernames
    elseif (!preg_match('!^[a-zA-Z0-9 ]+$!', $edit['name'])) {
      form_set_error('name', 'User Names can contain only letters, numbers and spaces');
    }
  }
}

/**
 * Implementation of hook_token_list()
 */
function subdomain_token_list($type = 'all') {
  if ($type == 'node') {
    $tokens['node']['subdomain'] = t('Subdomain token (place at the start of the path)');
  }
  if ($type == 'taxonomy') {
    $tokens['taxonomy']['subdomain'] = t('Subdomain token (place at the start of the path)');
  }
  elseif ($type == 'user') {
    $tokens['user']['subdomain'] = t('Subdomain token (place at the start of the path)');
  }
  return $tokens;
}

/**
 * Implementation of hook_token_values()
 */
function subdomain_token_values($type, $object = NULL) {
  $values['subdomain'] = NULL;
  switch ($type) {
    case 'node':
      switch (variable_get('subdomain_type', SUBDOMAIN_USER)) {
        case SUBDOMAIN_GROUP:
          if (og_is_group_type($object->type)) {
              $values['subdomain'] = subdomain_build_token($object->title, $object->nid);
          }
          elseif (isset($object->og_groups)) {
            // 1st Group in og_groups array is used for subdomain
            if ($groupname = db_result(db_query("SELECT title FROM {node} WHERE nid = %d", $object->og_groups[0]))) {
              $values['subdomain'] = subdomain_build_token($groupname, $object->og_groups[0]);
            };
          }
          else {
            $values['subdomain'] = '';
          }
          break;

        case SUBDOMAIN_USER:
          if (!empty($object->name)) {
            $values['subdomain'] = subdomain_build_token($object->name, $object->uid);
          }
          break;

        case SUBDOMAIN_VOCAB:
          $vocab = variable_get('subdomain_vocab', NULL);
          if (is_array($object->taxonomy) && in_array($vocab, array_keys($object->taxonomy))) {
            if ($tid = is_array($object->taxonomy[$vocab]) ? array_shift($object->taxonomy[$vocab]) : $object->taxonomy[$vocab]) {
              $term_name = db_result(db_query("SELECT name FROM {term_data} WHERE tid = %d", $tid));
              $values['subdomain'] = subdomain_build_token($term_name, $tid);
            }
          }
          break;
      }
      break;

    case 'taxonomy':
      if (variable_get('subdomain_type', SUBDOMAIN_USER) == SUBDOMAIN_VOCAB) {
        $values['subdomain'] = subdomain_build_token($object->name, $object->tid);
      }
      break;
    case 'user':
      if (variable_get('subdomain_type', SUBDOMAIN_USER) == SUBDOMAIN_USER) {
        $values['subdomain'] = subdomain_build_token($object->name, $object->uid);
      }
  }
  return $values;
}

/**
 * Implementation of hook_admin_settings()
 */
function subdomain_admin_settings() {

  drupal_add_js(drupal_get_path('module', 'subdomain') . '/subdomain.js');

  $reserved_default = implode("\n", array('www', 'static'));

  $options[SUBDOMAIN_USER] = 'Node authors and author content';
  if (module_exists('taxonomy')) {
    $vocabs = taxonomy_get_vocabularies();
    if (is_array($vocabs) && count($vocabs)) {
      foreach ($vocabs as $vocab) {
        if (!$vocab->hierarchy) {
          $vocab_options[$vocab->vid] = $vocab->name;
        }
      }
      if (is_array($vocab_options)) {
        $options[SUBDOMAIN_VOCAB] = 'Taxonomy Vocabulary and associated content';        
      }
    }
  }
  if (module_exists('og')) {
    $options[SUBDOMAIN_GROUP] = 'Organic groups and group content';
  }

  $form['subdomain_settings']['subdomain_base_domain'] = array(
    '#type' => 'textfield',
    '#title' => t('Base domain'),
    '#description' => 'Your website\'s root domain (e.g. example.com)',
    '#default_value' => variable_get('subdomain_base_domain', NULL)
  );
  $form['subdomain_settings']['subdomain_type'] = array(
    '#type' => 'select', 
    '#title' => t('Create subdomains for'), 
    '#description' => !isset($options[SUBDOMAIN_GROUP]) || !isset($options[SUBDOMAIN_VOCAB]) ? 'To use Taxonomy or Organic Group mode, enable and configure those modules' : '',
    '#options' => $options, 
    '#default_value' => variable_get('subdomain_type', SUBDOMAIN_USER)
  );
  if (module_exists('taxonomy') && is_array($vocab_options)) {
    $form['subdomain_settings']['subdomain_vocab'] = array(
      '#type' => 'select',
      '#title' => t('With the selected vocabulary\'s terms'),
      '#options' => $vocab_options,
      '#default_value' => variable_get('subdomain_vocab', NULL),
      '#prefix' => '<div id="vocab" style="display:none;">',
      '#suffix' => '</div>'
    );
  }
  $form['subdomain_settings']['subdomain_style'] = array(
    '#type' => 'select',
    '#title' => t('Using'),
    '#options' => array(SUBDOMAIN_STYLE_NAME => 'Group or Node Author name', SUBDOMAIN_STYLE_CUSTOM => 'Custom...'),
    '#default_value' => variable_get('subdomain_style', SUBDOMAIN_STYLE_NAME)
  );
  $form['subdomain_settings']['subdomain_custom'] = array(
    '#type' => 'textfield',
    '#title' => t('Custom subdomain'),
    '#description' => 'You may use the tokens %name% and %id% for group-name/user-name and group-nid/user-uid respectively',
    '#default_value' => variable_get('subdomain_custom', NULL),
    '#prefix' => '<div id="subdomain_custom">',
    '#suffix' => '</div>'
  );
  $form['subdomain_settings']['subdomain_reserved'] = array(
    '#type' => 'textarea', 
    '#title' => t('But disallow the following subdomains'), 
    '#rows' => 10, 
    '#description' => 'Enter 1 subdomain per line', 
    '#default_value' => variable_get('subdomain_reserved', $reserved_default)
  );
  $form['subdomain_settings']['subdomain_next'] = array(
    '#type' => 'markup',
    '#value' => '<div style="width:500px;">After saving, configure <strong>automated alias settings</strong> and place the <strong>[subdomain]</strong> token at the start of any paths you want on a subdomain</div>'
  );
  return system_settings_form($form);
}

function subdomain_admin_settings_validate($form, &$form_state) {
  if ($form_state['values']['subdomain_style'] == SUBDOMAIN_STYLE_CUSTOM) {
    $custom = trim($form_state['values']['subdomain_custom']);
    if (empty($custom)) {
      form_set_error('subdomain_custom', 'You forgot to specify a custom subdomain rewrite');
    }
    else {
      // get valid tokens out of the way
      $custom = str_replace(SUBDOMAIN_CUSTOM_TOKEN_NAME, '', $custom);
      $custom = str_replace(SUBDOMAIN_CUSTOM_TOKEN_ID, '', $custom);

      // Only allow letters and spaces in custom rewrite
      if (!preg_match('!^[a-zA-Z ]+$!', $custom)) {
        form_set_error('subdomain_custom', 'Apart from the ' . SUBDOMAIN_CUSTOM_TOKEN_NAME . ' and ' . SUBDOMAIN_CUSTOM_TOKEN_ID . ' tokens, the custom rewrite can only contain letters and spaces');
      }
    }
  }
  elseif (!empty($form_state['values']['subdomain_custom'])) {
    // Erase custom subdomain token, if "custom" isn't the selected subdomain style
    form_set_value($form['subdomain_settings']['subdomain_custom'], NULL, $form_state);
  }
  
  if ($form_state['values']['subdomain_type'] != SUBDOMAIN_VOCAB && isset($form['subdomain_settings']['subdomain_vocab'])) {
    // Erase selected vocab if taxonomy vocab isn't the selected subdomain type
    form_set_value($form['subdomain_settings']['subdomain_vocab'], NULL, $form_state);
  }
}

/**
 * Returns True if $subdomain is reserved
 */
function subdomain_reserved($name = NULL) {
  $subdomain = subdomain_raw_to_subdomain($name);
  $reserved_subdomain = strtolower(ereg_replace(chr(13) . chr(10), "\n", variable_get('subdomain_reserved', '')));
  if (in_array($subdomain, explode("\n", $reserved_subdomain))) {
    return TRUE;
  }
}

/**
 * Looks for subdomain token in Drupal generated URLs; if present, prepends to domain
 * NOTE: Requires a 1-line patch to includes/common.inc (See install.txt)
 */
function subdomain_url_rewrite(&$path, &$options) {

  $options['absolute'] = TRUE;
  $base_domain = variable_get('subdomain_base_domain', NULL);

  if (substr($path, 0, SUBDOMAIN_TOKEN_PREFIX_LENGTH) == SUBDOMAIN_TOKEN_PREFIX) {
    $path_parts = explode('/', substr($path, SUBDOMAIN_TOKEN_PREFIX_LENGTH));
    $options['base_url'] = 'http://'. array_shift($path_parts) .'.'. $base_domain;
    $path = implode('/', $path_parts);
  }
  else {
    $options['base_url'] = 'http://'. $base_domain;
  }
}

function subdomain_raw_to_subdomain($groupname) {
  return str_replace(' ', '-', trim(strtolower($groupname)));
}

function subdomain_subdomain_to_token($subdomain) {
  return SUBDOMAIN_TOKEN_PREFIX . $subdomain;
}

function subdomain_build_token($name, $id) {
  if (variable_get('subdomain_style', SUBDOMAIN_STYLE_NAME) == SUBDOMAIN_STYLE_NAME) {
    $raw = $name;
  }
  elseif ($custom = variable_get('subdomain_custom', NULL)) {
    $raw = str_replace(SUBDOMAIN_CUSTOM_TOKEN_NAME, $name, $custom);
    $raw = str_replace(SUBDOMAIN_CUSTOM_TOKEN_ID, $id, $raw);
  }
  return subdomain_subdomain_to_token(subdomain_raw_to_subdomain($raw));
}