add remove driver function

models/driver.go

@@ -38,7 +38,8 @@ next:
 	}
 }
 
-func (d *Driver) RemoveBus(bus *Bus) {
+func (d *Driver) RemoveBuses(buses []*Bus) {
+	for _, bus := range buses {
 		for _, currentBus := range d.Buses {
 			if currentBus.Name != bus.Name {
 				continue
@@ -69,6 +70,7 @@ func (d *Driver) RemoveBus(bus *Bus) {
 			}
 		}
 	}
+}
 
 func (d *Driver) GetBus(name string) *Bus {
 	for _, b := range d.Buses {

models/drivers.go

@@ -19,3 +19,11 @@ func (d *Drivers) AddDriver(drv *Driver) {
 	}
 	(*d)[drv.Type] = drv
 }
+
+func (d *Drivers) RemoveDriver(drv *Driver) {
+	if driver, ok := (*d)[drv.Type]; ok {
+		driver.RemoveBuses(drv.Buses)
+		(*d)[drv.Type] = driver
+		return
+	}
+}