misc_pterodactyl-panel/app/Repositories/UserRepository.php

136 lines
3.8 KiB
PHP
Raw Normal View History

<?php
namespace Pterodactyl\Repositories;
2016-01-15 04:13:26 +00:00
use DB;
use Hash;
2016-01-15 04:13:26 +00:00
use Validator;
2016-01-19 00:57:10 +00:00
use Mail;
2016-01-15 04:13:26 +00:00
use Pterodactyl\Models;
use Pterodactyl\Services\UuidService;
use Pterodactyl\Exceptions\DisplayValidationException;
2016-01-15 04:13:26 +00:00
use Pterodactyl\Exceptions\DisplayException;
class UserRepository
{
public function __construct()
{
//
}
/**
* Creates a user on the panel. Returns the created user's ID.
*
* @param string $email
* @param string $password An unhashed version of the user's password.
* @return bool|integer
*/
public function create($email, $password, $admin = false)
{
$validator = Validator::make([
'email' => $email,
'password' => $password,
2016-01-15 04:13:26 +00:00
'root_admin' => $admin
], [
'email' => 'required|email|unique:users,email',
'password' => 'required|regex:((?=.*\d)(?=.*[a-z])(?=.*[A-Z]).{8,})',
2016-01-15 04:13:26 +00:00
'root_admin' => 'required|boolean'
]);
// Run validator, throw catchable and displayable exception if it fails.
// Exception includes a JSON result of failed validation rules.
if ($validator->fails()) {
throw new DisplayValidationException($validator->errors());
}
2016-01-15 04:13:26 +00:00
$user = new Models\User;
$uuid = new UuidService;
2016-01-19 00:57:10 +00:00
DB::beginTransaction();
$user->uuid = $uuid->generate('users', 'uuid');
$user->email = $email;
$user->password = Hash::make($password);
2016-01-15 04:48:58 +00:00
$user->language = 'en';
$user->root_admin = ($admin) ? 1 : 0;
2016-01-19 00:57:10 +00:00
$user->save();
try {
2016-01-19 00:57:10 +00:00
Mail::queue('emails.new-account', [
'email' => $user->email,
'forgot' => route('auth.password'),
'login' => route('auth.login')
], function ($message) use ($email) {
$message->to($email);
$message->subject('Pterodactyl - New Account');
});
DB::commit();
return $user->id;
} catch (\Exception $ex) {
2016-01-19 00:57:10 +00:00
DB::rollBack();
throw $e;
}
}
/**
* Updates a user on the panel.
*
* @param integer $id
2016-01-15 04:13:26 +00:00
* @param array $data An array of columns and their associated values to update for the user.
* @return boolean
*/
2016-01-15 04:13:26 +00:00
public function update($id, array $data)
{
2016-01-15 04:13:26 +00:00
$validator = Validator::make($data, [
'email' => 'email|unique:users,email,' . $id,
'password' => 'regex:((?=.*\d)(?=.*[a-z])(?=.*[A-Z]).{8,})',
'root_admin' => 'boolean',
'language' => 'string|min:1|max:5',
'use_totp' => 'boolean',
'totp_secret' => 'size:16'
]);
// Run validator, throw catchable and displayable exception if it fails.
// Exception includes a JSON result of failed validation rules.
if ($validator->fails()) {
throw new DisplayValidationException($validator->errors());
}
2016-01-15 04:13:26 +00:00
if(array_key_exists('password', $data)) {
$user['password'] = Hash::make($data['password']);
}
return Models\User::findOrFail($id)->update($data);
}
/**
* Deletes a user on the panel, returns the number of records deleted.
*
* @param integer $id
* @return integer
*/
public function delete($id)
{
2016-01-15 04:13:26 +00:00
if(Models\Server::where('owner', $id)->count() > 0) {
throw new DisplayException('Cannot delete a user with active servers attached to thier account.');
}
DB::beginTransaction();
Models\Permission::where('user_id', $id)->delete();
Models\Subuser::where('user_id', $id)->delete();
Models\User::destroy($id);
try {
DB::commit();
return true;
} catch (\Exception $ex) {
throw $ex;
}
}
}