← All changes | modules/shortlinks.php +195 -0 16.2-beta → 16.3 View file →

		@@ -1,0 +1,195 @@
	1	+<?php
	2	+/**
	3	+ * Module Name: WP.me Shortlinks
	4	+ * Module Description: Share short, easy-to-remember links to your posts and pages.
	5	+ * Sort Order: 8
	6	+ * First Introduced: 1.1
	7	+ * Requires Connection: Yes
	8	+ * Auto Activate: No
	9	+ * Module Tags: Social
	10	+ * Feature: Writing
	11	+ * Additional Search Queries: shortlinks, wp.me
	12	+ *
	13	+ * @package automattic/jetpack
	14	+ */
	15	+
	16	+if ( ! defined( 'ABSPATH' ) ) {
	17	+ exit( 0 );
	18	+}
	19	+
	20	+add_filter( 'pre_get_shortlink', 'wpme_get_shortlink_handler', 1, 4 );
	21	+
	22	+if ( ! function_exists( 'wpme_dec2sixtwo' ) ) {
	23	+ /**
	24	+ * Converts number to base 62.
	25	+ *
	26	+ * @param int $num Number.
	27	+ *
	28	+ * @return string Value in base 62.
	29	+ */
	30	+ function wpme_dec2sixtwo( $num ) {
	31	+ $index = '0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ';
	32	+ $out = '';
	33	+
	34	+ if ( $num < 0 ) {
	35	+ $out = '-';
	36	+ $num = abs( $num );
	37	+ }
	38	+
	39	+ for ( $t = floor( log10( $num ) / log10( 62 ) ); $t >= 0; $t-- ) {
	40	+ $a = (int) floor( $num / pow( 62, $t ) );
	41	+ $out .= substr( $index, $a, 1 );
	42	+ $num -= $a * pow( 62, $t );
	43	+ }
	44	+
	45	+ return $out;
	46	+ }
	47	+}
	48	+
	49	+/**
	50	+ * Returns the WP.me shortlink.
	51	+ *
	52	+ * @param int $id Post ID, or 0 for the current post.
	53	+ * @param string $context The context for the link. One of 'post' or 'query'.
	54	+ * @param bool $allow_slugs Whether to allow post slugs in the shortlink.
	55	+ *
	56	+ * @return string
	57	+ */
	58	+function wpme_get_shortlink( $id = 0, $context = 'post', $allow_slugs = true ) {
	59	+ global $wp_query;
	60	+
	61	+ $blog_id = Jetpack_Options::get_option( 'id' );
	62	+
	63	+ if ( 'query' === $context ) {
	64	+ if ( is_singular() ) {
	65	+ $id = $wp_query->get_queried_object_id();
	66	+ $context = 'post';
	67	+ } elseif ( is_front_page() ) {
	68	+ $context = 'blog';
	69	+ } else {
	70	+ return '';
	71	+ }
	72	+ }
	73	+
	74	+ if ( 'blog' === $context ) {
	75	+ if ( empty( $id ) ) {
	76	+ $id = $blog_id;
	77	+ }
	78	+
	79	+ return 'https://wp.me/' . wpme_dec2sixtwo( $id );
	80	+ }
	81	+
	82	+ $post = get_post( $id );
	83	+
	84	+ if ( empty( $post ) ) {
	85	+ return '';
	86	+ }
	87	+
	88	+ $post_id = $post->ID;
	89	+ $type = '';
	90	+
	91	+ if ( $allow_slugs && 'publish' === $post->post_status && 'post' === $post->post_type && strlen( $post->post_name ) <= 8 && ! str_contains( $post->post_name, '%' )
	92	+ && ! str_contains( $post->post_name, '-' ) ) {
	93	+ $id = $post->post_name;
	94	+ $type = 's';
	95	+ } else {
	96	+ $id = wpme_dec2sixtwo( $post_id );
	97	+ if ( 'page' === $post->post_type ) {
	98	+ $type = 'P';
	99	+ } elseif ( 'post' === $post->post_type || post_type_supports( $post->post_type, 'shortlinks' ) ) {
	100	+ $type = 'p';
	101	+ } elseif ( 'attachment' === $post->post_type ) {
	102	+ $type = 'a';
	103	+ }
	104	+ }
	105	+
	106	+ if ( empty( $type ) ) {
	107	+ return '';
	108	+ }
	109	+
	110	+ return 'https://wp.me/' . $type . wpme_dec2sixtwo( $blog_id ) . '-' . $id;
	111	+}
	112	+
	113	+/**
	114	+ * Get the shortlink handler.
	115	+ *
	116	+ * Used with the Core pre_get_shortlink hook.
	117	+ *
	118	+ * @param string $shortlink Shortlink value from the action. Ignored.
	119	+ * @param int $id Post ID (0 for the current post).
	120	+ * @param string $context The context for the link. One of 'post' or 'query'.
	121	+ * @param bool $allow_slugs Whether to allow post slugs in the shortlink.
	122	+ *
	123	+ * @return string
	124	+ */
	125	+function wpme_get_shortlink_handler( $shortlink, $id, $context, $allow_slugs ) {
	126	+ return wpme_get_shortlink( $id, $context, $allow_slugs );
	127	+}
	128	+
	129	+/**
	130	+ * Add Shortlinks to the REST API responses.
	131	+ *
	132	+ * @since 6.9.0
	133	+ *
	134	+ * @action rest_api_init
	135	+ * @uses register_rest_field, wpme_rest_get_shortlink
	136	+ */
	137	+function wpme_rest_register_shortlinks() {
	138	+ // Post types that support shortlinks by default.
	139	+ $supported_post_types = array(
	140	+ 'attachment',
	141	+ 'page',
	142	+ 'post',
	143	+ );
	144	+
	145	+ // Add any CPT that may have declared support for shortlinks.
	146	+ foreach ( get_post_types() as $post_type ) {
	147	+ if (
	148	+ post_type_supports( $post_type, 'shortlinks' )
	149	+ && post_type_supports( $post_type, 'editor' )
	150	+ ) {
	151	+ $supported_post_types[] = $post_type;
	152	+ }
	153	+ }
	154	+
	155	+ register_rest_field(
	156	+ $supported_post_types,
	157	+ 'jetpack_shortlink',
	158	+ array(
	159	+ 'get_callback' => 'wpme_rest_get_shortlink',
	160	+ 'update_callback' => null,
	161	+ 'schema' => null,
	162	+ )
	163	+ );
	164	+}
	165	+
	166	+/**
	167	+ * Get the shortlink of a post.
	168	+ *
	169	+ * @since 6.9.0
	170	+ *
	171	+ * @param array $object Details of current post.
	172	+ *
	173	+ * @uses wpme_get_shortlink
	174	+ *
	175	+ * @return string
	176	+ */
	177	+function wpme_rest_get_shortlink( $object ) {
	178	+ $object_id = $object['id'] ?? 0;
	179	+ return wpme_get_shortlink( $object_id, array() );
	180	+}
	181	+
	182	+// Add shortlinks to the REST API Post response.
	183	+add_action( 'rest_api_init', 'wpme_rest_register_shortlinks' );
	184	+
	185	+/**
	186	+ * Set the Shortlink Gutenberg extension as available.
	187	+ */
	188	+function wpme_set_extension_available() {
	189	+ Jetpack_Gutenberg::set_extension_available( 'shortlinks' );
	190	+}
	191	+
	192	+add_action( 'init', 'wpme_set_extension_available' );
	193	+
	194	+require_once __DIR__ . '/shortlinks/abilities/class-shortlinks-abilities.php';
	195	+\Automattic\Jetpack\Plugin\Abilities\Shortlinks_Abilities::init();