PDO::ERRMODE_EXCEPTION,
PDO::ATTR_DEFAULT_FETCH_MODE => PDO::FETCH_ASSOC,
]);
}
function genCode($pdo, $len = 6) {
$alpha = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789';
do {
$code = '';
for ($i = 0; $i < $len; $i++) $code .= $alpha[random_int(0, strlen($alpha) - 1)];
} while ($pdo->query("SELECT COUNT(*) c FROM short_link WHERE code=" . $pdo->quote($code))->fetch()['c'] > 0);
return $code;
}
$pdo = null;
try { $pdo = db($cfg); } catch (Exception $e) { $dbError = '数据库连接失败，请检查 config.php 配置：' . $e->getMessage(); }
$shortened = null;
$error = null;
if ($_SERVER['REQUEST_METHOD'] === 'POST') {
$url = trim($_POST['url'] ?? '');
$alias = trim($_POST['alias'] ?? '');
if ($url === '' || !preg_match('#^https?://#i', $url)) {
$error = '请输入以 http(s):// 开头的链接';
} else {
try {
$code = '';
if ($alias !== '') {
if (!preg_match('#^[a-zA-Z0-9]{3,16}$#', $alias)) {
$error = '别名需为 3-16 位字母数字';
} else {
$exists = $pdo->query("SELECT COUNT(*) c FROM short_link WHERE code=" . $pdo->quote($alias))->fetch()['c'];
if ($exists > 0) $error = '该别名已被占用';
else $code = $alias;
}
} else {
$code = genCode($pdo);
}
if ($error === null) {
$st = $pdo->prepare("INSERT INTO short_link (code, url, alias) VALUES (?,?,?)");
$st->execute([$code, $url, $alias === '' ? null : $alias]);
$shortened = 'https://' . ($_SERVER['HTTP_HOST'] ?? 'dev.xjxbk.cn') . '/s/' . $code;
}
} catch (Exception $e) {
$error = '生成失败：' . $e->getMessage();
}
}
}
$recent = [];
if ($pdo) {
try {
$recent = $pdo->query("SELECT code, url, clicks, DATE_FORMAT(created_at,'%m-%d %H:%i') t FROM short_link ORDER BY id DESC LIMIT 10")->fetchAll();
} catch (Exception $e) {}
}
header('Content-Type: text/html; charset=utf-8');
?>
URL 短链接系统