learndash-import.php

<?php
/*
Plugin Name: LearnDash JSON Import
Plugin URI: https://www.corduroybeach.com/
Description: A plugin for importing Courses / Quizzes / Questions and other things through the use of JSON
Version: 0.5.0
Author: Brandon Tassone
Author URI: https://www.corduroybeach.com/
*/

require_once("vendor/autoload.php");

global $wpdb;

// TODO: Debugging tool, remove Kint for 1.0.0
\Kint::$maxLevels = 0;

// Constants - Post Types
define('COURSES_POST_TYPE', 'sfwd-courses');
define('QUIZ_POST_TYPE', 'sfwd-quiz');

// Constants - Database Table Names
define('QUIZ_MASTER_TABLE', $wpdb->prefix . 'wp_pro_quiz_master');
define('QUIZ_QUESTION_TABLE', $wpdb->prefix . 'wp_pro_quiz_question');
define('QUIZ_PREREQ_TABLE', $wpdb->prefix . 'wp_pro_quiz_prerequisite');

define('WP_POSTMETA_TABLE', $wpdb->prefix . 'postmeta');
define('WP_POSTS_TABLE', $wpdb->prefix . 'posts');

// Register Filters
add_filter('upload_mimes', 'learndash_import_add_json_mime', 1, 1);

// Register Actions
add_action('admin_menu', 'learndash_import_menu');
add_action('admin_enqueue_scripts', 'learndash_import_styles');
add_action('admin_enqueue_scripts', 'learndash_import_javascript');

// Filter Functions
function learndash_import_add_json_mime($mime_types) {
    $mime_types['json'] = 'application/json';

    return $mime_types;
}

// Action Functions
function learndash_import_menu() {
    add_menu_page('LearnDash Import', 'LearnDash Import', 'manage_options', 'learndash-import', 'learndash_import_menu_page');
}

function learndash_import_styles() {
    wp_register_style('learndash-import-main-style', plugin_dir_url(__FILE__) . "learndash-import.css");

    wp_enqueue_style('learndash-import-main-style');
}

function learndash_import_javascript() {
    wp_register_script('learndash-import-main-script', plugin_dir_url(__FILE__) . "learndash-import.js", array(), '', true);

    wp_enqueue_media();
    wp_enqueue_script('learndash-import-main-script');
}

// Admin Menu Functions
function learndash_import_menu_page() {
    $run = isset($_GET['run']) ? $_GET['run'] : '';
    $delete = isset($_GET['delete']) ? $_GET['delete'] : '';
    $import_url = isset($_POST['import_url']) ? $_POST['import_url'] : '';

    $mock_json_structure = file_get_contents(plugin_dir_url(__FILE__) . "mock-json-structure.json");
    ?>
    <form id="hidden-submit-form" method="post">
        <input id="hidden-url-field" type="hidden" name="import_url" value="" />
    </form>
    <div class="wrap learndash-import-wrap">
        <div class="main-action-wrap">
            <button id="run-import">Upload & Import</button>
            <button id="delete-all-data">Delete All LearnDash Data</button>
            <button id="go-back-to-main">Home</button>
        </div>
        <?php

        if($run && $import_url) {

            $json_file = file_get_contents($import_url);
            $data = json_decode($json_file);

            ?>
            <div class="heading-wrap">
                <div class="title-wrap">
                    <h1 class="heading-title">Importing Data</h1>
                </div>
            </div>
            <div class="courses-import-container">
                <?php

                foreach($data as $course):
                    $course_id = $course->course_id;
                    $course_title = $course->course_title;

                    $wp_course_id = learndash_import_create_course($course_id, $course_title);

                    $quiz_count = 0;
                    $question_count = 0;
                    $answer_count = 0;
                    ?>
                    <div class="course-item">
                        <?php
                        $quiz_prereq_info = array();

                        foreach($course->quizzes as $quiz):
                            $quiz_id = $quiz->quiz_id;
                            $quiz_title = $quiz->quiz_title;
                            $prereq_quiz_id = $quiz->prereq_quiz_id ?: 'NULL';

                            $prereq_info = learndash_import_create_quiz($quiz_id, $quiz_title, $wp_course_id, $prereq_quiz_id);
                            $quiz_prereq_info[] = $prereq_info;

                            foreach($quiz->questions as $question):
                                $question_text = $question->question_text;
                                $possible_answers = $question->possible_answers;

                                learndash_import_create_question($prereq_info["quiz_id_master"], $question_text, $possible_answers);

                                $answer_count += count($possible_answers);
                                $question_count++;
                            endforeach;

                            $quiz_count++;
                        endforeach;

                        // Update the recently imported prerequisites
                        learndash_import_set_prerequisites($quiz_prereq_info);
                        ?>
                        <div class="title-row">
                            <label>Course: </label>
                            <span><?php echo $course_title; ?></span>
                        </div>
                        <div class="information-row">
                            <div>
                                <label>Number of quizzes imported</label>
                                <span><?php echo $quiz_count; ?></span>
                            </div>
                            <div>
                                <label>Number of questions imported</label>
                                <span><?php echo $question_count; ?></span>
                            </div>
                            <div>
                                <label>Number of answers imported</label>
                                <span><?php echo $answer_count; ?></span>
                            </div>
                        </div>
                    </div>
                    <?php
                endforeach;
                ?>
            </div>
            <?php
        }

        if($delete) {
            ?>
            <div class="title-wrap delete-data">
                <h1 class="heading-title">Removing All LearnDash Data</h1>
            </div>
            <?php
            $passed = learndash_import_delete_all_data();
            $passed_text = $passed ? "Deleted everything successfully" : "Failed deleting a table somewhere.";
            $passed_class = $passed ? "success" : "failure";
            ?>
            <div class="deleted-tables-status <?php echo $passed_class; ?>">
                <?php echo $passed_text; ?>
            </div>
            <?php
        }

        if(!$run && !$delete) {
            ?>
            <div class="title-wrap description">
                <h1 class="heading-title">LearnDash Import</h1>
            </div>
            <div style="margin-bottom: 30px;">
                <p>
                    LearnDash Import is a JSON importer. Currently there are 2 supported actions Run Import which will run
                    the JSON import and Delete All LearnDash data. Best used when its a brand new site and you messed up
                    the import somehow and need to start again
                </p>
                <p>
                    Listed below is the current JSON template. More options will be added in the future
                </p>
            </div>
            <div>
                <h3>Example JSON Import Structure</h3>
                <pre class="json-structure"><?php echo $mock_json_structure; ?></pre>
            </div>
            <?php
        }

        ?>
    </div>
    <?php
}

// Program Functions
function learndash_import_create_course($course_id, $course_title) {

    // Create the course
    $wp_course_id = wp_insert_post(array(
        "post_title"    => $course_title,
        "post_type"     => COURSES_POST_TYPE,
        "post_status"   => "publish"
    ));

    // Add the associated import id from the json for later reference
    add_post_meta($wp_course_id, "imported_course_id", $course_id, true);

    return $wp_course_id;
}

function learndash_import_create_quiz($quiz_id, $quiz_title, $course_id, $prereq_quiz_id) {
    global $wpdb;

    $quiz_master_inserted_ids = array();

    // Create the quiz
    $wp_quiz_id = wp_insert_post(array(
        "post_title"     => $quiz_title,
        "post_type"     => QUIZ_POST_TYPE,
        "post_status"   => "publish"
    ));

    // Add the associated import id from the json for later reference
    add_post_meta($wp_quiz_id, "imported_quiz_id", $quiz_id, true);
    $quiz_insert_arr = array(
        "name"      => $quiz_title,
        "text"      => ""
    );

    $wpdb->insert(QUIZ_MASTER_TABLE, $quiz_insert_arr);
    $quiz_id_master = $wpdb->insert_id;

    add_post_meta($wp_quiz_id, "course_id", $course_id, true);
    add_post_meta($wp_quiz_id, "lesson_id", 0, true);
    add_post_meta($wp_quiz_id, "quiz_pro_id", $quiz_id_master, true);
    add_post_meta($wp_quiz_id, "quiz_pro_id_$quiz_id_master", $quiz_id_master, true);
    add_post_meta($wp_quiz_id, "_sfwd-quiz", array(
        "sfwd-quiz_course"              => $course_id,
        "sfwd-quiz_quiz_pro"            => $quiz_id_master,
        "sfwd-quiz_repeats"             => 0,
        "sfwd-quiz_threshold"           => 0.7,
        "sfwd-quiz_passingpercentage"   => 70,
        "sfwd-quiz_lesson"              => 0,
        "sfwd-quiz_certificate"         => 0
    ), true);

    return array(
        "wp_quiz_id"        => $wp_quiz_id,
        "quiz_id_master"    => $quiz_id_master,
        "quiz_uid"          => $quiz_id,
        "prereq_quiz_uid"   => $prereq_quiz_id
    );
}

function learndash_import_create_question($quiz_master_id, $question_text, $possible_answers){
    global $wpdb;

    $sort_next = $wpdb->get_results($wpdb->prepare("SELECT count(quiz_id) AS sort_next FROM wp_wp_pro_quiz_question WHERE quiz_id = %d", array($quiz_master_id)));
    $sort_next = $sort_next[0]->sort_next;
    $answer_types = array();

    foreach($possible_answers as $possible_answer) {
        $answer_id = $possible_answer->answer_id;
        $answer_text = $possible_answer->answer_text;
        $correct = $possible_answer->correct;

        $answer_type = new WpProQuiz_Model_AnswerTypes();
        $answer_type->setAnswer($answer_text);
        $answer_type->setCorrect($correct);

        $answer_types[] = $answer_type;
    }

    $answer_types = serialize($answer_types);

    $wpdb->insert(QUIZ_QUESTION_TABLE, array(
        "quiz_id"                               => $quiz_master_id,
        "online"                                => 1,
        "sort"                                  => intval($sort_next) + 1,
        "points"                                => 1,
        "title"                                 => "Question #" . (intval($sort_next) + 1),
        "question"                              => $question_text,
        "correct_same_text"                     => 0,
        "tip_enabled"                           => 0,
        "answer_type"                           => "multiple",
        "show_points_in_box"                    => 0,
        "answer_points_activated"               => 0,
        "answer_data"                           => $answer_types,
        "category_id"                           => 0,
        "answer_points_diff_modus_activated"    => 0,
        "disable_correct"                       => 0,
        "matrix_sort_answer_criteria_width"     => 0
    ));
}

function learndash_import_set_prerequisites($quiz_prereq_info) {
    global $wpdb;

    foreach($quiz_prereq_info as $prereq_info) {
        if($prereq_info["prereq_quiz_uid"] != 'NULL') {
            $wpdb->query($wpdb->prepare("UPDATE " . QUIZ_MASTER_TABLE . " SET prerequisite=%d WHERE id=%d", array(1, intval($prereq_info["quiz_id_master"]))));
            $prereq_master_id = learndash_import_find_prereq_quiz_master_id($prereq_info["prereq_quiz_uid"]);

            $wpdb->insert(QUIZ_PREREQ_TABLE, array(
                "prerequisite_quiz_id"       => intval($prereq_info["quiz_id_master"]),
                "quiz_id"                   => $prereq_master_id
            ));
        }
    }
}

function learndash_import_find_prereq_quiz_master_id($prereq_quiz_uid){
    global $wpdb;

    $query = "SELECT meta_value FROM wp_postmeta WHERE post_id IN (
                SELECT post_id FROM wp_postmeta WHERE meta_value = %s
              ) AND meta_key = 'quiz_pro_id'";

    $master_id = $wpdb->get_results($wpdb->prepare($query, array($prereq_quiz_uid)));
    $master_id = $master_id[0]->meta_value;

    return $master_id;
}

function learndash_import_delete_all_data() {
    global $wpdb;

    $passed = true;
    $tables = array(WP_POSTS_TABLE, QUIZ_MASTER_TABLE, QUIZ_QUESTION_TABLE, QUIZ_PREREQ_TABLE);
    $wp_tables = array(WP_POSTS_TABLE);

    echo "<pre class='rows-removed-container'>";

    foreach($tables as $table) {

        if(in_array($table, $wp_tables)) {
            $query = array(
                "posts_per_page"        => -1,
                "post_type"             => array(QUIZ_POST_TYPE, COURSES_POST_TYPE)
            );
            $posts = get_posts($query);

            foreach($posts as $post) {
                wp_delete_post($post->ID);
            }

            $passed = count($posts);
        } else {
            $passed = $wpdb->query($wpdb->prepare("DELETE FROM $table", array()));
        }

        echo "Table: $table | Rows Deleted: $passed<br />";

        if(!is_numeric($passed) && !$passed) {
            echo "</pre>";
            return false;
        }
    }

    echo "</pre>";

    return true;
}