S2OJ/web/app/libs/uoj-utility-lib.php

306 lines
6.7 KiB
PHP
Raw Normal View History

2016-07-18 16:39:37 +00:00
<?php
function mergeConfig(&$config, $default_config) {
2020-06-25 12:41:16 +00:00
foreach ($default_config as $key => $val) {
2016-07-18 16:39:37 +00:00
if (!isset($config[$key])) {
$config[$key] = $val;
} elseif (is_array($config[$key])) {
mergeConfig($config[$key], $val);
}
}
}
2022-11-06 02:26:21 +00:00
function is_assoc($arr) {
if (!is_array($arr)) {
return false;
}
foreach (array_keys($arr) as $key) {
if (!is_int($key)) {
return true;
}
}
return false;
}
2016-07-18 16:39:37 +00:00
function strStartWith($str, $pre) {
return substr($str, 0, strlen($pre)) === $pre;
}
function strEndWith($str, $suf) {
return substr($str, -strlen($suf)) === $suf;
}
function strOmit($str, $len) {
if (strlen($str) <= $len + 3) {
return $str;
} else {
return substr($str, 0, $len) . '...';
}
}
function uojTextEncode($str, $config = array()) {
mergeConfig($config, [
'allow_CR' => false,
'html_escape' => false
]);
2022-11-06 02:26:21 +00:00
2016-07-18 16:39:37 +00:00
$allow = array();
for ($c = 32; $c <= 126; $c++) {
$allow[chr($c)] = true;
}
$allow["\n"] = true;
$allow[" "] = true;
$allow["\t"] = true;
2022-11-06 02:26:21 +00:00
2016-07-18 16:39:37 +00:00
if ($config['allow_CR']) {
$allow["\r"] = true;
}
2022-11-06 02:26:21 +00:00
2016-07-18 16:39:37 +00:00
$len = strlen($str);
$ok = true;
for ($i = 0; $i < $len; $i++) {
$c = $str[$i];
if (!isset($allow[$c])) {
$ok = false;
}
}
if ($ok && mb_check_encoding($str, 'utf-8')) {
if (!$config['html_escape']) {
return $str;
} else {
return HTML::escape($str);
}
} else {
$len = strlen($str);
$res = '';
$i = 0;
while ($i < $len) {
$c = $str[$i];
if (ord($c) < 128) {
if (isset($allow[$c])) {
if ($config['html_escape']) {
if ($c == '&') {
$res .= '&amp;';
2020-06-25 12:41:16 +00:00
} elseif ($c == '"') {
2016-07-18 16:39:37 +00:00
$res .= '&quot;';
2020-06-25 12:41:16 +00:00
} elseif ($c == '<') {
2016-07-18 16:39:37 +00:00
$res .= '&lt;';
2020-06-25 12:41:16 +00:00
} elseif ($c == '>') {
2016-07-18 16:39:37 +00:00
$res .= '&gt;';
} else {
$res .= $c;
}
} else {
$res .= $c;
}
} else {
$res .= '<b>\x' . bin2hex($c) . '</b>';
}
$i++;
} else {
$ok = false;
$cur = $c;
for ($j = $i + 1; $j < $i + 4 && $j < $len; $j++) {
$cur .= $str[$j];
if (mb_check_encoding($cur, 'utf-8')) {
$ok = true;
break;
}
}
if ($ok) {
$res .= $cur;
$i = $j + 1;
} else {
$res .= '<b>\x' . bin2hex($c) . '</b>';
$i++;
}
}
}
return $res;
}
}
2020-06-25 12:41:16 +00:00
function base64url_encode($data) {
return rtrim(strtr(base64_encode($data), '+/', '-_'), '=');
2016-07-18 16:39:37 +00:00
}
2020-06-25 12:41:16 +00:00
function base64url_decode($data) {
return base64_decode(str_pad(strtr($data, '-_', '+/'), strlen($data) % 4, '=', STR_PAD_RIGHT));
2016-07-18 16:39:37 +00:00
}
function blog_name_encode($name) {
$name = str_replace('-', '_', $name);
if (!strStartWith($name, '_') && !strEndWith($name, '_')) {
$name = str_replace('_', '-', $name);
}
$name = strtolower($name);
return $name;
}
function blog_name_decode($name) {
$name = str_replace('-', '_', $name);
$name = strtolower($name);
return $name;
}
2022-11-06 02:26:21 +00:00
function camelize($str, $delimiters = '-_') {
$str = ucwords($str, $delimiters);
foreach (str_split($delimiters) as $c) {
$str = str_replace($c, '', $str);
}
return $str;
}
2016-07-18 16:39:37 +00:00
function isSuperUser($user) {
return $user != null && $user['usergroup'] == 'S';
}
2022-11-06 02:26:21 +00:00
function isTmpUser($user) {
return $user != null && $user['usergroup'] == 'T';
}
2023-02-15 08:46:47 +00:00
function isBannedUser($user) {
return $user != null && $user['usergroup'] == 'B';
}
2016-07-18 16:39:37 +00:00
function getProblemExtraConfig($problem) {
$extra_config = json_decode($problem['extra_config'], true);
2022-11-06 02:26:21 +00:00
2016-07-18 16:39:37 +00:00
$default_extra_config = array(
'view_content_type' => 'ALL',
'view_all_details_type' => 'ALL',
2022-09-28 11:48:49 +00:00
'view_details_type' => 'ALL',
'view_solution_type' => 'ALL',
'submit_solution_type' => 'ALL_AFTER_AC',
2016-07-18 16:39:37 +00:00
);
2022-11-06 02:26:21 +00:00
2016-07-18 16:39:37 +00:00
mergeConfig($extra_config, $default_extra_config);
2022-11-06 02:26:21 +00:00
2016-07-18 16:39:37 +00:00
return $extra_config;
}
function getProblemSubmissionRequirement($problem) {
return json_decode($problem['submission_requirement'], true);
}
function getProblemCustomTestRequirement($problem) {
$extra_config = json_decode($problem['extra_config'], true);
if (isset($extra_config['custom_test_requirement'])) {
return $extra_config['custom_test_requirement'];
} else {
$answer = array(
'name' => 'answer',
'type' => 'source code',
'file_name' => 'answer.code'
);
foreach (getProblemSubmissionRequirement($problem) as $req) {
if ($req['name'] == 'answer' && $req['type'] == 'source code' && isset($req['languages'])) {
$answer['languages'] = $req['languages'];
}
}
return array(
$answer,
array(
'name' => 'input',
'type' => 'text',
'file_name' => 'input.txt'
)
);
}
}
function sendSystemMsg($username, $title, $content) {
2022-11-06 02:26:21 +00:00
DB::insert([
2023-01-22 09:00:12 +00:00
"insert into user_system_msg",
"(receiver, title, content, send_time)",
"values", DB::tuple([$username, $title, $content, DB::now()])
]);
2023-02-13 12:29:32 +00:00
sendEmail($username, $title, $content);
}
2023-02-13 12:32:00 +00:00
function sendEmail($username, $title, $content, $priority = 0) {
2023-02-13 12:29:32 +00:00
DB::insert([
"insert into emails",
2023-02-13 12:32:00 +00:00
"(receiver, subject, content, created_at, priority)",
"values", DB::tuple([$username, $title, $content, DB::now(), $priority])
2023-02-13 12:29:32 +00:00
]);
2022-11-06 02:26:21 +00:00
}
function retry_loop(callable $f, $retry = 5, $ms = 10) {
for ($i = 0; $i < $retry; $i++) {
$ret = $f();
if ($ret !== false) {
return $ret;
}
usleep($ms * 1000);
}
return $ret;
2016-07-18 16:39:37 +00:00
}
2023-01-22 09:00:12 +00:00
function getAbsoluteUrl($relativeUrl, $baseUrl) {
// if already absolute URL
if (parse_url($relativeUrl, PHP_URL_SCHEME) !== null) {
return $relativeUrl;
}
// queries and anchors
if ($relativeUrl[0] === '#' || $relativeUrl[0] === '?') {
return $baseUrl . $relativeUrl;
}
// parse base URL and convert to: $scheme, $host, $path, $query, $port, $user, $pass
extract(parse_url($baseUrl));
// if base URL contains a path remove non-directory elements from $path
if (isset($path) === true) {
$path = preg_replace('#/[^/]*$#', '', $path);
} else {
$path = '';
}
// if realtive URL starts with //
if (substr($relativeUrl, 0, 2) === '//') {
return $scheme . ':' . $relativeUrl;
}
// if realtive URL starts with /
if ($relativeUrl[0] === '/') {
$path = null;
}
$abs = null;
// if realtive URL contains a user
if (isset($user) === true) {
$abs .= $user;
// if realtive URL contains a password
if (isset($pass) === true) {
$abs .= ':' . $pass;
}
$abs .= '@';
}
$abs .= $host;
// if realtive URL contains a port
if (isset($port) === true) {
$abs .= ':' . $port;
}
$abs .= $path . '/' . $relativeUrl . (isset($query) === true ? '?' . $query : null);
// replace // or /./ or /foo/../ with /
$re = ['#(/\.?/)#', '#/(?!\.\.)[^/]+/\.\./#'];
for ($n = 1; $n > 0; $abs = preg_replace($re, '/', $abs, -1, $n)) {
}
// return absolute URL
return $scheme . '://' . $abs;
}
2023-02-15 08:46:47 +00:00
function array_to_csv($data, $delimiter = ',', $enclosure = '"', $escape_char = "\\") {
$f = fopen('php://memory', 'r+');
foreach ($data as $item) {
fputcsv($f, $item, $delimiter, $enclosure, $escape_char);
}
rewind($f);
return stream_get_contents($f);
}