86 lines
2.8 KiB
PHP
86 lines
2.8 KiB
PHP
<?php
|
|
/*
|
|
Plugin Name: YOURLS HTML Sitemap Generator
|
|
Plugin URI: https://github.com/SophiaAtkinson/yourls-html-sitemap
|
|
Description: Generates a sitemap.html file with all of your short URLs
|
|
Version: 1.0
|
|
Author: Sophia Atkinson
|
|
Author URI: https://sophia.wtf
|
|
*/
|
|
|
|
// No direct call
|
|
if (!defined('YOURLS_ABSPATH')) die();
|
|
|
|
// Include YOURLS loader
|
|
require_once YOURLS_ABSPATH . '/includes/load-yourls.php';
|
|
|
|
// Hook to add the sitemap action
|
|
yourls_add_action('pre_html_head', 'generate_html_sitemap');
|
|
|
|
// Function to generate the sitemap.html file
|
|
function generate_html_sitemap() {
|
|
global $ydb;
|
|
|
|
try {
|
|
// Initialize PDO connection
|
|
$pdo = new PDO('mysql:host=' . YOURLS_DB_HOST . ';dbname=' . YOURLS_DB_NAME, YOURLS_DB_USER, YOURLS_DB_PASS);
|
|
|
|
// Set PDO to throw exceptions
|
|
$pdo->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION);
|
|
|
|
// Start building the HTML string
|
|
$html = '<!DOCTYPE html>';
|
|
$html .= '<html>';
|
|
$html .= '<head>';
|
|
$html .= '<title>YOURLS HTML Sitemap</title>';
|
|
$html .= '</head>';
|
|
$html .= '<body>';
|
|
$html .= '<h1>YOURLS HTML Sitemap</h1>';
|
|
$html .= '<ul>';
|
|
|
|
// Check if the 'private' column exists in the YOURLS database table
|
|
$table_name = YOURLS_DB_TABLE_URL;
|
|
$stmt = $pdo->query("DESCRIBE `$table_name`");
|
|
$private_column_exists = false;
|
|
while ($row = $stmt->fetch(PDO::FETCH_ASSOC)) {
|
|
if ($row['Field'] === 'private') {
|
|
$private_column_exists = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if ($private_column_exists) {
|
|
// Retrieve all public short URLs
|
|
$stmt = $pdo->prepare("SELECT `keyword`, `url`, `timestamp` FROM `$table_name` WHERE `private` = '0'");
|
|
$stmt->execute();
|
|
} else {
|
|
// Retrieve all short URLs (assuming all are public)
|
|
$stmt = $pdo->prepare("SELECT `keyword`, `url`, `timestamp` FROM `$table_name`");
|
|
$stmt->execute();
|
|
}
|
|
|
|
// Loop through each link and add it to the HTML sitemap
|
|
while ($row = $stmt->fetch(PDO::FETCH_ASSOC)) {
|
|
$shorturl = yourls_link($row['keyword']);
|
|
$longurl = htmlspecialchars($row['url']);
|
|
$timestamp = date('c', strtotime($row['timestamp']));
|
|
|
|
$html .= '<li>';
|
|
$html .= '<a href="' . $shorturl . '">' . $shorturl . '</a>';
|
|
$html .= ' - ' . $longurl . ' - Last Modified: ' . $timestamp;
|
|
$html .= '</li>';
|
|
}
|
|
|
|
// Close the HTML
|
|
$html .= '</ul>';
|
|
$html .= '</body>';
|
|
$html .= '</html>';
|
|
|
|
// Save the HTML to sitemap.html file
|
|
file_put_contents(YOURLS_ABSPATH . '/sitemap.html', $html);
|
|
} catch (PDOException $e) {
|
|
// Handle PDO exceptions
|
|
yourls_die('Error: ' . $e->getMessage());
|
|
}
|
|
}
|