Skip to content

Commit a98faac

Browse files
committed
Converted router check to a regex
1 parent 90df91c commit a98faac

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

internal/cloning/cloning_service.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@ import (
55
"fmt"
66
"log"
77
"os"
8-
"strings"
8+
"regexp"
99
"time"
1010

1111
"github.com/cpp-cyber/proclone/internal/ldap"
@@ -85,11 +85,11 @@ func (cs *CloningService) CloneTemplate(req CloneRequest) error {
8585
// 3. Identify router and other VMs
8686
var router *proxmox.VM
8787
var templateVMs []proxmox.VM
88+
routerPattern := regexp.MustCompile(`(?i)(router|pfsense|vyos)`)
8889

8990
for _, vm := range templatePool {
9091
// Check to see if this VM is the router
91-
lowerVMName := strings.ToLower(vm.Name)
92-
if strings.Contains(lowerVMName, "router") || strings.Contains(lowerVMName, "pfsense") || strings.Contains(lowerVMName, "vyos") {
92+
if routerPattern.MatchString(vm.Name) {
9393
router = &proxmox.VM{
9494
Name: vm.Name,
9595
Node: vm.NodeName,

0 commit comments

Comments
 (0)