102 lines
2.5 KiB
PHP
102 lines
2.5 KiB
PHP
|
<?php
|
|||
|
|
|
|||
|
|
if (!class_exists('WP_CLI')) {
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
class OSS_CLI_Commands
|
|||
|
|
{
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 同步文件夹到 OSS
|
|||
|
|
*
|
|||
|
|
* ## OPTIONS
|
|||
|
|
*
|
|||
|
|
* <path>
|
|||
|
|
* : 要同步的文件夹
|
|||
|
|
*
|
|||
|
|
* ## EXAMPLES
|
|||
|
|
*
|
|||
|
|
* wp oss upload wp-content/uploads
|
|||
|
|
*
|
|||
|
|
* @when after_wp_load
|
|||
|
|
*/
|
|||
|
|
public function upload($args, $assoc_args)
|
|||
|
|
{
|
|||
|
|
[$path] = $args;
|
|||
|
|
$dir = ABSPATH . $path;
|
|||
|
|
if (!is_dir($dir)) {
|
|||
|
|
WP_CLI::error("Directory not found: [{$dir}]");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
WP_CLI::line("Uploading files from [{$dir}] to OSS...");
|
|||
|
|
|
|||
|
|
$files = oss_read_dir_queue(ABSPATH, $path);
|
|||
|
|
if (empty($files)) {
|
|||
|
|
WP_CLI::success('No files to upload.');
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
foreach ($files as $file) {
|
|||
|
|
$status = oss_file_upload($file['key'], $file['filepath']);
|
|||
|
|
if ($status) {
|
|||
|
|
WP_CLI::line("Uploaded: {$file['key']}");
|
|||
|
|
} else {
|
|||
|
|
WP_CLI::line("Failed: {$file['key']}");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
$total = count($files);
|
|||
|
|
WP_CLI::success("Uploaded {$total} files.");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 同步文件到 OSS
|
|||
|
|
*
|
|||
|
|
* ## OPTIONS
|
|||
|
|
*
|
|||
|
|
* <path>
|
|||
|
|
* : 要同步的文件
|
|||
|
|
*
|
|||
|
|
* [--delete]
|
|||
|
|
* : 如果设置，上传后会删除本地文件
|
|||
|
|
* [--key=<key>]
|
|||
|
|
* : 指定上传到 OSS 的 key，默认和文件路径一致
|
|||
|
|
*
|
|||
|
|
* ## EXAMPLES
|
|||
|
|
*
|
|||
|
|
* wp oss upload-file wp-content/uploads/2021/01/1.jpg
|
|||
|
|
* wp oss upload-file wp-content/uploads/2021/01/1.jpg --delete
|
|||
|
|
* wp oss upload-file wp-content/uploads/2021/01/1.jpg --key=2021/01/1.jpg
|
|||
|
|
*
|
|||
|
|
* @when after_wp_load
|
|||
|
|
* @subcommand upload-file
|
|||
|
|
*/
|
|||
|
|
public function upload_file($args, $assoc_args)
|
|||
|
|
{
|
|||
|
|
[$path] = $args;
|
|||
|
|
$file = ABSPATH . $path;
|
|||
|
|
if (!is_file($file)) {
|
|||
|
|
WP_CLI::error("File not found: {$file}");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
$delete = false;
|
|||
|
|
if (isset($assoc_args['delete'])) {
|
|||
|
|
$delete = true;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
$key = isset($assoc_args['key']) ? $assoc_args['key'] : $path;
|
|||
|
|
|
|||
|
|
WP_CLI::line("Uploading file [{$file}] to OSS with key [$key]...");
|
|||
|
|
|
|||
|
|
$status = oss_file_upload("/{$key}", $file, $delete);
|
|||
|
|
if ($status) {
|
|||
|
|
WP_CLI::success("Uploaded: {$path}");
|
|||
|
|
} else {
|
|||
|
|
WP_CLI::error("Failed: {$path}");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
WP_CLI::add_command('oss', 'OSS_CLI_Commands', ['shortdesc' => 'Commands used to operate OSS.']);
|