misc_pterodactyl-panel/app/Repositories/UserRepository.php

168 lines
5.5 KiB
PHP
Raw Normal View History

<?php
2016-01-20 00:10:39 +00:00
/**
2016-01-20 21:05:16 +00:00
* Pterodactyl - Panel
2016-01-20 00:10:39 +00:00
* Copyright (c) 2015 - 2016 Dane Everitt <dane@daneeveritt.com>
* Some Modifications (c) 2015 Dylan Seidt <dylan.seidt@gmail.com>
2016-01-20 00:10:39 +00:00
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
2016-01-20 00:10:39 +00:00
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
2016-01-20 00:10:39 +00:00
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
2016-01-20 00:10:39 +00:00
*/
namespace Pterodactyl\Repositories;
2016-01-15 04:13:26 +00:00
use DB;
2016-01-21 03:08:09 +00:00
use Settings;
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);
2016-01-21 03:08:09 +00:00
$message->from(Settings::get('email_from', env('MAIL_FROM')), Settings::get('email_sender_name', env('MAIL_FROM_NAME', 'Pterodactyl Panel')));
$message->subject(Settings::get('company') . ' - New Account');
2016-01-19 00:57:10 +00:00
});
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)
{
$user = Models\User::findOrFail($id);
2016-01-15 04:13:26 +00:00
$validator = Validator::make($data, [
'email' => 'sometimes|required|email|unique:users,email,' . $id,
'password' => 'sometimes|required|regex:((?=.*\d)(?=.*[a-z])(?=.*[A-Z]).{8,})',
'root_admin' => 'sometimes|required|boolean',
'language' => 'sometimes|required|string|min:1|max:5',
'use_totp' => 'sometimes|required|boolean',
'totp_secret' => 'sometimes|required|size:16'
2016-01-15 04:13:26 +00:00
]);
// 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)) {
2016-02-04 16:11:38 +00:00
$data['password'] = Hash::make($data['password']);
}
if (isset($data['password_confirmation'])) {
unset($data['password_confirmation']);
}
$user->fill($data);
$user->save();
}
/**
* 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();
try {
Models\Permission::where('user_id', $id)->delete();
Models\Subuser::where('user_id', $id)->delete();
Models\User::destroy($id);
2016-01-15 04:13:26 +00:00
DB::commit();
return true;
} catch (\Exception $ex) {
DB::rollBack();
2016-01-15 04:13:26 +00:00
throw $ex;
}
}
}