invoiceninja/app/Console/Commands/CreateTestData.php

<?php namespace App\Console\Commands;

use stdClass;
use Auth;
use DB;
use Utils;
use Artisan;
use Illuminate\Console\Command;
use Faker\Factory;
use App\Models\User;

use App\Ninja\Repositories\ClientRepository;
use App\Ninja\Repositories\InvoiceRepository;
use App\Ninja\Repositories\PaymentRepository;
use App\Ninja\Repositories\VendorRepository;
use App\Ninja\Repositories\ExpenseRepository;

class CreateTestData extends Command
{
    //protected $name = 'ninja:create-test-data';
    protected $description = 'Create Test Data';
    protected $signature = 'ninja:create-test-data {count=1}';

    protected $token;

    public function __construct(
        ClientRepository $clientRepo, 
        InvoiceRepository $invoiceRepo, 
        PaymentRepository $paymentRepo,
        VendorRepository $vendorRepo,
        ExpenseRepository $expenseRepo)
    {
        parent::__construct();

        $this->faker = Factory::create();
                
        $this->clientRepo = $clientRepo;
        $this->invoiceRepo = $invoiceRepo;
        $this->paymentRepo = $paymentRepo;
        $this->vendorRepo = $vendorRepo;
        $this->expenseRepo = $expenseRepo;
    }

    public function fire()
    {
        if (Utils::isNinjaProd()) {
            return false;
        }
        
        $this->info(date('Y-m-d').' Running CreateTestData...');

        Auth::loginUsingId(1);
        $this->count = $this->argument('count');
        
        $this->createClients();
        $this->createVendors();
        
        $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);
        }
    }

    private function createInvoices($client)
    {   
        for ($i=0; $i<$this->count; $i++) {
            $data = [
                'client_id' => $client->id,
                '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);
            
            $this->createPayment($client, $invoice);
        }
    }
    
    private function createPayment($client, $invoice)
    {
        $data = [
            'invoice_id' => $invoice->id,
            'client_id' => $client->id,
            'amount' => $this->faker->randomFloat(2, 0, $invoice->amount)
        ];
        
        $payment = $this->paymentRepo->save($data);
        
        $this->info('Payment: ' . $payment->amount);
    }

    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);
        }
    }
    
    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' => null,
                'public_notes' => null,
            ];
            
            $expense = $this->expenseRepo->save($data);
            $this->info('Expense: ' . $expense->amount);
        }
    }

    protected function getArguments()
    {
        return array(
            //array('example', InputArgument::REQUIRED, 'An example argument.'),
        );
    }

    protected function getOptions()
    {
        return array(
            //array('example', null, InputOption::VALUE_OPTIONAL, 'An example option.', null),
        );
    }
}