snipe-it/app/Console/Commands/LicenseImportCommand.php

<?php
namespace App\Console\Commands;

use Illuminate\Console\Command;
use Symfony\Component\Console\Input\InputOption;
use Symfony\Component\Console\Input\InputArgument;
use League\Csv\Reader;
use App\Models\User;
use App\Models\Supplier;
use App\Models\License;
use App\Models\LicenseSeat;

class LicenseImportCommand extends Command
{

    /**
     * The console command name.
     *
     * @var string
     */
    protected $name = 'snipeit:license-import';

    /**
     * The console command description.
     *
     * @var string
     */
    protected $description = 'Import Licenses from CSV';

    /**
     * Create a new command instance.
     *
     * @return void
     */
    public function __construct()
    {
        parent::__construct();
    }

    /**
     * Execute the console command.
     *
     * @return mixed
     */
    public function fire()
    {
        $filename = $this->argument('filename');


        if (!$this->option('testrun')=='true') {
            $this->comment('======= Importing Licenses from '.$filename.' =========');
        } else {
            $this->comment('====== TEST ONLY License Import for '.$filename.' ====');
            $this->comment('============== NO DATA WILL BE WRITTEN ==============');
        }

        if (! ini_get("auto_detect_line_endings")) {
            ini_set("auto_detect_line_endings", '1');
        }

        $csv = Reader::createFromPath($this->argument('filename'));
        $csv->setNewline("\r\n");
        $csv->setOffset(1);
        $duplicates = '';

        // Loop through the records
        $nbInsert = $csv->each(function ($row) use ($duplicates) {
            $status_id = 1;

            // Let's just map some of these entries to more user friendly words

            if (array_key_exists('0', $row)) {
                $user_name = trim($row[0]);
            } else {
                $user_name = '';
            }

            if (array_key_exists('1', $row)) {
                $user_email = trim($row[1]);
            } else {
                $user_email = '';
            }

            if (array_key_exists('2', $row)) {
                $user_username = trim($row[2]);
            } else {
                $user_username = '';
            }

            if (array_key_exists('3', $row)) {
                $user_license_name = trim($row[3]);
            } else {
                $user_license_name = '';
            }

            if (array_key_exists('4', $row)) {
                $user_license_serial = trim($row[4]);
            } else {
                $user_license_serial = '';
            }

            if (array_key_exists('5', $row)) {
                $user_licensed_to_name = trim($row[5]);
            } else {
                $user_licensed_to_name = '';
            }

            if (array_key_exists('6', $row)) {
                $user_licensed_to_email = trim($row[6]);
            } else {
                $user_licensed_to_email = '';
            }

            if (array_key_exists('7', $row)) {
                $user_license_seats = trim($row[7]);
            } else {
                $user_license_seats = '';
            }

            if (array_key_exists('8', $row)) {
                $user_license_reassignable = trim($row[8]);
                if ($user_license_reassignable!='') {
                    if ((strtolower($user_license_reassignable)=='yes') || (strtolower($user_license_reassignable)=='true') || ($user_license_reassignable=='1')) {
                        $user_license_reassignable = 1;
                    }
                } else {
                    $user_license_reassignable = 0;
                }
            } else {
                $user_license_reassignable = 0;
            }

            if (array_key_exists('9', $row)) {
                $user_license_supplier = trim($row[9]);
            } else {
                $user_license_supplier = '';
            }

            if (array_key_exists('10', $row)) {
                $user_license_maintained = trim($row[10]);

                if ($user_license_maintained!='') {
                    if ((strtolower($user_license_maintained)=='yes') || (strtolower($user_license_maintained)=='true') || ($user_license_maintained=='1')) {
                        $user_license_maintained = 1;
                    }
                } else {
                    $user_license_maintained = 0;
                }


            } else {
                $user_license_maintained = '';
            }

            if (array_key_exists('11', $row)) {
                $user_license_notes = trim($row[11]);
            } else {
                $user_license_notes = '';
            }

            if (array_key_exists('12', $row)) {
                if ($row[12]!='') {
                    $user_license_purchase_date = date("Y-m-d 00:00:01", strtotime($row[12]));
                } else {
                    $user_license_purchase_date = '';
                }
            } else {
                $user_license_purchase_date = 0;
            }

            // A number was given instead of a name
            if (is_numeric($user_name)) {
                $this->comment('User '.$user_name.' is not a name - assume this user already exists');
                $user_username = '';
            // No name was given

            } elseif ($user_name=='') {
                $this->comment('No user data provided - skipping user creation, just adding license');
                $first_name = '';
                $last_name = '';
                $user_username = '';

            } else {

                    $name = explode(" ", $user_name);
                    $first_name = $name[0];
                    $email_last_name = '';
                    $email_prefix = $first_name;

                if (!array_key_exists(1, $name)) {
                    $last_name='';
                    $email_last_name = $last_name;
                    $email_prefix = $first_name;
                } else {
                    $last_name = str_replace($first_name, '', $user_name);

                    if ($this->option('email_format')=='filastname') {
                        $email_last_name.=str_replace(' ', '', $last_name);
                        $email_prefix = $first_name[0].$email_last_name;

                    } elseif ($this->option('email_format')=='firstname.lastname') {
                        $email_last_name.=str_replace(' ', '', $last_name);
                        $email_prefix = $first_name.'.'.$email_last_name;

                    } elseif ($this->option('email_format')=='firstname') {
                        $email_last_name.=str_replace(' ', '', $last_name);
                        $email_prefix = $first_name;
                    }


                }


                    $user_username = $email_prefix;

                    // Generate an email based on their name if no email address is given
                if ($user_email=='') {
                    if ($first_name=='Unknown') {
                        $status_id = 7;
                    }
                    $email = strtolower($email_prefix).'@'.$this->option('domain');
                    $user_email = str_replace("'", '', $email);
                }
            }

            $this->comment('Full Name: '.$user_name);
            $this->comment('First Name: '.$first_name);
            $this->comment('Last Name: '.$last_name);
            $this->comment('Username: '.$user_username);
            $this->comment('Email: '.$user_email);
            $this->comment('License Name: '.$user_license_name);
            $this->comment('Serial No: '.$user_license_serial);
            $this->comment('Licensed To Name: '.$user_licensed_to_name);
            $this->comment('Licensed To Email: '.$user_licensed_to_email);
            $this->comment('Seats: '.$user_license_seats);
            $this->comment('Reassignable: '.$user_license_reassignable);
            $this->comment('Supplier: '.$user_license_supplier);
            $this->comment('Maintained: '.$user_license_maintained);
            $this->comment('Notes: '.$user_license_notes);
            $this->comment('Purchase Date: '.$user_license_purchase_date);

            $this->comment('------------- Action Summary ----------------');

            if ($user_username!='') {
                if ($user = User::where('username', $user_username)->whereNotNull('username')->first()) {
                    $this->comment('User '.$user_username.' already exists');
                } else {

                    $user = new \App\Models\User;
                    $password  = substr(str_shuffle("0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"), 0, 20);

                    $user->first_name = $first_name;
                    $user->last_name = $last_name;
                    $user->username = $user_username;
                    $user->email = $user_email;
                    $user->permissions = '{user":1}';
                    $user->password = bcrypt($password);
                    $user->activated = 1;
                    if ($user->save()) {
                        $this->comment('User '.$first_name.' created');
                    } else {
                        $this->error('ERROR CREATING User '.$first_name.' '.$last_name);
                        $this->error($user->getErrors());
                    }

                    $this->comment('User '.$first_name.' created');
                }
            } else {
                $user = new User;
                $user->user_id = null;
            }


            // Check for the supplier match and create it if it doesn't exist
            if ($supplier = Supplier::where('name', $user_license_supplier)->first()) {
                $this->comment('Supplier '.$user_license_supplier.' already exists');
            } else {
                $supplier = new Supplier();
                $supplier->name = e($user_license_supplier);
                $supplier->user_id = 1;

                if ($supplier->save()) {
                    $this->comment('Supplier '.$user_license_supplier.' was created');
                } else {
                    $this->comment('Something went wrong! Supplier '.$user_license_supplier.' was NOT created');
                }

            }


            // Add the license
            $license = new License();
            $license->name = e($user_license_name);
            if ($user_license_purchase_date!='') {
                $license->purchase_date = $user_license_purchase_date;
            } else {
                $license->purchase_date = null;
            }
            $license->serial = e($user_license_serial);
            $license->seats = e($user_license_seats);
            $license->supplier_id = $supplier->id;
            $license->user_id = 1;
            if ($user_license_purchase_date!='') {
                $license->purchase_date = $user_license_purchase_date;
            } else {
                $license->purchase_date = null;
            }
            $license->license_name = $user_licensed_to_name;
            $license->license_email = $user_licensed_to_email;
            $license->notes = e($user_license_notes);

            if ($license->save()) {
                $this->comment('License '.$user_license_name.' with serial number '.$user_license_serial.' was created');


                $license_seat_created = 0;

                for ($x = 0; $x < $user_license_seats; $x++) {
                    // Create the license seat entries
                    $license_seat = new LicenseSeat();
                    $license_seat->license_id = $license->id;

                    // Only assign the first seat to the user
                    if ($x==0) {
                        $license_seat->assigned_to = $user->id;
                    } else {
                        $license_seat->assigned_to = null;
                    }

                    if ($license_seat->save()) {
                        $license_seat_created++;
                    }
                }

                if ($license_seat_created > 0) {
                    $this->comment($license_seat_created.' seats were created');
                } else {
                    $this->comment('Something went wrong! NO seats for '.$user_license_name.' were created');
                }



            } else {
                $this->comment('Something went wrong! License '.$user_license_name.' was NOT created');
            }


            $this->comment('=====================================');

            return true;

        });


    }

    /**
     * Get the console command arguments.
     *
     * @return array
     */
    protected function getArguments()
    {
        return array(
            array('filename', InputArgument::REQUIRED, 'File for the CSV import.'),
        );
    }


    /**
     * Get the console command options.
     *
     * @return array
     */
    protected function getOptions()
    {
        return array(
            array('domain', null, InputOption::VALUE_REQUIRED, 'Email domain for generated email addresses.', null),
            array('email_format', null, InputOption::VALUE_REQUIRED, 'The format of the email addresses that should be generated. Options are firstname.lastname, firstname, filastname', null),
            array('testrun', null, InputOption::VALUE_REQUIRED, 'Test the output without writing to the database or not.', null),
        );
    }
}