aboutsummaryrefslogtreecommitdiff
path: root/six/helpers.go
blob: 694a79e323b0880961a5d24e4adfd1f4775e7ee9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
// SPDX-License-Identifier: GPL-2.0-only
// Copyright (C) 2020 Michael Crute <mike@crute.us>. All rights reserved.
//
// Use of this source code is governed by a license that can be found in the
// LICENSE file.

package six

import (
	"net"
	"strconv"
	"strings"
	"time"
)

func mustParseInt(a string) int {
	a = strings.TrimSpace(a)

	if a == "" {
		return 0
	}

	i, err := strconv.Atoi(a)
	if err != nil {
		panic(err)
	}

	return i
}

func parseYesNo(b string) bool {
	return strings.ToLower(strings.TrimSpace(b)) == "yes"
}

func parseIP(i string) *net.IP {
	o := net.ParseIP(i)
	return &o
}

func parseIPNetFromCIDR(i string) *net.IPNet {
	_, ipnet, _ := net.ParseCIDR(strings.TrimSpace(i))
	return ipnet
}

func mustParseTime(t string) *time.Time {
	t = strings.TrimSpace(t)

	if t == "" {
		return nil
	}

	i, err := time.Parse("2006-01-02", t)
	if err != nil {
		panic(err)
	}

	return &i
}

func mustParseLongTime(t string) *time.Time {
	t = strings.TrimSpace(t)

	if t == "" {
		return nil
	}

	i, err := time.Parse("2006-01-02 15:04:05 MST", t)
	if err != nil {
		panic(err)
	}

	return &i
}

func mustParseLongTimeNoZone(t string) *time.Time {
	t = strings.TrimSpace(t)

	if t == "" {
		return nil
	}

	i, err := time.Parse("2006-01-02 15:04:05", t)
	if err != nil {
		panic(err)
	}

	return &i
}

func allEmpty(d []string) bool {
	for _, v := range d {
		if strings.TrimSpace(v) != "" {
			return false
		}
	}
	return true
}

func parseASPath(p string) []int {
	out := []int{}

	for _, i := range strings.Split(p, " ") {
		ii, err := strconv.Atoi(strings.TrimSpace(i))
		// Some AS paths contain { and } which are not valid and need to be discard
		if err != nil {
			continue
		}
		out = append(out, ii)
	}

	return out
}