anspress-paid-membership/functions.php at master · anspress/anspress-paid-membership · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
<?php
/**
 * Insert level meta to pmpro_membership_levelmeta.
 * @param  integer $level_id   Level id.
 * @param  string  $meta_key   Meta key.
 * @param  string  $meta_value Meta value.
 * @return boolean|integer
 */
function ap_pm_add_level_meta( $level_id, $meta_key, $meta_value ) {
	global $wpdb;

	if ( ! $wpdb->pmpro_membership_levelmeta ) {
		return false;
	}

	$meta_value = maybe_serialize( $meta_value );

	$row = $wpdb->insert(
		$wpdb->pmpro_membership_levelmeta,
		array(
			'pmpro_membership_level_id' => $level_id,
			'meta_key' => $meta_key,
			'meta_value' => $meta_value,
		),
		array(
			'%d',
			'%s',
			'%s',
		)
	);

	if ( false !== $row ) {
		return $wpdb->insert_id;
	}

	return false;
}


function ap_pm_get_level_meta( $level_id, $meta_key = '', $single = false ) {
	return get_metadata( 'pmpro_membership_level', $level_id, $meta_key , $single );
}


function ap_pm_delete_level_meta( $level_id, $meta_key, $meta_value = '', $delete_all = false ) {
	return delete_metadata( 'pmpro_membership_level', $level_id, $meta_key, $meta_value, $delete_all );
}

/**
 * Update pmpro level meta
 * @param  [type]  $meta_id    [description]
 * @param  [type]  $meta_key   [description]
 * @param  [type]  $meta_value [description]
 * @param  boolean $prev_value [description]
 * @return [type]              [description]
 */
function ap_pm_update_meta( $level_id, $meta_key, $meta_value, $prev_value = false ) {
	return update_metadata( 'pmpro_membership_level', $level_id, $meta_key, $meta_value, $prev_value );
}

/**
 * Check if user have a capabilty based on user membership level.
 * @param  string          $cap     Capability to check.
 * @param  boolean|integer $user_id User ID.
 * @return boolean
 */
function ap_pm_user_level_can( $cap, $user_id = false ) {
	if ( false === $user_id ) {
		$user_id = get_current_user_id();
	}

	$level = pmpro_getMembershipLevelForUser( $user_id );

	if ( ! $level ) {
		return;
	}

	if ( ! pmpro_hasMembershipLevel( $level->ID, $user_id ) ) {
		return;
	}

	$level_caps = ap_pm_get_level_meta( $level->ID, 'ap_caps', true );

	if ( $level_caps && isset( $level_caps[$cap] ) && $level_caps[$cap] ) {
		return true;
	}

	return false;
}

/**
 * Returns level categories.
 * @param  integer $level_id Level ID.
 * @return array
 */
function ap_pm_get_level_categories( $level_id ) {
	global $wpdb;

	$cache = wp_cache_get( $level_id, 'ap_pm_level_category' );

	if ( false !== $cache ) {
		return $cache;
	}

	$level_categories = $wpdb->get_col("SELECT c.category_id FROM $wpdb->pmpro_memberships_categories c WHERE c.membership_id = '" . (int) $level_id . "'" );

	wp_cache_set( $level_id, $level_categories, 'ap_pm_level_category' );

	return $level_categories;
}

/**
 * Check if a user level have permission to access category.
 * @param  intgere         $term_id Term ID.
 * @param  boolean|integer $user_id User ID.
 * @return boolean
 */
function ap_pm_user_level_can_access_category( $term_id, $user_id = false ) {
	if ( false === $user_id ) {
		$user_id = get_current_user_id();
	}

	$level = pmpro_getMembershipLevelForUser( $user_id );

	if ( ! $level ) {
		return true;
	}

	$catgeories = ap_pm_get_level_categories( $level->id );

	if ( empty( $catgeories ) ) {
		return true;
	}

	if ( in_array( $term_id, $catgeories ) ) {
		return true;
	}

	return false;
}