summaryrefslogtreecommitdiff
path: root/domain.go
blob: 252cab4b04f17f90544c6aa3101acbffbb1df206 (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
package main

import (
	"log"
	"net"
	"reflect"
	"strings"
)

func (vD VirtualDomain) DomainExists() bool {
	query := Db.Where("name = ?", vD.Name).Find(&vD)
	if query.Error != nil {
		log.Println("Info: Query error.", query.Error)
		return false
	}
	if vD.Name == "" {
		return false
	}
	return true
}

func (vD VirtualDomain) CreateDomain() bool {
	if !Db.NewRecord(vD) {
		log.Println("Info: Creating new record failed.", vD.Name)
		return false
	}
	query := Db.Create(&vD)
	if query.Error != nil {
		log.Println("Info: Query error.", query.Error)
		return false
	}
	return true
}

func (vD VirtualDomain) ValidateDomain(ref string) bool {
	log.Println("Info: Validating " + vD.Name)
	if vD.Name == "" {
		return false
	}
	addr, err := net.LookupIP(vD.Name)
	if err != nil {
		log.Println("Info: ", vD.Name+" "+err.Error())
		return false
	}
	if len(addr) == 0 {
		return false
	}
	serverIP, err := net.LookupIP(ref)
	if err != nil {
		log.Println("Info: ", ref+" "+err.Error())
		return false
	}
	return reflect.DeepEqual(serverIP, addr)
}
func (vD VirtualDomain) ValidateDomainMX(ref string) bool {
	log.Println("Info: Validating MX " + vD.Name)
	if vD.Name == "" {
		log.Println("Info: Empty domain.")
		return false
	}
	mx, err := net.LookupMX(vD.Name)
	if err != nil {
		log.Println("Info: Lookup error " + vD.Name + " " + err.Error())
		return false
	}
	match := false

	serverIP, err := net.LookupIP(ref)
	if err != nil {
		log.Println("Info: Lookup error for server " + ref + " " + err.Error())
		return false
	}
	for _, v := range mx {
		mxIP, err := net.LookupIP(v.Host)
		if err != nil {
			log.Println("Info: ", err)
		}
		if reflect.DeepEqual(serverIP, mxIP) {
			match = true
		}
	}
	return match
}

func GetDomain(email string) string {
	fragments := strings.Split(email, "@")
	return fragments[len(fragments)-1]
}

func (vD VirtualDomain) GetPrimaryKey() int64 {
	Db.Find(&vD)
	return vD.Id
}