package chapter9;

import java.util.*;

import java.util.function.*;

import java.time.*;

enum STATUS {ACTIVE, SUSPENDED}

class Vehicle

{

String make;

String model;

int year;

public Vehicle(String ma, String mo, int y)

{

make = ma;

model = mo;

year = y;

}

@Override

public String toString()

{

return year + " " + make + " " + model;

}

}

class License

{

LocalDate expDate;

String insurance;

STATUS status;

ArrayList<Vehicle> vehicles;

public License(LocalDate e, String i, Vehicle... v)

{

expDate = e;

insurance = i;

status = STATUS.ACTIVE;

vehicles = new ArrayList<>();

for (Vehicle r : v)

vehicles.add(r);

}

public License(LocalDate e, String i, STATUS s,

ArrayList<Vehicle> vs)

{

expDate = e;

insurance = i;

status = s;

vehicles = new ArrayList<>(vs);

}

public License(License l)

{

expDate = l.expDate;

insurance = l.insurance;

status = l.status;

vehicles = new ArrayList<>(l.vehicles);

}

@Override

public String toString()

{

return expDate + " " + insurance + " " + status

+ " " + vehicles;

}

}

public class DMV

{

public static void main(String[] args)

{

Map<String,License> d = new TreeMap<>();

System.out.println(

"Welcome to the Department of Motor Vehicles.");

Supplier<Integer> selectOperation = () -> {

int selection = -1;

Scanner input = new Scanner(System.in);

while (selection < 0 || selection > 7)

{

System.out.println("Select and Operation:");

System.out.println(

" 0 - Add Driver 1 - Remove Driver 2 - Change Ins");

System.out.println(

" 3 - Suspend Lic 4 - Renew License 5 - Add Veh");

System.out.print (

" 6 - Remove Veh 7 - Quit:");

selection = Integer.parseInt(input.nextLine());

if (selection < 0 || selection > 7)

System.out.println("Imvalid selection");

}

return selection;

};

Supplier<String> licenseId = () -> {

Scanner input = new Scanner(System.in);

System.out.print("Enter licenseId:");

return input.nextLine();

};

Supplier<String> insurance = () -> {

Scanner input = new Scanner(System.in);

System.out.print("Enter insurance:");

return input.nextLine();

};

Supplier<LocalDate> expDate = () -> {

Scanner input = new Scanner(System.in);

System.out.print("Enter year:");

int year = Integer.parseInt(input.nextLine());

System.out.print("Enter month:");

Month month = Month.valueOf(input.nextLine());

System.out.print("Enter day:");

int day = Integer.parseInt(input.nextLine());

return LocalDate.of(year, month, day);

};

Supplier<Vehicle> vehicle = () -> {

Scanner input = new Scanner(System.in);

System.out.print("Enter make:");

String make = input.nextLine();

System.out.print("Enter model:");

String model = input.nextLine();

System.out.print("Enter year:");

int year = Integer.parseInt(input.nextLine());

return new Vehicle(make, model, year);

};

int operation = 0;

while (operation != 7)

{

operation = selectOperation.get();

switch (operation)

{

case 0: // Add Driver

BiFunction<License,License,License> addDriver

= (ov,nv) -> ov;

System.out.println(

d.merge(licenseId.get(),

new License(expDate.get(), null),

addDriver));

break;

case 1: // Remove Driver

BiFunction<String,License,License> remove

= (k,v) -> null;

System.out.println(

d.computeIfPresent(licenseId.get(),remove));

break;

case 2: // Change Insurance

BiFunction<License,License,License> changeInsurance

= (ov,nv) -> new License(ov.expDate,

nv.insurance,

ov.status,

ov.vehicles);

System.out.println(

d.merge(licenseId.get(),

new License(null, insurance.get()),

changeInsurance));

break;

case 3: // Suspend License

BiFunction<String,License,License> suspend

= (k,v) -> new License(v.expDate, v.insurance,

STATUS.SUSPENDED,

v.vehicles);

System.out.println(

d.computeIfPresent(licenseId.get(),suspend));

break;

case 4: // Renew License

BiFunction<License,License,License> renew

= (ov,nv) -> new License(nv.expDate,

ov.insurance,

STATUS.ACTIVE,

ov.vehicles);

System.out.println(

d.merge(licenseId.get(),

new License(expDate.get(), null),

renew));

break;

case 5: // Add Vehicle

BiFunction<License,License,License>

addVehicle = (ov,nv) -> {

License temp = new License(ov);

temp.vehicles.add(nv.vehicles.get(0));

return temp;

};

System.out.println(

d.merge(licenseId.get(),

new License(null, null,vehicle.get()),

addVehicle));

break;

case 6: // Remove Vehicle

BiFunction<License,License,License>

removeVehicle = (ov,nv) -> {

License temp = new License(ov);

Vehicle r = nv.vehicles.get(0);

temp.vehicles.removeIf(

x -> x.make.equals(r.make)

&& x.model.equals(r.model)

&& x.year == r.year);

return temp;

};

System.out.println(

d.merge(licenseId.get(),

new License(null, null,vehicle.get()),

removeVehicle));

}

}

}

}