functions.php

<?php
/**
 * @package Make Child
 */

/**
 * The theme version.
 */
define( 'TTFMAKE_CHILD_VERSION', '1.1.0' );

/**
 * Turn off the parent theme styles.
 *
 * If you would like to use this child theme to style Make from scratch, rather
 * than simply overriding specific style rules, simply remove the '//' from the
 * 'add_filter' line below. This will tell the theme not to enqueue the parent
 * stylesheet along with the child one.
 */
//add_filter( 'make_enqueue_parent_stylesheet', '__return_false' );

/**
 * Add your custom theme functions here.
 */


/* redirect users to blog page after login */
function redirect_to_front_page() {
global $redirect_to;
if (!isset($_GET['redirect_to'])) {
//$redirect_to = get_option('siteurl');

$redirect_to = '/blog';
}
}
add_action('login_form', 'redirect_to_front_page');

add_action('wp_logout','auto_redirect_after_logout');
function auto_redirect_after_logout(){
  wp_redirect( home_url() );
  exit();
}
add_action( 'woocommerce_before_my_account', 'my_laser_usage' );

function my_laser_usage() {

    print("<hr><h2>Laser Cutter Usage</h2>\n");

    $userid = get_current_user_id();
    $auth = md5($userid."3e532d5d228f9594c55d72978e3d3663ef32ed64dad4d078236ebd24c6a1a2d4fb872d4ef9a84eca78ffeb449afd98a6b4080cecb4124a9cb0dbd3188bf80415");
    $data = json_decode(file_get_contents("https://ssl.acemonstertoys.org/member/mylaserfees.php?userid=$userid&auth=$auth"),true);

    if(!isset($data['ok'])) {
        print("Error loading laser cutter usage, sorry!<br>");
        return;
    }

    if(!isset($data['rows']) || count($data['rows']) == 0) {
        print("No laser cutter usage found under this account. Updates hourly.<br>");
        return;
    }
    
    print("<table border=1>");
    print("<thead>");
    print("<tr>");
    print("<th>date</th>");
    print("<th>usage minutes</th>");
    print("<th>fee</th>");
    print("<th>billed</th>");
    print("<th>note</th>");
    print("</tr>");
    print("</thead>");

    foreach($data['rows'] as $row) {
        print("<tr>");
        print("<td>");
        print(implode("</td><td>",$row));
        print("</td>");
        print("</tr>");
    }

    print("</table>");

}

// return true if a given RFID is a member
function is_member($rfid, $opts = array()) {
  $args = array( 'meta_key' => 'member_rfid', 'meta_value' => $rfid );
  $user_query = new WP_User_Query( array_merge($args, $opts) );
  return !empty($user_query->results);
}

//////////////// API ///////////////////////

// Find user ID associated with rfid
function find_member_ids_from_rfid($rfid) { #API2
  $args = array( 'meta_key' => 'member_rfid', 'meta_value' => trim($rfid) );
  $user_query = new WP_User_Query( $args );
  $ids = array();

  //
  //$test_query = new WP_User_Query( array( 'number' => 200) );
  //echo var_export($test_query->results, true);
  //
  //echo var_export(get_field('member_rfid', 'user_149'), true);


  //echo var_export($user_query->results,true);
  foreach($user_query->results as $user){
    array_push($ids, $user->ID);
  }
  return $ids;  
}

// Get all memberships as slugs
function membership_plan_slugs() { #API3
  $plan_slugs = array();
  foreach(wc_memberships_get_membership_plans() as $plan) {
    if($plan->slug == "bot-bot-program-membership") continue; // ignore box bot plan
    array_push($plan_slugs, $plan->slug);
  }
  return $plan_slugs;
}

// Check if a given user ID is active or not
function is_member_active($user_id) { #API2
  $active = false;

  foreach(membership_plan_slugs() as $slug) {    
    if(wc_memberships_is_user_active_member($user_id, $slug)) {
      $active = true;
    }      
  }
  return $active;
}

// Check if a given RFID is good or not
function rfid_status( $request ) { #API
  if(validate_request($request)) {
    return invalid_request_response();
  }

  $success = false;
  $rfid = $request->get_params('id');
  //still handle one rfid assigned to multiple users
  $user_ids = find_member_ids_from_rfid($rfid); 

  foreach($user_ids as $user_id) {
    if(is_member_active($user_id)) {
      $success = true;
    }
  }

  return $success;
};

// Get the RFID for a specific user given their UserID
function rfid_for_user( $user_id ) { #API2
  // $args = array( 'meta_key' => 'member_rfid', 'user_id' => $user_id );
  // $user_query = new WP_User_Query( $args );  
  // $rfid = $user_query->results;
  $rfid = get_user_meta($user_id, 'member_rfid', true);
  return $rfid; 
}

// Loop over all users, find someone whose membership has expired, 
// then if they are 'author' or 'customer', set their role to 'subscriber' instead
function expire_user_role() {
/*    if (! wp_next_scheduled ( 'amt_expire_user_role' )) {
        wp_schedule_event(time(), 'hourly', 'amt_expire_user_role');
    }*/

  // Get the list of valid membership plans
  $plan_slugs = membership_plan_slugs();

  $out = array();

  foreach(get_users() as $user) {
    $activePlan = false;

    // loop over all the plans, see if the member has one
    foreach($plan_slugs as $slug) {    
      if(wc_memberships_is_user_active_member($user->ID, $slug)) {
        $activePlan = true;
        $sslug = $slug;
        break;
      }      
    }

    $activeRole = false;
    $adminRole = false;
    $rrole = $user->roles[0];
    $arole = "--";

    // loop over all the user's roles, look for a role that is not author or customer
    foreach($user->roles as $role) {
      if($role == 'author' || $role == 'editor') {
        $activeRole = true;
        $rrole = $role;
      }
      if($role == 'shop_manager' || $role == 'administrator') {
          $adminRole = true;
          $arole = $role;
      }
    }

    // Big logic tree of all possible combinations of active/inactive/etc
    if($activePlan) {
        if($adminRole) { // active membership & is admin -> skip
            $out[] = array('id'=>$user->user_email, 'r'=>"active $sslug admin $arole");
            continue;
        }
        if(!$activeRole) { // active membership, no active role -> upgrade to active role
            $out[] = array('id'=>$user->user_email, 'r'=>'UPGRADE');
            $user->set_role('author');
            // log it on the admin site
            @file_get_contents("http://acemonstertoys.org/member/changerole.php?up=1&msg=".urlencode($user->user_email));
            continue;
        }
        // active membership, active role -> skip
        $out[] = array('id'=>$user->user_email, 'r'=>"active $sslug role $rrole");
        continue;
    } else {
        if($adminRole) { // no active membership & is admin -> skip
            $out[] = array('id'=>$user->user_email, 'r'=>"inactive admin $arole");
            continue;
        }
        if(!$activeRole) { // no active membership & no active role -> skip
            $out[] = array('id'=>$user->user_email, 'r'=>"inactive role $rrole"); 
            continue;
        }

        // no active membership but active role -> downgrade role
        $out[] = array('id'=>$user->user_email, 'r'=>'DOWNGRADE');
        $user->set_role('subscriber');

        // log it on the admin site
        @file_get_contents("http://acemonstertoys.org/member/changerole.php?down=1&msg=".urlencode($user->user_email));
    }
  }

  return $out;
}



// Slurp all users and their info
function get_all_users_info($request) { #API
  if(!validate_request($request)) {
    return invalid_request_response();
  }

  $out = array();

  // Get all the plans (so we can put their names & slugs into the results without looking up each time)
  $plans_in = wc_memberships_get_membership_plans();
  $plans = array();
  foreach($plans_in as $plan)
    $plans[$plan->id] = $plan;

  foreach(get_users() as $user) {
    $umeta = get_user_meta($user->ID);
    $meta = array();

    // List of fields from user metadata that we'd like to pull
$fields = array('nickname','description','wp_capabilities','wp_user_level','ing_phone','ing_email','member_rfid','member_door_code','amt_membership_legal','amt_status_handle','membership_date','membership_agreement_date','liability_waiver_date','has_taken_laser_class','twitter_handle','billing_first_name','billing_last_name','billing_company','billing_address_1','billing_address_2','billing_city','billing_postcode','billing_country','billing_state','billing_phone','billing_email','shipping_first_name','shipping_last_name','shipping_company','shipping_address_1','shipping_address_2','shipping_city','shipping_postcode','shipping_country','shipping_state','account_status','role','_stripe_customer_id','_order_count','_money_spent','paying_customer');
    foreach($fields as $field) {
       if(isset($umeta[$field]))
         if(is_array($umeta[$field]))
           $meta[$field] = $umeta[$field][0];
         else
           $meta[$field] = $umeta[$field];
    }

    // List of fields from the user object we'd like to pull
    $fields = array('ID','user_login','user_nicename','user_email','user_url','display_name','first_name','last_name');
    foreach($fields as $field)
      if(isset($user, $field))
        $meta[$field] = $user->{$field};

    // Get this guy's WC memberships
    $uplans_in = wc_memberships_get_user_memberships($user->ID);
    $uplans = array();

    // Pull out the data for each plan that we are interested in
    foreach($uplans_in as $uplan_in) {
      $uplan = array();
      $uplan['id'] = $uplan_in->plan_id;
      $uplan['status'] = $uplan_in->get_status();
      $uplan['name'] = $plans[$uplan_in->plan_id]->name;
      $uplan['slug'] = $plans[$uplan_in->plan_id]->slug;
      $uplan['end'] = $uplan_in->get_end_date('timestamp');
      $uplans[] = $uplan;
    }

    $meta['plans'] = $uplans;
    $out[] = $meta;
  }

  return $out;
}

// Return list of all active RFIDs
function active_rfids($request) { #API
  if(!validate_request($request)) {
    return invalid_request_response();
  }

  $rfids = array();

  foreach(get_users() as $user) {
    $user_id = $user->ID;
    if(is_member_active($user_id)) {
      $rfid = rfid_for_user($user_id);
      if(!empty($rfid))
        array_push($rfids, $rfid);
    }
  }
  
  $out = array("OK",$rfids);

  return $out;
}

// Counts the number of active, inactive, and operator users
function count_active_users($request) { #API
  $countA = $countI = $countO = 0;

  foreach(get_users() as $user)
    if(wc_memberships_is_user_active_member($user->ID, 'ops-membership')) 
      $countO++;
    elseif(is_member_active($user->ID)) 
      $countA++;
    else
      $countI++;

  return array($countA, $countI, $countO);
}

// Authenticates the request (just check if they have the right password)
function validate_request($request) { #API2
  $header = $request->get_header('X-Amt-Auth');
  /* FIXME: move to someplace not here */
 return $header == "3e532d5d228f9594c55d72978e3d3663ef32ed64dad4d078236ebd24c6a1a2d4fb872d4ef9a84eca78ffeb449afd98a6b4080cecb4124a9cb0dbd3188bf80415";
   // return true;
}

// Standard error for wrong pass
function invalid_request_response() { #API2
  return "['ERROR: Request is invalid']";
}

// Find the user associated with a given RFID
function get_user_for_rfid($request) { #API
  if(!validate_request($request)) {
    return invalid_request_response();
  }
    
  $rfid = trim($request->get_param('id'));
  $member_id = find_member_ids_from_rfid($rfid);

  if(empty($member_id))
    return "Could not find user with ".var_export($rfid, true);
  else
    $member_id = $member_id[0]; // hopefully there aren't multiple, but maybe there are, pick first

  $amtStatusHandle = get_field('amt_status_handle', 'user_'.$member_id);
  $userData = get_user_by('ID', $member_id)->to_array();
  $userData["amt_status_handle"] = $amtStatusHandle;
  return $userData;
}


// Set the end date for a user's membership
// Pass in:
// 'email' (can be email address or userID)
// 'time' unix timestamp to set end date. Must set it in the future for
function set_user_end_date($request) { #API
  if(!validate_request($request)) {
    return invalid_request_response();
  }
    
  if(!isset($request['email']))
      return "ERROR: Missing email";

  if(!isset($request['time']))
      return "ERROR: Missing end time";

  $userid = trim($request['email']);
  $date = trim($request['time']);

  if(isset($request['slug']))
      $searchslug = $request['slug'];
  else
      $searchslug = 'legacy';

  $user = get_user_by('ID', $userid);
  if(!$user)
    $user = get_user_by('email', $userid );
  if(!$user)
    return "ERROR: Cannot find user '$userid'";

  // Get all plans up front so we can look at info about them later
  $plans_in = wc_memberships_get_membership_plans();
  $plans = array();
  foreach($plans_in as $plan)
    $plans[$plan->id] = $plan;

  $slugs = array();
  foreach($plans as $plan)
      $slugs[$plan->id] = $plan->slug;

  // Get all the memberships belonging to the user
  $uplans = wc_memberships_get_user_memberships($user->ID);

  $out = array();

  // Look for plans that are active and match the slug we are searching for, then cancel them
  foreach($uplans as &$plan) {
      $status = $plan->get_status();
      $slug = $plans[$plan->plan_id]->slug;
      if($status == 'active' && $slug == $searchslug) {
          $out[] = array('id'=>$plan->plan_id, 'slug'=>$plans[$plan->plan_id]->slug, 'end'=>$date);
          $plan->set_end_date($date);
          $plan->cancel_membership("Expired on old site");
      }
  }
  
  return array('id'=>$user->id, 'email'=>$user->user_email, 'plans_updated'=>$out, 'searchslug'=>$searchslug);
}

// Send out customer invoice email for an order
// Pass in:
// 'orderid'
function send_customer_invoice($request) { #API
  if(!validate_request($request)) {
    return invalid_request_response();
  }
    
  if(!isset($request['orderid']))
      return "ERROR: Missing email";

  $order = wc_get_order(absint($request['orderid']));
  if(!$order) 
      return "ERROR: Couldn't find order '".$request['orderid']."'";

  $wc = WC();
  if(!$wc) return "ERROR: Couldn't get WC?";
  $mailer = $wc->mailer();
  if(!$mailer) return "ERROR: Couldn't get mailer?";
  $mailer->customer_invoice($order);

  return "Probably OK";
}




// Map API URLs to PHP methods
add_action( 'rest_api_init', function () {
 register_rest_route( 'amt/v1', '/rfid/(?P<id>[a-zA-Z0-9]+)', array(
    'methods' => 'GET',
    'callback' => 'rfid_status'
  ) );

 register_rest_route( 'amt/v1', '/rfids/active', array(
    'methods' => 'GET',
    'callback' => 'active_rfids'
  ) ); 

 register_rest_route( 'amt/v1', '/rfid/(?P<id>[a-zA-Z0-9]+)/user', array(
    'methods' => 'GET',
    'callback' => 'get_user_for_rfid'
  ) ); 

 register_rest_route( 'amt/v1', '/user/count', array(
    'methods' => 'GET',
    'callback' => 'count_active_users'
  ) ); 

 register_rest_route( 'amt/v1', '/user/info', array(
    'methods' => 'GET',
    'callback' => 'get_all_users_info'
  ) ); 

 register_rest_route( 'amt/v1', '/user/expire', array(
    'methods' => 'GET',
    'callback' => 'set_user_end_date'
  ) ); 

 register_rest_route( 'amt/v1', '/order/invoice', array(
    'methods' => 'GET',
    'callback' => 'send_customer_invoice'
  ) ); 

 // uncomment to test the expire-all-users logic remotely and see results
/* register_rest_route( 'amt/v1', '/user/expireALL', array(
    'methods' => 'GET',
    'callback' => 'expire_user_role'
) ); */
} );


// Link action name to our function
add_action('amt_expire_user_role', 'expire_user_role');

// Schedule our action when theme is activated
function custom_amt_activation($oldname, $oldtheme=false) {
    if (! wp_next_scheduled ( 'amt_expire_user_role' )) {
        wp_schedule_event(time(), 'hourly', 'amt_expire_user_role');
    }
}

// Link theme activation to our custom activation function
add_action("after_switch_theme", "custom_amt_activation", 10 ,  2);

// Remove scheduled action when theme deactivated
function custom_amt_deactivation($newname, $newtheme) {
    wp_clear_scheduled_hook('amt_expire_user_role');
}

// Link theme deactivation to our custom deactivation function
add_action("switch_theme", "custom_amt_deactivation", 10 , 2);

// Send toybot notification on new orders
function notify_new_order($order_id) {
    $order = wc_get_order($order_id);
    $total = $order->get_formatted_order_total();
    @file_get_contents("http://acemonstertoys.org/member/neworder.php?amt=".urlencode($total));
}
add_action('woocommerce_order_status_completed', 'notify_new_order');


//Member Directory

if ( ! function_exists( 'skills_taxonomy' ) ) {

// Register Custom Taxonomy
function skills_taxonomy() {

  $labels = array(
    'name'                       => _x( 'Skills', 'Taxonomy General Name', 'text_domain' ),
    'singular_name'              => _x( 'Skill', 'Taxonomy Singular Name', 'text_domain' ),
    'menu_name'                  => __( 'Taxonomy', 'text_domain' ),
    'all_items'                  => __( 'All Items', 'text_domain' ),
    'parent_item'                => __( 'Parent Item', 'text_domain' ),
    'parent_item_colon'          => __( 'Parent Item:', 'text_domain' ),
    'new_item_name'              => __( 'New Item Name', 'text_domain' ),
    'add_new_item'               => __( 'Add New Item', 'text_domain' ),
    'edit_item'                  => __( 'Edit Item', 'text_domain' ),
    'update_item'                => __( 'Update Item', 'text_domain' ),
    'view_item'                  => __( 'View Item', 'text_domain' ),
    'separate_items_with_commas' => __( 'Separate items with commas', 'text_domain' ),
    'add_or_remove_items'        => __( 'Add or remove items', 'text_domain' ),
    'choose_from_most_used'      => __( 'Choose from the most used', 'text_domain' ),
    'popular_items'              => __( 'Popular Items', 'text_domain' ),
    'search_items'               => __( 'Search Items', 'text_domain' ),
    'not_found'                  => __( 'Not Found', 'text_domain' ),
    'no_terms'                   => __( 'No items', 'text_domain' ),
    'items_list'                 => __( 'Items list', 'text_domain' ),
    'items_list_navigation'      => __( 'Items list navigation', 'text_domain' ),
  );
  $args = array(
    'labels'                     => $labels,
    'hierarchical'               => false,
    'public'                     => true,
    'show_ui'                    => true,
    'show_admin_column'          => true,
    'show_in_nav_menus'          => true,
    'show_tagcloud'              => true,
  );
  register_taxonomy( 'skills', array( 'amt_profile' ), $args );

}
add_action( 'init', 'skills_taxonomy', 0 );

}

// Adding a custom post type

if ( ! function_exists('amt_profile_post_type') ) {

// Register Custom Post Type
function amt_profile_post_type() {

  $labels = array(
    'name'                  => _x( 'AMT Profiles', 'Post Type General Name', 'text_domain' ),
    'singular_name'         => _x( 'AMT Profile', 'Post Type Singular Name', 'text_domain' ),
    'menu_name'             => __( 'AMT Profiles', 'text_domain' ),
    'name_admin_bar'        => __( 'AMT Profile', 'text_domain' ),
    'archives'              => __( 'AMT Profile Archives', 'text_domain' ),
    'parent_item_colon'     => __( 'Parent Item:', 'text_domain' ),
    'all_items'             => __( 'All AMT Profiles', 'text_domain' ),
    'add_new_item'          => __( 'Add New AMT Profiles', 'text_domain' ),
    'add_new'               => __( 'Add New', 'text_domain' ),
    'new_item'              => __( 'New Item', 'text_domain' ),
    'edit_item'             => __( 'Edit Item', 'text_domain' ),
    'update_item'           => __( 'Update Item', 'text_domain' ),
    'view_item'             => __( 'View Item', 'text_domain' ),
    'search_items'          => __( 'Search Item', 'text_domain' ),
    'not_found'             => __( 'Not found', 'text_domain' ),
    'not_found_in_trash'    => __( 'Not found in Trash', 'text_domain' ),
    'featured_image'        => __( 'Featured Image', 'text_domain' ),
    'set_featured_image'    => __( 'Set featured image', 'text_domain' ),
    'remove_featured_image' => __( 'Remove featured image', 'text_domain' ),
    'use_featured_image'    => __( 'Use as featured image', 'text_domain' ),
    'insert_into_item'      => __( 'Insert into item', 'text_domain' ),
    'uploaded_to_this_item' => __( 'Uploaded to this item', 'text_domain' ),
    'items_list'            => __( 'Items list', 'text_domain' ),
    'items_list_navigation' => __( 'Items list navigation', 'text_domain' ),
    'filter_items_list'     => __( 'Filter items list', 'text_domain' ),
  );
  $args = array(
    'label'                 => __( 'AMT Profile', 'text_domain' ),
    'description'           => __( 'AMT Profiles for members', 'text_domain' ),
    'labels'                => $labels,
    'supports'              => array( 'title', 'editor', 'excerpt', 'author', 'thumbnail', 'comments', 'revisions', 'page-attributes', ),
    'taxonomies'            => array( 'skills' ),
    'hierarchical'          => false,
    'public'                => true,
    'show_ui'               => true,
    'show_in_menu'          => true,
    'menu_position'         => 5,
    'menu_icon'             => 'dashicons-palmtree',
    'show_in_admin_bar'     => true,
    'show_in_nav_menus'     => true,
    'can_export'            => true,
    'has_archive'           => true,    
    'exclude_from_search'   => true,
    'publicly_queryable'    => true,
    'capability_type'       => 'post',
  );
  register_post_type( 'amt_profile', $args );

}
add_action( 'init', 'amt_profile_post_type', 0 );

}


function get_amt_profile_page_link($user)
{

  $args=array(
  'post_type' => 'amt_profile',
  'post_status' => 'any',
  'author' => $user->ID,
  'posts_per_page' => -1
);


$amt_profile_link = 'profile_not_found';
$my_query = null;
$my_query = new WP_Query($args);
if( $my_query->have_posts() ) {

  $amt_profile_posts = $my_query->get_posts();

  foreach($amt_profile_posts as $post) {
      $amt_profile_link = get_permalink($post);
                        break;
  }
}
wp_reset_query();

return $amt_profile_link;

}

// extending author abilities for skills tags

function allow_author_to_manage_categories()
{
  $user_role = 'author';
  $author    = get_role($user_role):
  $author->add_cap('manage_categories');
}
add_action('init', 'allow_author_to_manage_categories');

?>