add remove buses when empty

models/driver.go

@@ -1,6 +1,8 @@
 package models
 
-import "slices"
+import (
+	"slices"
+)
 
 // driver model
 type Driver struct {
@@ -38,34 +40,38 @@ next:
 	}
 }
 
-func (d *Driver) RemoveBus(bus *Bus) {
-	for _, currentBus := range d.Buses {
-		if currentBus.Name != bus.Name {
-			continue
-		}
+func (d *Driver) RemoveBuses(buses []*Bus) {
+	for _, bus := range buses {
+		for _, currentBus := range d.Buses {
+			if currentBus.Name != bus.Name {
+				continue
+			}
 
-		currentBus.RemoveAddress(bus.Address...)
-		if bus.Topic != nil {
-			currentBus.RemoveSubscription(bus.Topic.Subscribe...)
-			currentBus.RemovePublish(bus.Topic.Publish...)
-			if len(currentBus.Address) == 0 {
-				var remove bool = true
-				if currentBus.Topic != nil {
-					remove = len(currentBus.Topic.Subscribe) == 0 && len(currentBus.Topic.Publish) == 0
-				}
-				if remove {
-					if len(d.Buses) == 1 {
-						d.Buses = []*Bus{}
-						return
+			currentBus.RemoveAddress(bus.Address...)
+
+			if bus.Topic != nil {
+				currentBus.RemoveSubscription(bus.Topic.Subscribe...)
+				currentBus.RemovePublish(bus.Topic.Publish...)
+
+				if len(currentBus.Address) == 0 {
+					var remove bool = true
+					if currentBus.Topic != nil {
+						remove = len(currentBus.Topic.Subscribe) == 0 && len(currentBus.Topic.Publish) == 0
 					}
-					for i, b := range d.Buses {
-						if bus.Name == b.Name {
-							d.Buses = slices.Delete(d.Buses, i, i+1)
+					if remove {
+						if len(d.Buses) == 1 {
+							d.Buses = []*Bus{}
+							return
+						}
+						for i, b := range d.Buses {
+							if bus.Name == b.Name {
+								d.Buses = slices.Delete(d.Buses, i, i+1)
+							}
 						}
 					}
 				}
+				break
 			}
-			break
 		}
 	}
 }

models/drivers.go

@@ -19,3 +19,17 @@ 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)
+		if len(driver.Buses) == 1 {
+			if len(driver.Buses[0].Address) == 0 && driver.Buses[0].Topic == nil {
+				(*d)[drv.Type] = &Driver{}
+				return
+			}
+		}
+		(*d)[drv.Type] = driver
+		return
+	}
+}

models/topic.go

@@ -1,6 +1,8 @@
 package models
 
-import "slices"
+import (
+	"slices"
+)
 
 // topic model
 type Topic struct {
@@ -33,6 +35,7 @@ func (t *Topic) RemoveSubscription(subs ...string) {
 		t.Subscribe = []string{}
 		return
 	}
+
 	for _, toDelete := range subs {
 		for i, a := range t.Subscribe {
 			if a == toDelete {