短縮URLとQRコードを作るツール

CREATE TABLE sumidai_short_url (
  uri    VARCHAR(10)   NOT NULL,   -- 短縮コード（例: Ab3x）
  domain VARCHAR(255)  NOT NULL,   -- 短縮URLのドメイン
  url    TEXT          NOT NULL,   -- 元のURL
  title  VARCHAR(255)  DEFAULT '', -- 元のページのタイトル
  time   DATETIME      NOT NULL,   -- 作成日時
  ip     VARCHAR(45)   DEFAULT '', -- 作成した人の IP アドレス
  PRIMARY KEY (uri)
);

<script>
$(function () {
    $('#make').on('click', function () {
        var url = $('input[name="url"]').val();
        $('#qrcode').empty();

        $.ajax({
            url: 'create.php',          // 同じフォルダの create.php に送る
            type: 'POST',
            data: { url: url },
            timeout: 10000,
            dataType: 'html'
        })
        .done(function (data) {
            $('#short').html(data);     // 返ってきた HTML をそのまま表示
            var link = $('#link').val();
            if (link) {                 // 短縮URLができていたら QR コードを作る
                new QRCode(document.getElementById('qrcode'), { text: link, width: 160, height: 160 });
                $('input[name="url"]').val('');
            }
        })
        .fail(function () {
            $('#short').html('<div class="alert alert-danger">通信に失敗しました。</div>');
        });
    });

    // コピーボタン（あとから追加された要素なので document に対して設定する）
    $(document).on('click', '#copy', function () {
        var text = $('#link').val();
        var done = function () { $('#complete').text('コピーしました'); };
        var oldWay = function () {          // 古いブラウザ用
            $('#link').trigger('select');
            document.execCommand('copy');
            done();
        };
        if (navigator.clipboard && window.isSecureContext) {
            navigator.clipboard.writeText(text).then(done, oldWay);
        } else {
            oldWay();
        }
    });
});
</script>

// 入力されたURL（保存は「そのまま」、画面に出すときにエスケープする）
$url = trim((string)($_POST['url'] ?? ''));

// http:// か https:// で始まるURLだけ受け付ける
if (!preg_match('#^https?://#i', $url) || filter_var($url, FILTER_VALIDATE_URL) === false || strlen($url) > 2000) {
    echo '<div class="alert alert-danger" role="alert">入力された文字がURLの形式ではありません。</div>';
    exit;
}

$title = getPageTitle($url);
$ip    = $_SERVER['REMOTE_ADDR'] ?? '';
$host  = $_SERVER['HTTP_HOST'] ?? 'localhost';

// 4文字のコードを作り、使われていたら作り直す
do {
    $uri = makeRandStr(4);
} while (confirmation($uri, $db));

$stmt = $db->prepare("INSERT INTO sumidai_short_url (uri, domain, url, title, time, ip)
                      VALUES (:uri, :domain, :url, :title, now(), :ip)");
$stmt->execute([':uri' => $uri, ':domain' => $host, ':url' => $url, ':title' => $title, ':ip' => $ip]);

// 公開デモなので、古いものから削除して件数を一定に保つ
demo_prune_table($db, 'sumidai_short_url', 'time', DEMO_MAX_URLS);

// 短縮URL（このサイトの go.php?c=コード）
$scheme = (!empty($_SERVER['HTTPS']) && $_SERVER['HTTPS'] !== 'off') ? 'https' : 'http';
$short  = $scheme . '://' . $host . url('php/short-url/go.php') . '?c=' . $uri;

// ランダムな英数字を作る  makeRandStr(文字数)
function makeRandStr($length) {
    $str = array_merge(range('a', 'z'), range('0', '9'), range('A', 'Z'));
    $r_str = '';
    for ($i = 0; $i < $length; $i++) {
        $r_str .= $str[random_int(0, count($str) - 1)];   // 推測されにくい乱数を使う
    }
    return $r_str;
}

// 外部から取りに行ってよいURLか確認する
// （サーバーの内側＝社内ネットワークなどにアクセスさせないための対策）
function isPublicUrl($url) {
    $host = parse_url($url, PHP_URL_HOST);
    if (!is_string($host) || $host === '') {
        return false;
    }
    $ip = gethostbyname($host);
    return filter_var($ip, FILTER_VALIDATE_IP, FILTER_FLAG_NO_PRIV_RANGE | FILTER_FLAG_NO_RES_RANGE) !== false;
}

// ページのタイトルを取得する  getPageTitle(URL)
function getPageTitle($url){
    if (!isPublicUrl($url)) {
        return '';
    }
    $ch = curl_init();
    curl_setopt($ch, CURLOPT_URL, $url);
    curl_setopt($ch, CURLOPT_RETURNTRANSFER, true);     // 結果を画面に出さず、文字列で受け取る
    curl_setopt($ch, CURLOPT_FOLLOWLOCATION, false);    // 転送先は追わない
    curl_setopt($ch, CURLOPT_TIMEOUT, 5);               // 5秒で諦める
    $html = curl_exec($ch);
    curl_close($ch);
    if (!is_string($html) || $html === '') {            // 取得に失敗したとき（false）は空文字
        return '';
    }
    $html = mb_convert_encoding($html, 'UTF-8', 'ASCII,JIS,UTF-8,CP51932,SJIS-win');
    return preg_match('@<title>([^<]++)</title>@i', $html, $m) ? trim($m[1]) : '';
}

<?php
// 短縮URLを開いたときに、元のURLへ転送する  例: go.php?c=Ab3x
require_once 'db.php';

$code = (string)($_GET['c'] ?? '');

$url = false;
if (preg_match('/^[A-Za-z0-9]{4}$/', $code)) {          // コードは英数字4文字だけ
    $db = db_connect();
    $stmt = $db->prepare("SELECT url FROM sumidai_short_url WHERE uri = :uri");
    $stmt->bindValue(':uri', $code);
    $stmt->execute();
    $url = $stmt->fetchColumn();
}

// 見つかって、http/https のURLなら転送する
if (is_string($url) && preg_match('#^https?://#i', $url)) {
    header('Location: ' . $url, true, 302);
    exit;
}

// 見つからなかったとき
http_response_code(404);