Allow associations between posts and sessions with the Posts 2 Posts …

…plugin

functions.php

@@ -16,17 +16,18 @@ class ona11
 
 	function __construct() {
 
+		$this->session = new ona11_session();
+
 		add_action( 'after_setup_theme', array( &$this, 'enqueue_resources' ) );
 		add_action( 'after_setup_theme', array( &$this, 'register_custom_taxonomies' ) );
+		add_action( 'after_setup_theme', array( &$this, 'associate_post_types' ) );
 
 		if ( !is_admin() ) {
 			require_once( 'php/template_tags.php' );
 			add_action( 'wp_head', 'ona11_head_title' );
 			add_action( 'wp_head', array( &$this, 'wp_head' ) );			
 		}
 
-		$this->session = new ona11_session();
-
 	}
 
 	/**
@@ -69,6 +70,17 @@ function register_custom_taxonomies() {
 
 	}
 
+	function associate_post_types() {
+		if ( !function_exists( 'p2p_register_connection_type' ) )
+	        return;
+
+		// Writers should be able to associate posts with sessions
+	    p2p_register_connection_type( array( 
+			'from' => 'post',
+	        'to' => 'ona11_session'
+	    ) );
+	}
+
 	/**
 	 * Other resources to add to the head
 	 */

php/class.ona11_session.php

@@ -15,7 +15,7 @@ class ona11_session
 	 */
 	function __construct() {
 
-		add_action( 'init', array( &$this, 'create_post_type' ) );
+		add_action( 'after_setup_theme', array( &$this, 'create_post_type' ) );
 
 		// Load necessary scripts and stylesheets
 		add_action( 'admin_enqueue_scripts', array( &$this, 'add_admin_resources' ) );