invoiceninja/app/Http/Controllers/SetupController.php

<?php

/**
 * Invoice Ninja (https://invoiceninja.com).
 *
 * @link https://github.com/invoiceninja/invoiceninja source repository
 *
 * @copyright Copyright (c) 2020. Invoice Ninja LLC (https://invoiceninja.com)
 *
 * @license https://opensource.org/licenses/AAL
 */

namespace App\Http\Controllers;

use App\Http\Requests\Setup\CheckDatabaseRequest;
use App\Http\Requests\Setup\CheckMailRequest;
use App\Http\Requests\Setup\StoreSetupRequest;
use App\Jobs\Account\CreateAccount;
use App\Jobs\Util\VersionCheck;
use App\Models\Account;
use App\Utils\CurlUtils;
use App\Utils\SystemHealth;
use App\Utils\Traits\AppSetup;
use DB;
use Exception;
use Illuminate\Contracts\Foundation\Application;
use Illuminate\Contracts\Routing\ResponseFactory;
use Illuminate\Http\JsonResponse;
use Illuminate\Http\Response;
use Illuminate\Support\Facades\Artisan;
use Illuminate\Support\Facades\Request;
use Illuminate\Support\Facades\Schema;
use Illuminate\Support\Facades\Storage;
use Spatie\Browsershot\Browsershot;

/**
 * Class SetupController.
 */
class SetupController extends Controller
{
    use AppSetup;

    public function index()
    {
        $check = SystemHealth::check(false);

        if ($check['system_health'] == true && $check['simple_db_check'] && Schema::hasTable('accounts') && $account = Account::all()->first()) {
            return redirect('/');
        }

        return view('setup.index', ['check' => $check]);
    }

    public function doSetup(StoreSetupRequest $request)
    {
        $check = SystemHealth::check(false);

        if ($check['system_health'] === false) {
            info($check);

            return response('Oops, something went wrong. Check your logs.'); /* We should never reach this block, but jic. */
        }

        $mail_driver = $request->input('mail_driver');

        if (! $this->failsafeMailCheck($request)) {
            $mail_driver = 'log';
        }

        $url = $request->input('url');

        if (substr($url, -1) != '/')
            $url = $url . '/';

        $_ENV['APP_KEY'] = config('app.key');
        $_ENV['APP_URL'] = $url;
        $_ENV['APP_DEBUG'] = $request->input('debug') ? 'true' : 'false';
        $_ENV['REQUIRE_HTTPS'] = $request->input('https') ? 'true' : 'false';
        $_ENV['DB_TYPE'] = 'mysql';
        $_ENV['DB_HOST1'] = $request->input('host');
        $_ENV['DB_DATABASE1'] = $request->input('database');
        $_ENV['DB_USERNAME1'] = $request->input('db_username');
        $_ENV['DB_PASSWORD1'] = $request->input('db_password');
        $_ENV['MAIL_MAILER'] = $mail_driver;
        $_ENV['MAIL_PORT'] = $request->input('mail_port');
        $_ENV['MAIL_ENCRYPTION'] = $request->input('encryption');
        $_ENV['MAIL_HOST'] = $request->input('mail_host');
        $_ENV['MAIL_USERNAME'] = $request->input('mail_username');
        $_ENV['MAIL_FROM_NAME'] = $request->input('mail_name');
        $_ENV['MAIL_FROM_ADDRESS'] = $request->input('mail_address');
        $_ENV['MAIL_PASSWORD'] = $request->input('mail_password');
        $_ENV['NINJA_ENVIRONMENT'] = 'selfhost';
        $_ENV['DB_CONNECTION'] = 'db-ninja-01';

        $config = '';

        try {
            foreach ($_ENV as $key => $val) {
                if (is_array($val)) {
                    continue;
                }
                if (preg_match('/\s/', $val)) {
                    $val = "'{$val}'";
                }
                $config .= "{$key}={$val}\n";
            }

            /* Write the .env file */
            $filePath = base_path() . '/.env';
            $fp = fopen($filePath, 'w');
            fwrite($fp, $config);
            fclose($fp);

            /* We need this in some environments that do not have STDIN defined */
            define('STDIN', fopen('php://stdin', 'r'));

            /* Make sure no stale connections are cached */
            DB::purge('db-ninja-01');

            /* Run migrations */
            Artisan::call('optimize');
            Artisan::call('migrate', ['--force' => true]);
            Artisan::call('db:seed', ['--force' => true]);

            Storage::disk('local')->delete('test.pdf');

            /* Create the first account. */
            if (Account::count() == 0) {
                CreateAccount::dispatchNow($request->all());
            }

            VersionCheck::dispatchNow();
            
            $this->buildCache();

            return redirect('/');
        } catch (Exception $e) {
            info($e->getMessage());

            return redirect()
                ->back()
                ->with('setup_error', $e->getMessage());
        }
    }

    /**
     * Return status based on check of database connection.
     *
     * @param CheckDatabaseRequest $request
     * @return Response
     */
    public function checkDB(CheckDatabaseRequest $request): Response
    {
        $status = SystemHealth::dbCheck($request);

        if (is_array($status) && $status['success'] === true) {
            return response([], 200);
        }

        return response($status, 400);
    }

    /**
     * Return status based on check of SMTP connection.
     *
     * @param CheckMailRequest $request
     * @return Application|ResponseFactory|JsonResponse|Response
     */
    public function checkMail(CheckMailRequest $request)
    {
        try {
            $response_array = SystemHealth::testMailServer($request);

            if (count($response_array) == 0) {
                return response([], 200);
            } else {
                return response()->json($response_array, 200);
            }
        } catch (Exception $e) {
            info(['message' => $e->getMessage(), 'action' => 'SetupController::checkMail()']);

            return response()->json(['message' => $e->getMessage()], 400);
        }
    }

    private function failsafeMailCheck($request)
    {
        $response_array = SystemHealth::testMailServer($request);

        if ($response_array instanceof Response) {
            return true;
        }

        return false;
    }

    public function checkPdf(Request $request)
    {
        try {
            if (config('ninja.phantomjs_key')) {
                return $this->testPhantom();
            }

            Browsershot::html('GENERATING PDFs WORKS! Thank you for using Invoice Ninja!')
                ->setNodeBinary(config('ninja.system.node_path'))
                ->setNpmBinary(config('ninja.system.npm_path'))
                ->noSandbox()
                ->savePdf(
                    public_path('test.pdf')
                );

            return response(['url' => asset('test.pdf')], 200);
        } catch (Exception $e) {
            info($e->getMessage());

            return response([], 500);
        }
    }

    private function testPhantom()
    {
        try {
            $key = config('ninja.phantomjs_key');
            $url = 'https://www.invoiceninja.org/';

            $phantom_url = "https://phantomjscloud.com/api/browser/v2/{$key}/?request=%7Burl:%22{$url}%22,renderType:%22pdf%22%7D";
            $pdf = CurlUtils::get($phantom_url);

            Storage::disk(config('filesystems.default'))->put('test.pdf', $pdf);
            Storage::disk('local')->put('test.pdf', $pdf);

            return response(['url' => Storage::disk('local')->url('test.pdf')], 200);
        } catch (Exception $e) {
            return response([], 500);
        }
    }
}