我正在使用CMB脚本(WordPress的自定义元框和字段)为具有cliente自定义角色的不同用户上传文件。
我能够显示字段并使它们工作(尽管它们只在编辑用户时出现，而不是在创建用户时出现，但这是另一个问题)。我现在想要实现的是为不同的用户上传文件到不同的文件夹。
代码如下:
// Here we add a new user role, "cliente"
add_role( 'cliente', 'Cliente' );
add_filter('wp_handle_upload_prefilter', 'my_upload_prefilter');
add_filter('wp_handle_upload', 'my_upload_postfilter');
function my_upload_prefilter( $file ) {
add_filter('upload_dir', 'custom_upload_dir');
return $file;
}
function my_upload_postfilter( $fileinfo ) {
remove_filter('upload_dir', 'custom_upload_dir');
return $fileinfo;
}
function custom_upload_dir( $path ) {
global $pagenow;
// Check if we are on the user-edit.php page
if ( $pagenow == 'user-edit.php' && isset($_GET['user_id']) ) {
// Set the role we want to change the path for
$role_to_check = 'cliente';
// Get a bunch of user info for later use
$user_id = filter_var( $_GET['user_id'], FILTER_SANITIZE_NUMBER_INT );
$meta = get_user_meta($user_id);
$roles = unserialize($meta['wp_capabilities'][0]);
// If we are on the chosen role page, set the $customdir to first_name + last_name
if ( !empty($roles[$role_to_check]) ) {
$customdir = '/' . $meta['first_name'][0] . $meta['last_name'][0];
}
} else {
// Here we are not on the user-edit.php page. This is just a check to prove that WP is not recognizing the correct page, maybe because we are doing an Ajax call when this function is called. Confusing.
$customdir = '/did-not-work';
}
// If there is any error, just return the $path and abort the rest.
if ( !empty( $path['error'] ) ) {
return $path;
}
// Here we set the new $path with the $customdir set above
$path['path'] = str_replace($path['subdir'], '', $path['path']); //remove default subdir (year/month)
$path['url'] = str_replace($path['subdir'], '', $path['url']);
$path['subdir'] = $customdir;
$path['path'] .= $customdir;
$path['url'] .= $customdir;
return $path;
}
从几个检查，我已经运行它似乎我的代码检索用户id和存储在数据库中的数据，但它不检测它时上传的图像。这是否与我们通过Ajax或其他方式上传图像有关?
只是为了清楚，我不希望上传基于当前登录的用户，但在用户我，作为一个超级管理员，正在编辑使用edit-user.php页面。
我自己想出来的。首先是正确的代码，然后是解释:
// Here we add a new user role, "cliente".
add_role( 'cliente', 'Cliente' );
// These are the filters we need to add in order to modify the default upload path.
add_filter('wp_handle_upload_prefilter', 'my_upload_prefilter');
add_filter('wp_handle_upload', 'my_upload_postfilter');
function my_upload_prefilter( $file ) {
add_filter('upload_dir', 'custom_upload_dir');
return $file;
}
function my_upload_postfilter( $fileinfo ) {
remove_filter('upload_dir', 'custom_upload_dir');
return $fileinfo;
}
function custom_upload_dir( $path ) {
// When uploading, the file gets sent to upload_async.php, so we need to take the referral page in order to be able to get the user_id we need. We then take the query string, pass it through parse_str and store it in a $query_array. Took me a while to figure it out, but now it works like a charm.
$actual_page = $_SERVER['HTTP_REFERER'];
parse_str( parse_url($actual_page, PHP_URL_QUERY), $query_array );
// Check if we are uploading from the user-edit.php page.
if ( strpos($actual_page, 'user-edit.php') ) {
// Set the role we want to change the path for.
$role_to_check = 'cliente';
// Get a bunch of user info for later use
$user_id = filter_var( $query_array['user_id'], FILTER_SANITIZE_NUMBER_INT );
$meta = get_user_meta( $user_id );
$roles = unserialize( $meta['wp_capabilities'][0] );
// If we are on the chosen role page, set the $customdir to first_name + last_name
if ( !empty($roles[$role_to_check]) ) {
$customdir = '/docs/' . $meta['first_name'][0] . $meta['last_name'][0];
// If there is any error, just return the $path and abort the rest.
if ( !empty( $path['error'] ) ) {
return $path;
}
// Here we set the new $path with the $customdir set above
$new_subdir = $customdir . $path['subdir'];
$path['path'] = str_replace( $path['subdir'], $new_subdir, $path['path'] );
$path['url'] = str_replace( $path['subdir'], $new_subdir, $path['url'] );
$path['subdir'] = $new_subdir;
return $path;
}
} else {
// We are not uploading from user-edit.php, so go ahead as per default.
return $path;
}
}
问题是，当通过Ajax上传，$pagenow正确存储async-upload.php页，而不是我们在url。我只需要通过php $_SERVER['HTTP_REFERER']检索推荐页面(请注意，referer的拼写错误是因为http规范中的遗留拼写错误，有趣的东西)。
还请注意PHP规范不鼓励使用HTTP_REFERER，因为它可能根据服务器配置产生意想不到的结果，但在这种情况下，我应该完全控制服务器，所以它应该不是问题。如果您遇到任何问题，我建议您检查一下。
一旦我有了正确的url，我就能够解析它并检查我们是否在user-edit.php上，如果我们是，从查询字符串中获得user_id并从那里继续。
我花了一些时间才弄明白，但事后看来，这很容易。
希望它能在将来帮助到别人。