Remove hardcode table references

InternetSociety#21
paulvanbuuren · May 20, 2021 · 3c7f2ae · 3c7f2ae
1 parent a9b743f
commit 3c7f2ae
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/od/includes/hooks-setup.php b/od/includes/hooks-setup.php
@@ -66,7 +66,7 @@ function od_create_page( $sPageTitle, $iMenuOrder = NULL, $sPageContent = NULL )
 	// early exit - slug exists
 	if( od_slug_exists($sPageSlug) ) {
 		global $wpdb;
-		$oPost = $wpdb->get_row("SELECT post_name FROM od_posts WHERE post_name = '".$sPageSlug."'");
+		$oPost = $wpdb->get_row( "SELECT post_name FROM $wpdb->posts WHERE post_name = '" . sanitize_title( $sPageSlug ) . "'" );
 		// we've got ourselves a page
 		if( $oPost->post_type == 'page' ) { return $oPost->ID; }
 		// not a page, better create one then

diff --git a/od/includes/utils.php b/od/includes/utils.php
@@ -79,6 +79,9 @@ function od_get_id_by_slug( $object_slug ) {
  */
 function od_slug_exists( $slug ) {
 	global $wpdb;
-	if( $wpdb->get_row("SELECT post_name FROM od_posts WHERE post_name = '".$slug."'", 'ARRAY_A') ) { return TRUE; }
-	else { return FALSE; }
+	if ( $wpdb->get_row( "SELECT post_name FROM $wpdb->posts WHERE post_name = '" . sanitize_title( $slug ) . "'", 'ARRAY_A' ) ) {
+		return true;
+	} else {
+		return false;
+	}
 }