Как я могу создать файл с именами интерфейсов в определенных частях и затем проанализировать их из другого файла, чтобы изменить интерфейс с данными в GO? - PullRequest
0 голосов
/ 15 октября 2018

Я создаю программу для изменения IP-адреса машины в Ubuntu 18.04.1 LTS.Поэтому в основном я создаю шаблон файла YAML с

network:
   ethernets:
      {INTERFACE}:
          dhcp4: no
          addresses: [{IP}/24]
          gateway4: {GATEWAY}
          optional: true
          nameservers:
           addresses: [{DNS}]
     {INTERFACE2}:
          dhcp4: no
          addresses: [{IP2}/24]
          gateway4: {GATEWAY2}
          optional: true
          nameservers:
           addresses: [{DNS2}]

Теперь мне нужно отсканировать этот файл шаблона и добавить значения из другого файла либо с помощью вызова JSON, либо в качестве параметров в CLI.

Таким образом, приложение примет файл шаблона, который был предварительно подготовлен.Например, шаблон примера для файла yaml.

Для этого файла шаблона потребуется 3 параметра: файл перевода (JSON) и выходные данные

, файл перевода также может быть проанализирован в командной строке.

Он основан на 2 столбцах, поисковой переменной, такой как IP, и ее новом значении. Утилита выполнит поиск переменной {IP} и заменит ее значением второго столбца.

НаCLI может выглядеть так:

. / Searchreplace -intemplate = 'templateip.yaml' -translation = '{IP}, xxx.xxx;{SUBNET}, yyy.yyy.yyy.y} -outfile = / tmp / newipfile.yaml

Мой текущий код жестко кодирует данные и меняет IP, но я хочу, чтобы он был в отдельномфайл шаблона.Это мой текущий код

    package main

import (
    "bufio"
    "fmt"
    "log"
    "net"
    "os"
    "os/exec"
    "regexp"
    "strings"
)

var pathdelete = "/etc/netplan/01-network-manager-all.yaml"
var ipv4Regex = regexp.MustCompile(`^(?:[0-9]{1,3}\.){3}[0-9]{1,3}/[0-9]+$`)
var gatewayRegex = regexp.MustCompile(`^(?:[0-9]{1,3}\.){3}[0-9]{1,3}$`)
var nsRegex = regexp.MustCompile(`^(?:[0-9]\.){3}[0-9]{1,3}\,(?:[0-9]\.){3}[0-9]{1,3}$`)
var nicRegex = regexp.MustCompile(`e([a-z]+)`)

func main() {
    reader := bufio.NewReader(os.Stdin)
    fmt.Print("Enter the IP address: ")
    IPaddress, _ := reader.ReadString('\n')
    IPaddress = strings.TrimSuffix(IPaddress, "\n")

    //sanity check
    testInput := net.ParseIP(IPaddress)
    if testInput.To4() == nil {
        fmt.Printf("%v is not a valid IPv4 address\n", testInput)
    } else {
        fmt.Printf("It is a valid IP address\n")
        writeFile()
        readFile()
    }
}
func writeFile() {
    err01 := os.Truncate(pathdelete, 0)
    if err01 != nil {
        log.Fatal(err01)
    }

    //reading Network Cards
    reader := bufio.NewReader(os.Stdin)
    fmt.Print("Enter the Ethernet Card (for example : 'ens0N' or 'eth0') ")
    nic, _ := reader.ReadString('\n')
    nic = strings.TrimSuffix(nic, "\n")
    if nicRegex.MatchString(nic) {
    } else {
        log.Fatal("Please check the name of the NIC Card you have entered")
        os.Exit(1)
    }

    //reading IP Address and DNS
    fmt.Print("Enter the IP address along with the DNS in the format xxx.xxx.x.xxx/yy ")
    IP, _ := reader.ReadString('\n')
    IP = strings.TrimSuffix(IP, "\n")

    if ipv4Regex.MatchString(IP) {
    } else {
        log.Fatal("Please check the IP address you have entered")
        os.Exit(1)
    }

    //reading Gateway
    fmt.Print("Enter the gateway in the format xxx.xxx.x.x ")
    gateway, _ := reader.ReadString('\n')
    gateway = strings.TrimSuffix(gateway, "\n")
    if gatewayRegex.MatchString(gateway) {
    } else {
        log.Fatal("Please check the Gateway you have entered")
        os.Exit(1)
    }

    //reading Name servers
    fmt.Print("Enter the nameservers seperated by comma(,) in the format x.x.x.x,y.y.y.y ")
    ns, _ := reader.ReadString('\n')
    ns = strings.TrimSuffix(ns, "\n")
    if nsRegex.MatchString(ns) {
    } else {
        log.Fatal("Please check the Name Servers you have entered")
        os.Exit(1)
    }

    // open file using READ & WRITE permission
    var file, err = os.OpenFile(pathdelete, os.O_RDWR, 0644)
    checkError(err)
    defer file.Close()

    // write some text to file
    _, err = file.WriteString("# Let NetworkManager manage all devices on this system\nnetwork:\n  version: 2\n  renderer: NetworkManager\n")
    if err != nil {
        fmt.Println(err.Error())
        return //must return here for defer statements to be called
    }
    _, err = file.WriteString("  ethernets:\n    " + nic + ":\n      dhcp4: no\n      dhcp6: no\n")
    if err != nil {
        fmt.Println(err.Error())
        return //same as above
    }
    _, err = file.WriteString("      addresses: [" + IP + "]\n      gateway4: " + gateway + "\n      nameservers:\n        addresses: [" + ns + "]")
    if err != nil {
        fmt.Println(err.Error())
        return //same as above
    }

    // save changes
    err = file.Sync()
    if err != nil {
        fmt.Println(err.Error())
        return //same as above
    }
    applyCmd := exec.Command("bash", "-c", "sudo netplan apply")
    _, err2 := applyCmd.Output()
    if err2 != nil {
        panic(err2)
    }
}

func readFile() {
    // re-open file
    var file, err = os.OpenFile(pathdelete, os.O_RDWR, 0644)
    checkError(err)
    defer file.Close()

    // read file
    var text = make([]byte, 1024)
    n, err := file.Read(text)
    if n > 0 {
        fmt.Println("\nThe data you have entered is \n ")
        fmt.Println(string(text))
    }
    //if there is an error while reading
    //just print however much was read if any
    //at return file will be closed
}

func checkError(err error) {
    if err != nil {
        fmt.Println(err.Error())
        os.Exit(1)
    }
}

1 Ответ

0 голосов
/ 22 октября 2018

Мне удалось выполнить это путем преобразования ввода командной строки

. / Program --xyz.yaml - {interface} eth0 - IP-адрес 192.168.1.233/24 - {gateway} 192.168.0.1 --DNS 8.8.3.8,8.8.8.3

Я использовал функцию отображения для сопоставления CLI со строками с этим

func manyRandomArg() map[string]string {
    rv := make(map[string]string)
    for ix, x := range os.Args {
        if x[:2] == "--" {
            rv[x] = os.Args[ix+1]
        }
    }
    return rv
}

Тогда я сортировал толькоключи и выполнил функцию поиска замены

keys := make([]string, 0, len(rv))
    for key := range rv {
        keys = append(keys, key)
    }
    sort.Strings(keys) //sort keys alphabetically

пришлось обрезать часть '-' в ключе

key0 = trimLeftChars(keys[0], 2)

Использовалась функция замены строки

r := strings.NewReplacer(key0, rv[keys[0]], key1, rv[keys[1]], key2, rv[keys[2]], key3, rv[keys[3]], key4, rv[keys[4]])
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...