invoiceninja/app/Console/Commands/CreateTestData.php

<?php

namespace App\Console\Commands;

use App\Ninja\Repositories\AccountRepository;
use App\Ninja\Repositories\ClientRepository;
use App\Ninja\Repositories\ExpenseRepository;
use App\Ninja\Repositories\InvoiceRepository;
use App\Ninja\Repositories\PaymentRepository;
use App\Ninja\Repositories\VendorRepository;
use App\Ninja\Repositories\TaskRepository;
use App\Ninja\Repositories\ProjectRepository;
use App\Models\Client;
use App\Models\TaxRate;
use App\Models\Project;
use App\Models\ExpenseCategory;
use Auth;
use Faker\Factory;
use Illuminate\Console\Command;
use Utils;

/**
 * Class CreateTestData.
 */
class CreateTestData extends Command
{
    /**
     * @var string
     */
    protected $description = 'Create Test Data';
    /**
     * @var string
     */
    protected $signature = 'ninja:create-test-data {count=1} {create_account=false} {--database}';

    /**
     * @var
     */
    protected $token;

    /**
     * CreateTestData constructor.
     *
     * @param ClientRepository  $clientRepo
     * @param InvoiceRepository $invoiceRepo
     * @param PaymentRepository $paymentRepo
     * @param VendorRepository  $vendorRepo
     * @param ExpenseRepository $expenseRepo
     * @param TaskRepository $taskRepo
     * @param AccountRepository $accountRepo
     */
    public function __construct(
        ClientRepository $clientRepo,
        InvoiceRepository $invoiceRepo,
        PaymentRepository $paymentRepo,
        VendorRepository $vendorRepo,
        ExpenseRepository $expenseRepo,
        TaskRepository $taskRepo,
        ProjectRepository $projectRepo,
        AccountRepository $accountRepo)
    {
        parent::__construct();

        $this->faker = Factory::create();

        $this->clientRepo = $clientRepo;
        $this->invoiceRepo = $invoiceRepo;
        $this->paymentRepo = $paymentRepo;
        $this->vendorRepo = $vendorRepo;
        $this->expenseRepo = $expenseRepo;
        $this->taskRepo = $taskRepo;
        $this->projectRepo = $projectRepo;
        $this->accountRepo = $accountRepo;
    }

    /**
     * @return bool
     */
    public function fire()
    {
        if (Utils::isNinjaProd()) {
            $this->info('Unable to run in production');
            return false;
        }

        $this->info(date('r').' Running CreateTestData...');
        $this->count = $this->argument('count');

        if ($database = $this->option('database')) {
            config(['database.default' => $database]);
        }

        if (filter_var($this->argument('create_account'), FILTER_VALIDATE_BOOLEAN)) {
            $this->info('Creating new account...');
            $account = $this->accountRepo->create(
                $this->faker->firstName,
                $this->faker->lastName,
                $this->faker->safeEmail
            );
            Auth::login($account->users[0]);
        } else {
            $this->info('Using first account...');
            Auth::loginUsingId(1);
        }

        $this->createClients();
        $this->createVendors();
        $this->createOtherObjects();

        $this->info('Done');
    }

    private function createClients()
    {
        for ($i = 0; $i < $this->count; $i++) {
            $data = [
                'name' => $this->faker->name,
                'address1' => $this->faker->streetAddress,
                'address2' => $this->faker->secondaryAddress,
                'city' => $this->faker->city,
                'state' => $this->faker->state,
                'postal_code' => $this->faker->postcode,
                'contacts' => [[
                    'first_name' => $this->faker->firstName,
                    'last_name' => $this->faker->lastName,
                    'email' => $this->faker->safeEmail,
                    'phone' => $this->faker->phoneNumber,
                ]],
            ];

            $client = $this->clientRepo->save($data);
            $this->info('Client: ' . $client->name);

            $this->createInvoices($client);
            $this->createInvoices($client, true);
            $this->createTasks($client);
        }
    }

    /**
     * @param $client
     */
    private function createInvoices($client, $isQuote = false)
    {
        for ($i = 0; $i < $this->count; $i++) {
            $data = [
                'is_public' => true,
                'is_quote' => $isQuote,
                'client_id' => $client->id,
                'invoice_date_sql' => date_create()->modify(rand(-100, 100) . ' days')->format('Y-m-d'),
                'due_date_sql' => date_create()->modify(rand(-100, 100) . ' days')->format('Y-m-d'),
                'invoice_items' => [[
                    'product_key' => $this->faker->word,
                    'qty' => $this->faker->randomDigit + 1,
                    'cost' => $this->faker->randomFloat(2, 1, 10),
                    'notes' => $this->faker->text($this->faker->numberBetween(50, 300)),
                ]],
            ];

            $invoice = $this->invoiceRepo->save($data);
            $this->info('Invoice: ' . $invoice->invoice_number);

            if (! $isQuote) {
                $this->createPayment($client, $invoice);
            }
        }
    }

    /**
     * @param $client
     * @param $invoice
     */
    private function createPayment($client, $invoice)
    {
        $data = [
            'invoice_id' => $invoice->id,
            'client_id' => $client->id,
            'amount' => $this->faker->randomFloat(2, 0, $invoice->amount),
            'payment_date_sql' => date_create()->modify(rand(-100, 100) . ' days')->format('Y-m-d'),
        ];

        $payment = $this->paymentRepo->save($data);

        $this->info('Payment: ' . $payment->amount);
    }

    private function createTasks($client)
    {
        $data = [
            'client_id' => $client->id,
            'name' => $this->faker->sentence(3),
        ];
        $project = $this->projectRepo->save($data);

        for ($i = 0; $i < $this->count; $i++) {
            $startTime = date_create()->modify(rand(-100, 100) . ' days')->format('U');
            $endTime = $startTime + (60 * 60 * 2);
            $timeLog = "[[{$startTime},{$endTime}]]";
            $data = [
                'client_id' => $client->id,
                'project_id' => $project->id,
                'description' => $this->faker->text($this->faker->numberBetween(50, 300)),
                'time_log' => $timeLog,
            ];

            $this->taskRepo->save(false, $data);
        }
    }



    private function createVendors()
    {
        for ($i = 0; $i < $this->count; $i++) {
            $data = [
                'name' => $this->faker->name,
                'address1' => $this->faker->streetAddress,
                'address2' => $this->faker->secondaryAddress,
                'city' => $this->faker->city,
                'state' => $this->faker->state,
                'postal_code' => $this->faker->postcode,
                'vendor_contacts' => [[
                    'first_name' => $this->faker->firstName,
                    'last_name' => $this->faker->lastName,
                    'email' => $this->faker->safeEmail,
                    'phone' => $this->faker->phoneNumber,
                ]],
            ];

            $vendor = $this->vendorRepo->save($data);
            $this->info('Vendor: ' . $vendor->name);

            $this->createExpense($vendor);
        }
    }

    /**
     * @param $vendor
     */
    private function createExpense($vendor)
    {
        for ($i = 0; $i < $this->count; $i++) {
            $data = [
                'vendor_id' => $vendor->id,
                'amount' => $this->faker->randomFloat(2, 1, 10),
                'expense_date' => date_create()->modify(rand(-100, 100) . ' days')->format('Y-m-d'),
                'public_notes' => '',
            ];

            $expense = $this->expenseRepo->save($data);
            $this->info('Expense: ' . $expense->amount);
        }
    }

    private function createOtherObjects()
    {
        $this->createTaxRate('Tax 1', 10, 1);
        $this->createTaxRate('Tax 2', 20, 2);

        $this->createCategory('Category 1', 1);
        $this->createCategory('Category 1', 2);

        $this->createProject('Project 1', 1);
        $this->createProject('Project 2', 2);
    }

    private function createTaxRate($name, $rate, $publicId)
    {
        $taxRate = new TaxRate();
        $taxRate->name = $name;
        $taxRate->rate = $rate;
        $taxRate->account_id = 1;
        $taxRate->user_id = 1;
        $taxRate->public_id = $publicId;
        $taxRate->save();
    }

    private function createCategory($name, $publicId)
    {
        $category = new ExpenseCategory();
        $category->name = $name;
        $category->account_id = 1;
        $category->user_id = 1;
        $category->public_id = $publicId;
        $category->save();
    }

    private function createProject($name, $publicId)
    {
        $project = new Project();
        $project->name = $name;
        $project->account_id = 1;
        $project->client_id = 1;
        $project->user_id = 1;
        $project->public_id = $publicId;
        $project->save();
    }

    /**
     * @return array
     */
    protected function getArguments()
    {
        return [];
    }

    /**
     * @return array
     */
    protected function getOptions()
    {
        return [];
    }
}