sigmasternchen/drnk.me
1
0
Fork 0
mirror of https://github.com/sigmasternchen/drnk.me synced 2025-03-15 01:38:55 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: Add return existing slug if available

Browse source
This commit is contained in:
overflowerror 2023-12-03 13:36:47 +01:00
parent 96ca76cd93
commit f9a1419608
2 changed files with 33 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
controllers/manage/POST.php
View file

@ -86,6 +86,10 @@ function generateSlug(URLs $repository) {
return $candidates[0];
}
function generateAccessKey(string $slug, string $url) {
return sha1($url . "-" . $url . "-" . microtime() . "-" . rand());
}
return function (array &$context) {
$url = $_POST["url"] ?? "";
$url = validateInput($url);
@ -94,19 +98,25 @@ return function (array &$context) {
}
$repository = $context[REPOSITORIES]->urls;
$slug = generateSlug($repository);
$accessKey = sha1($url . "-" . $url . "-" . microtime() . "-" . rand());
$result = $repository->getByUrl($url);
if ($result) {
// don't leak existing access key
$result->accessKey = "";
} else {
$slug = generateSlug($repository);
$accessKey = generateAccessKey($slug, $url);
$result = $context[REPOSITORIES]->urls->add(new URL(
$slug,
$url,
$accessKey
));
$result = $repository->add(new URL(
$slug,
$url,
$accessKey
));
}
$data = [
"url" => "https://drnk.me/$slug",
"accessKey" => $accessKey,
"url" => "https://drnk.me/" . $result->slug,
"accessKey" => $result->accessKey,
];
require(ROOT . "/templates/pages/components/creation-successful.php");
};

14
persistence/repositories/URLs.php
View file

@ -71,4 +71,18 @@ class URLs {
return $this->entityFromRow($statement->fetch());
}
}
public function getByUrl(string $url) {
$statement = $this->connection->prepare(<<<EOF
SELECT * FROM `$this->table`
WHERE `url` = ?
EOF);
$statement->execute([$url]);
if ($statement->rowCount() == 0) {
return null;
} else {
return $this->entityFromRow($statement->fetch());
}
}
}