WPML compatibility tweaks

includes/class-sp-wpml.php

@@ -0,0 +1,66 @@
+<?php
+/**
+ * WPML Class
+ *
+ * The SportsPress WPML class handles all WPML-related localization hooks.
+ *
+ * @class 		SP_WPML
+ * @version		1.1.9
+ * @package		SportsPress/Classes
+ * @category	Class
+ * @author 		ThemeBoy
+ */
+
+if ( ! defined( 'ABSPATH' ) ) exit; // Exit if accessed directly
+
+if ( ! class_exists( 'SP_WPML' ) ) :
+
+/**
+ * SP_WPML Class
+ */
+class SP_WPML {
+
+	/**
+	 * Constructor.
+	 */
+	public function __construct() {
+		add_filter( 'the_title', array( $this, 'the_title' ), 5, 2 );
+		add_filter( 'post_type_link', array( $this, 'post_type_link' ), 5, 2 );
+	}
+
+	public static function the_title( $title, $id = null ) {
+		if ( self::can_localize( $id, get_post_type( $id ) ) ):
+			// Get translated post ID
+			$translated_id = icl_object_id( $id, 'any', false, ICL_LANGUAGE_CODE );
+			
+			if ( $translated_id ):
+				$post = get_post( $translated_id );
+				$title = $post->post_title;
+			endif;
+		endif;
+
+		return $title;
+	}
+
+	public static function post_type_link( $url, $post ) {
+		if ( self::can_localize( $post->ID, $post->post_type ) ):
+			// Get translated post ID
+			$translated_id = icl_object_id( $post->ID, 'any', false, ICL_LANGUAGE_CODE );
+
+			if ( $translated_id ):
+				$url .= '?lang=ja';
+				//$url = get_permalink( $translated_id );
+			endif;
+		endif;
+
+		return $url;
+	}
+
+	public static function can_localize( $id, $post_type ) {
+		return ( function_exists( 'icl_object_id' ) && is_sp_post_type( $post_type ) && $id != get_the_ID() );
+	}
+}
+
+endif;
+
+return new SP_WPML();