Changeset 489010

Timestamp:

01/12/2012 09:21:19 PM (14 years ago)

Author:

divinenephron

Message:

Now generates PDFs from posts and attaches it to the post.

File:

• latex-everything/trunk/article-to-latex.php (modified) (4 diffs)

latex-everything/trunk/article-to-latex.php

@@ -27 +27 @@
     var $html_to_latex_script;
 
+    var $post;
+
     var $html_file;
     var $latex_file;
@@ -32 +34 @@
     var $uploaded_file;
 
-
-    function __construct () {
+    var $error;
+
+
+    function __construct ( $post_id ) {
         $this->plugin_dir = plugin_dir_path(__FILE__);
         $this->html_to_latex_script = $this->plugin_dir . 'html2latex.pl';
+
+        $this->post = get_post( $post_id );
     }
 
@@ -46 +52 @@
         unlink( $this->html_file . '.log' );
     }
+
+    function create_pdf() {
+        $html_result = $this->make_html_file();
+        if ( is_wp_error( $html_result ) ) {
+            $this->record_error( $html_result );
+            return;
+        }
+    
+        $latex_result = $this->html_file_to_latex_file();
+        if ( is_wp_error( $latex_result ) ) {
+            $this->record_error( $latex_result );
+            return;
+        }
+    
+        $pdf_result = $this->latex_file_to_pdf_file();
+        if ( is_wp_error( $pdf_result ) ) {
+            $this->record_error( $pdf_result );
+            return;
+        }
+    
+        $attach_id = $this->attach_pdf_file();
+        if ( is_wp_error( $attach_id ) ) {
+            $this->record_error( $pdf_result );
+            return;
+        }
+        return;
+    }
     
     function make_html_file () {
-        $html = '<h1>Header</h1><p>Paragraph.</p>';
+        $html = $this->post->post_content;
+        $html = apply_filters('the_content', $html);
+        $html = str_replace(']]>', ']]&gt;', $html);
 
         if ( !$this->html_file = tempnam( sys_get_temp_dir(), 'a2l-' ) ) // Should fall back on system's temp dir if /tmp does not exist
@@ -96 +131 @@
 
     function attach_pdf_file () {
-        $upload_dir = wp_upload_dir();
-        $this->uploaded_file = $upload_dir['path'] . '/' . 'test' . '.pdf';
-    
+        $filename = 'a2l-postie-' . $this->post->ID . '.pdf';
+
+        $wp_filetype = wp_check_filetype( '.pdf' );
+        $attachment_data = array(
+            'post_mime_type' => $wp_filetype['type'],
+            'post_title' => $this->post->post_title,
+            'post_name' => 'pdf',
+            'post_content' => '',
+            'post_status' => 'inherit',
+        );
+
+        /* Check whether this post has an older file and attachment
+         * object we can update
+         */
+        $args = array( 'post_type' => 'attachment', 'numberposts' => -1, 'post_status' => null, 'post_parent' => $this->post->ID ); 
+        $attachments = get_posts($args);
+        if ($attachments) {
+            foreach ( $attachments as $attachment ) {
+                $attached_file = get_attached_file( $attachment->ID );
+                if ( basename( $attached_file ) == $filename ) {
+                    $this->uploaded_file = $attached_file;
+                    $attachment_data['ID'] = $attachment->ID;
+                }
+            }
+        }
+
+        // If it doesn't, find a location for a new file
+        if ( empty( $this->uploaded_file ) ) {
+            $upload_dir = wp_upload_dir();
+            $this->uploaded_file = $upload_dir['path'] . '/' . $filename;
+        }
+
+        
+        // Create the attachment
         copy( $this->pdf_file, $this->uploaded_file );
+        $attach_id = wp_insert_attachment( $attachment_data, $this->uploaded_file, $this->post->ID );
+        if ( $attach_id == 0 ) { // Attachment error
+            return WP_Error( 'wp_insert_attachment', 'Could not attach generated pdf' );
+        }
+        return $attach_id;
+    }
+
+    /* Adds a2l_error to the redirect query when the error is first encounterd.
+     * After the redirect it prints the errors to the top of the admin page.
+     */
+    function record_error ( $error ) {
+        if ( isset( $_GET['a2l_error'] ) ) {
+            add_action( 'admin_notices', array( $this, 'display_error' ) );
+        } else { // First time the error has been encountered.
+            add_action('redirect_post_location', array( $this, 'change_redirect_query' ), 99 );
+        }
+        $this->error = $error;
+    }
+    /* Add a query to the redirect after the post has been saved to remind the post edit
+     * screen to re-run the pdf create so that the error can be displayed.
+     */
+    function change_redirect_query ($location) {
+        return add_query_arg('a2l_error', $this->post->ID, $location);
+    }
+    /* Displays the error on the admin screen.
+     */
+    function display_error () {
+        printf( "<div class=\"error\"><p>Article to Latex error</p><p>%s</p><p>%s</p></div>\n", $this->error->get_error_code(), $this->error->get_error_message() );
     }
 }
 
-function random_test() {
-    $a2l = new Divinenephron_Article_To_Latex;
-
-    $html_result = $a2l->make_html_file();
-    if ( is_wp_error( $html_result ) ) {
-        print $html_result->get_error_message();
-    }
-
-    $latex_result = $a2l->html_file_to_latex_file();
-    if ( is_wp_error( $latex_result ) ) {
-        print $latex_result->get_error_message();
-    }
-
-    $pdf_result = $a2l->latex_file_to_pdf_file();
-    if ( is_wp_error( $pdf_result ) ) {
-        printf("<!--\n%s\n-->", $pdf_result->get_error_message());
-    }
-
-    $a2l->attach_pdf_file();
+/* Every time a post is saved run the converter.
+ */
+function a2l_save_post ( $post_id ) {
+    if ( get_post_type( $post_id ) == 'post' && !wp_is_post_revision( $post_id ) ) {
+        $a2l = new Divinenephron_Article_To_Latex( $post_id );
+        $a2l->create_pdf();
+    }
 }
-add_action('wp_head', 'random_test');
+add_action('save_post', 'a2l_save_post');
+
+/* If a2l_error is in the query, a previous post save created an error
+ * for a post (the post ID is its value). Thus we need to re-run the
+ * converter to figure out what the error was and display it.
+ */
+function a2l_show_errors () {
+    global $wp_query;
+    if ( isset( $_GET['a2l_error'] ) ) {
+        $post_id = $_GET[ 'a2l_error' ];
+        $a2l = new Divinenephron_Article_To_Latex( $post_id );
+        $a2l->create_pdf();
+    }
+}
+add_action( 'admin_head', 'a2l_show_errors' );
+