[ 'blue' => 'Blue', 'green' => 'Green', 'pink' => 'Pink', 'red' => 'Red', 'orange' => 'Orange', 'cyan' => 'Cyan', ], 'admin' => [ 'purple' => 'Purple', ], ]; const AVAILABLE_COLORS = [ 'blue', 'green', 'pink', 'red', 'orange', 'cyan', 'purple', ]; public static $visibility_codes = [ 'all' => [ 'html' => '', ], 'self' => [ 'html' => '（仅自己可见）', ], ]; public static function query($username) { if (!validateUsername($username)) { return null; } return DB::selectFirst([ "select * from user_info", "where", ['username' => $username] ]); } public static function checkBasicInfo($user, $cfg = []) { $cfg += [ 'check_email' => true, ]; if (!isset($user['username']) || !validateUsername($user['username'])) { throw new UOJInvalidArgumentException('无效用户名'); } if (UOJUser::query($user['username'])) { throw new UOJInvalidArgumentException('用户名已存在'); } if (!isset($user['password']) || !validatePassword($user['password'])) { throw new UOJInvalidArgumentException('无效密码'); } if ($cfg['check_email'] && (!isset($user['email']) || !validateEmail($user['email']))) { throw new UOJInvalidArgumentException('无效电子邮箱'); } } public static function register($user, $cfg = []) { $cfg += [ 'extra' => [], ]; UOJUser::checkBasicInfo($user, $cfg); $password = getPasswordToStore($user['password'], $user['username']); $extra = [ 'school' => $user['school'] ?: null, ] + $cfg['extra']; $info = [ 'username' => $user['username'], 'realname' => $user['realname'] ?: '', 'email' => $user['email'], 'password' => $password, 'svn_password' => uojRandString(20), 'register_time' => DB::now(), 'extra' => json_encode($extra, JSON_UNESCAPED_UNICODE), ]; // 0 means non-existence, false means DB error. if (DB::selectExists("select 1 from user_info") === 0) { $info['usergroup'] = 'S'; } DB::insert([ "insert into user_info", DB::bracketed_fields(array_keys($info)), "values", DB::tuple($info) ]); return $user; } public static function registerTmpAccount($user, $cfg = []) { $cfg += [ 'extra' => [], 'check_email' => false, 'type' => 'student', ]; UOJUser::checkBasicInfo($user, $cfg); if (!isset($user['expiration_time'])) { throw new UOJInvalidArgumentException('无效账号过期时间'); } $password = getPasswordToStore($user['password'], $user['username']); $extra = [ 'school' => $user['school'] ?: null, 'permissions' => [ 'problems' => [ 'view' => false, 'download_testdata' => false, 'create' => false, 'manage' => false, ], 'contests' => [ 'view' => false, 'register' => false, 'create' => false, 'start_final_test' => false, 'manage' => false, ], 'lists' => [ 'view' => false, 'create' => false, 'manage' => false, ], 'groups' => [ 'view' => false, 'create' => false, 'manage' => false, ], 'blogs' => [ 'view' => false, 'create' => false, 'manage' => false, ], 'users' => [ 'view' => false, 'upload_image' => false, ], ], ] + $cfg['extra']; $info = [ 'username' => $user['username'], 'usergroup' => 'T', 'usertype' => $cfg['type'], 'realname' => $user['realname'] ?: '', 'email' => $user['email'], 'password' => $password, 'svn_password' => uojRandString(20), 'register_time' => DB::now(), 'expiration_time' => $user['expiration_time'], 'extra' => json_encode($extra, JSON_UNESCAPED_UNICODE), ]; DB::insert([ "insert into user_info", DB::bracketed_fields(array_keys($info)), "values", DB::tuple($info) ]); return $user; } public static function registerTmpACMTeamAccount($team, $cfg = []) { $cfg += [ 'check_email' => false, 'extra' => [], 'type' => 'team', ]; UOJUser::checkBasicInfo($team, $cfg); if (!isset($team['expiration_time'])) { throw new UOJInvalidArgumentException('无效账号过期时间'); } $cfg['extra'] += [ 'acm' => [ 'contest_name' => $team['contest_name'], 'team_name' => $team['team_name'], 'members' => $team['members'], ], ]; return UOJUser::registerTmpAccount($team, $cfg); } public static function registerTmpACMTeamAccountFromText($text, $contest_name, $expiration_time) { $fields = array_map('trim', str_getcsv($text)); if (count($fields) < 7) { throw new UOJInvalidArgumentException('格式不合规范'); } $num = (int)$fields[4]; if (count($fields) != 5 + $num * 2) { throw new UOJInvalidArgumentException('格式不合规范'); } $mem = []; for ($i = 0; $i < $num; $i++) { $mem[] = [ 'name' => $fields[5 + $i * 2], 'organization' => $fields[5 + $i * 2 + 1] ]; } $team = [ 'username' => $fields[0], 'password' => hash_hmac('md5', $fields[1], getPasswordClientSalt()), 'email' => $fields[2], 'contest_name' => $contest_name, 'expiration_time' => $expiration_time, 'team_name' => $fields[3], 'members' => $mem, ]; return UOJUser::registerTmpACMTeamAccount($team); } public static function convertACMTeamInfoToText($mem) { $csv_array = [count($mem)]; foreach ($mem as $member) { $csv_array[] = $member['name']; $csv_array[] = $member['organization']; } return array_to_csv([$csv_array]); } public static function parseACMTeamInfoFromText($text) { $fields = array_map('trim', str_getcsv($text)); if (empty($fields)) { throw new UOJInvalidArgumentException('格式不合规范'); } $num = (int)$fields[0]; if (count($fields) != 1 + $num * 2) { throw new UOJInvalidArgumentException('格式不合规范'); } $mem = []; for ($i = 0; $i < $num; $i++) { $mem[] = [ 'name' => $fields[1 + $i * 2], 'organization' => $fields[1 + $i * 2 + 1] ]; } return $mem; } public static function getAccountStatus($user) { if ($user['usergroup'] == 'B') { return 'banned'; } elseif ($user['usergroup'] == 'T' && UOJTime::$time_now > new DateTime($user['expiration_time'])) { return 'expired'; } else { return 'ok'; } } public static function getRealname($user) { $realname = $user['realname']; if ($user['usertype'] == 'teacher') { $realname .= '老师'; } return $realname; } public static function getUserColor($user) { $extra = UOJUser::getExtra($user); return UOJUser::getUserColor2($user['usergroup'], $extra['username_color']); } public static function getUserColor2($usergroup, $custom_color = null) { if ($usergroup == 'B') { return 'brown'; } if ($usergroup == 'T') { return 'gray'; } if ($usergroup == 'S') { return $custom_color ?: 'purple'; } // 前管理员设置颜色为紫色的，颜色改为蓝色 if ($custom_color == 'purple') { return 'blue'; } return $custom_color ?: 'blue'; } public static function getLink($user, $cfg = []) { $cfg += [ 'color' => true, ]; if (is_string($user)) { $info = UOJUser::query($user); if (!$info) { return $user; } else { $user = $info; } } $realname = UOJUser::getRealname($user); // 未登录不可查看真实姓名 if (!Auth::check()) { $realname = ''; } $color = $cfg['color'] ? UOJUser::getUserColor($user) : ''; return HTML::tag('span', [ 'class' => "uoj-username", 'data-realname' => trim(HTML::escape($realname)), 'data-color' => $color, ], $user['username']); } public static function getUpdatedExtraVisitHistory($history, $cur = null) { $new_h = []; $oldest = clone UOJTime::$time_now; $oldest->modify('-1 month'); for ($i = 0; $i < count($history); $i++) { if (UOJTime::str2time($history[$i]['last']) >= $oldest) { $new_h[] = $history[$i]; } } if ($cur === null) { return $new_h; } for ($i = 0; $i < count($new_h); $i++) { if ( $new_h[$i]['addr'] == $cur['addr'] && $new_h[$i]['forwarded_addr'] == $cur['forwarded_addr'] && $new_h[$i]['ua'] == $cur['ua'] ) { $new_h[$i]['last'] = $cur['last']; return $new_h; } } if (count($new_h) < UOJUser::MAX_HISTORY_LEN) { $new_h[] = $cur; return $new_h; } $p = 0; for ($i = 1; $i < count($new_h); $i++) { if (strcmp($new_h[$i]['last'], $new_h[$p]['last']) < 0) { $p = $i; } } $new_h[$p] = $cur; return $new_h; } public static function sortExtraVisitHistory(&$history) { usort($history, function ($a, $b) { return -strcmp($a['last'], $b['last']); }); } public static function getExtra($user, $key = null) { $extra = json_decode($user['extra'], true); if ($extra === null) { $extra = []; } mergeConfig($extra, [ 'school' => null, 'permissions' => UOJContext::getMeta('users_default_permissions'), 'social' => [ 'codeforces' => null, 'github' => null, 'website' => null, ], 'image_hosting' => [ 'total_size_limit' => 104857600, // 100 MiB ], 'history' => [], 'show_email' => 'all', 'show_qq' => 'all', 'avatar_source' => 'gravatar', 'username_color' => isSuperUser($user) ? '#9d3dcf' : '#0d6efd', ]); if ($key === null) { return $extra; } return $extra[$key]; } public static function checkVisibility(string $code, array $user, ?array $viewer) { switch ($code) { case 'all': return true; case 'self': return $viewer && $user['username'] === $viewer['username']; default: return false; } } public static function getVisibilityHTML(string $code) { return static::$visibility_codes[$code]['html']; } public static function viewerCanSeeComponents(array $user, ?array $viewer) { // assert($viewer can view $user's probile). This is always true because everyone's profile is public. $extra = UOJUser::getExtra($user); return [ 'email' => UOJUser::checkVisibility($extra['show_email'], $user, $viewer), 'qq' => UOJUser::checkVisibility($extra['show_qq'], $user, $viewer) ]; } public static function checkPermission(array $user = null, string $perm = '') { if ($user == null) { return false; } $extra = UOJUser::getExtra($user); $cur = $extra['permissions']; foreach (explode('.', $perm) as $p) { if (!is_assoc($cur) || !isset($cur[$p])) { return false; } $cur = $cur[$p]; } return $cur; } public static function getMatchedVisitHistory($user, $info) { $extra = UOJUser::getExtra($user); $new_h = UOJUser::getUpdatedExtraVisitHistory($extra['history']); foreach ($new_h as $history) { if ( $history['addr'] == $info['addr'] && $history['forwarded_addr'] == $info['forwarded_addr'] && $history['ua'] == substr($info['ua'], 0, UOJUser::MAX_UA_LEN) ) { return $history; } } return null; } public static function updateVisitHistory($user, $info) { $extra = UOJUser::getExtra($user); $cur = [ 'addr' => $info['remote_addr'], 'forwarded_addr' => $info['http_x_forwarded_for'], 'ua' => substr($info['http_user_agent'], 0, UOJUser::MAX_UA_LEN), 'last' => UOJTime::$time_now_str, ]; $extra['history'] = UOJUser::getUpdatedExtraVisitHistory($extra['history'], $cur); $user['remote_addr'] = $info['remote_addr']; $user['http_x_forwarded_for'] = $info['http_x_forwarded_for']; $user['extra'] = json_encode($extra, JSON_UNESCAPED_UNICODE); DB::update([ 'update user_info', 'set', [ 'remote_addr' => $user['remote_addr'], 'http_x_forwarded_for' => $user['http_x_forwarded_for'], 'last_visit_time' => DB::now(), 'extra' => $user['extra'] ], "where", ["username" => $user['username']] ]); return $user; } }