| Current File : /home/pacjaorg/www/copwordpres/wp-content/plugins/betterdocs/includes/Admin/CSVExporter.php |
<?php
namespace WPDeveloper\BetterDocs\Admin;
if ( ! defined( 'ABSPATH' ) ) {
exit; // Exit if accessed directly.
}
#[\AllowDynamicProperties]
class CSVExporter {
private static $default_args = [
'content' => 'docs',
'author' => false,
'category' => false,
'start_date' => false,
'end_date' => false,
'status' => false,
'offset' => 0,
'limit' => -1,
'meta_query' => [], // If specified `meta_key` then will include all post(s) that have this meta_key.
'query_args' => []
];
/**
* @var array
*/
private $args;
/**
* @var wpdb
*/
private $wpdb;
public function __construct( array $args = [] ) {
global $wpdb;
$this->args = wp_parse_args( $args, self::$default_args );
$this->wpdb = $wpdb;
}
public function combine_csv_data($csv_data_array) {
// Combine headers
$headers_combined = $csv_data_array[0][0];
foreach ($csv_data_array as $csv_data) {
$headers_combined = array_merge($headers_combined, array_slice($csv_data[0], 1));
}
$csv_data_combined = [$headers_combined];
// Combine data
for ($i = 1; $i < count($csv_data_array[0]); $i++) {
$combined_row = [];
foreach ($csv_data_array as $csv_data) {
$combined_row = array_merge($combined_row, array_fill(0, count($csv_data_combined[0]) - count($combined_row)), [$csv_data[$i][0]], array_slice($csv_data[$i], 1));
}
$csv_data_combined[] = $combined_row;
}
return $csv_data_combined;
}
public function run(): array {
$post_args = [
'post_type' => 'docs',
'posts_per_page' => -1
];
if ( isset($this->args['post__in']) ) {
$post_args['post__in'] = $this->args['post__in'];
}
if ( isset($this->args['category_terms']) ) {
$post_args['tax_query'] = [
[
'taxonomy' => 'doc_category',
'field' => 'slug',
'terms' => $this->args['category_terms'],
],
];
}
if ( isset($this->args['kb_terms']) ) {
$post_args['tax_query'] = [
[
'taxonomy' => 'knowledge_base',
'field' => 'slug',
'terms' => $this->args['kb_terms'],
],
];
}
$posts = get_posts($post_args);
$post_ids = array_map(function ($post) {
return $post->ID;
}, $posts);
$csv_data_terms = $this->get_terms_csv_data($post_ids);
$csv_data_author = $this->authors_list($post_ids);
$csv_data_posts = $this->get_posts_csv_data($posts);
// Combine headers
$headers_combined = array_merge($csv_data_posts[0], array_slice($csv_data_author[0], 1), array_slice($csv_data_terms[0], 1));
// Initialize the combined array with headers
$csv_data_combined = [$headers_combined];
// Combine term data with existing combined array
for ($i = 1; $i < count($csv_data_posts); $i++) {
$combined_row = array_merge($csv_data_posts[$i], array_fill(0, count($headers_combined) - count($csv_data_posts[$i]), ''));
$csv_data_combined[] = $combined_row;
}
// Calculate the starting index for author data
$author_start_index = count($csv_data_posts[0]);
// Combine author data with existing combined array
for ($i = 1; $i < count($csv_data_author); $i++) {
$combined_row = array_merge([$csv_data_author[$i][0]], array_fill(1, $author_start_index - 1, ''), array_slice($csv_data_author[$i], 1));
$csv_data_combined[] = $combined_row;
}
// Calculate the starting index for post data
$terms_start_index = $author_start_index + count($csv_data_author[0]) - 1;
// Combine post data with existing combined array
for ($i = 1; $i < count($csv_data_terms); $i++) {
$combined_row = array_merge([$csv_data_terms[$i][0]], array_fill(1, $terms_start_index - 1, ''), array_slice($csv_data_terms[$i], 1));
$csv_data_combined[] = $combined_row;
}
$filename = 'betterdocs.' . date('Y-m-d') . '.csv';
$csv_content = $this->generate_csv($csv_data_combined);
return [
'success' => true,
'data' => [
'filename' => $filename,
'filetype' => 'text/csv',
'download' => $csv_content,
],
];
}
public function get_terms( array $post_ids ) {
return wp_get_object_terms( $post_ids, get_object_taxonomies( 'docs' ) );
}
/**
* Sorting function for sorting term data.
*
* @param array $a The first array to compare.
* @param array $b The second array to compare.
*
* @return int Returns an integer less than, equal to, or greater than zero if the first array is considered
* to be respectively less than, equal to, or greater than the second.
*/
public function sort_terms($a, $b) {
// Compare parent values
$parentComparison = strcmp($a[7], $b[7]);
// If parent values are equal, compare Term ID values
return ($parentComparison === 0) ? strcmp($a[2], $b[2]) : $parentComparison;
}
private function get_terms_csv_data($post_ids) {
$csv_data_terms = [];
// Add CSV headers for terms
$csv_data_terms[] = [
'Type',
'Taxonomy',
'Term ID',
'Term name',
'Term slug',
'Term group',
'Term description',
'Term parent',
'Assigned Docs',
'Assigned KBs',
'Doc Category order',
'KB order', // Add additional term meta headers here
];
$terms = $this->get_terms($post_ids);
foreach ($terms as $term) {
$term_meta = '';
// Add term meta based on taxonomy
switch ($term->taxonomy) {
case 'doc_category':
$doc_category_knowledge_base = maybe_unserialize(get_term_meta($term->term_id, 'doc_category_knowledge_base', true));
if (is_array($doc_category_knowledge_base) && $doc_category_knowledge_base !== false) {
$doc_category_knowledge_base = implode(', ', array_filter($doc_category_knowledge_base));
} else {
$doc_category_knowledge_base = '';
}
$term_meta = [
'_docs_order' => get_term_meta($term->term_id, '_docs_order', true),
'doc_category_knowledge_base' => $doc_category_knowledge_base,
'doc_category_order' => get_term_meta($term->term_id, 'doc_category_order', true),
];
break;
case 'knowledge_base':
$term_meta = [
'kb_order' => get_term_meta($term->term_id, 'kb_order', true),
];
break;
}
// Add CSV row for term
$csv_data_terms[] = [
'Term',
$term->taxonomy,
$term->term_id,
$term->name,
$term->slug,
$term->term_group,
$term->description,
$term->parent,
isset($term_meta['_docs_order']) ? $term_meta['_docs_order'] : '',
isset($term_meta['doc_category_knowledge_base']) ? $term_meta['doc_category_knowledge_base'] : '',
isset($term_meta['doc_category_order']) ? $term_meta['doc_category_order'] : '',
isset($term_meta['kb_order']) ? $term_meta['kb_order'] : '',
];
}
return $csv_data_terms;
}
/**
* Return list of authors with posts.
*
* @param int[] $post_ids Optional. Array of post IDs to filter the query by.
*
* @return string
*/
private function authors_list( $post_ids ) {
$authors = [];
// Add CSV headers for terms
$authors[] = [
'Type',
'Author id',
'Author login',
'Author email',
'Author display name',
'Author first name',
'Author last name'
];
if ( ! empty( $post_ids ) ) {
$post_ids = array_map( 'absint', $post_ids );
$and = 'AND ID IN ( ' . implode( ', ', $post_ids ) . ')';
} else {
$and = '';
}
$authors_data = [];
$results = $this->wpdb->get_results( "SELECT DISTINCT post_author FROM {$this->wpdb->posts} WHERE post_status != 'auto-draft' $and" );// phpcs:ignore
foreach ( (array) $results as $r ) {
$authors_data[] = get_userdata( $r->post_author );
}
$authors_data = array_filter( $authors_data );
foreach ( $authors_data as $author ) {
$authors[] = [
'Author',
$author->ID,
$author->user_login,
$author->user_email,
$author->display_name,
$author->first_name,
$author->last_name
];
}
return $authors;
}
private function get_posts_csv_data($posts) {
$csv_data_posts = [];
// Add CSV headers for posts
$csv_data_posts[] = [
'Type',
'Docs ID',
'Docs author',
'Docs date',
'Docs date gmt',
'Docs title',
'Docs content',
'Docs excerpt',
'Docs status',
'Docs password',
'Docs slug',
'Docs modified date',
'Docs modified date gmt',
'Docs parent',
'Docs menu order',
'Docs mime type',
'Comment count',
'Doc Categories',
'Doc Tags',
'Knowledge Bases',
'Docs attachement url',
'Docs attachement ID'
];
foreach ($posts as $post) {
$attachment_id = get_post_thumbnail_id($post->ID);
$attachment_url = get_the_post_thumbnail_url($post->ID);
// Add CSV row for post
$csv_data_posts[] = [
'Docs',
$post->ID,
$post->post_author,
$post->post_date,
$post->post_date_gmt,
$post->post_title,
$post->post_content,
$post->post_excerpt,
$post->post_status,
$post->post_password,
$post->post_name,
$post->post_modified,
$post->post_modified_gmt,
$post->post_parent,
$post->menu_order,
$post->post_mime_type,
$post->comment_count,
$this->get_term_ids( $post->ID, 'doc_category' ),
$this->get_term_ids( $post->ID, 'doc_tag' ),
$this->get_term_ids( $post->ID, 'knowledge_base' ),
$attachment_url ? $attachment_url : '',
$attachment_id ? $attachment_id : ''
];
}
return $csv_data_posts;
}
public function get_term_ids($post_id, $taxonomy) {
$terms = get_the_terms($post_id, $taxonomy);
if ($terms && !is_wp_error($terms)) {
$term_ids = wp_list_pluck($terms, 'term_id');
return implode(', ', $term_ids);
}
return '';
}
private function generate_csv(array $data): string {
ob_start();
$output = fopen('php://output', 'w');
// Add CSV rows
foreach ($data as $row) {
fputcsv($output, $row);
}
fclose($output);
return ob_get_clean();
}
}