package peerfinder

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

type Time time.Time

func (t *Time) UnmarshalJSON(b []byte) error {
	time, err := time.Parse(`"2006-01-02 15:04:05"`, string(b))
	*t = Time(time)
	return err
}
func (t *Time) MarshalJSON() ([]byte, error) {
	if t == nil {
		return nil, nil
	}
	i := *t
	return time.Time(i).MarshalJSON()
}

type ipFamily string

const (
	ipFamilyV4   ipFamily = "IPv4"
	ipFamilyV6   ipFamily = "IPv6"
	ipFamilyBoth ipFamily = "both"
	ipFamilyNone ipFamily = "none"
)

func (t *ipFamily) UnmarshalJSON(b []byte) error {
	i, err := strconv.Atoi(strings.Trim(string(b), `"`))
	switch i {
	case 1:
		*t = ipFamilyV4
	case 2:
		*t = ipFamilyV6
	case 3:
		*t = ipFamilyBoth
	default:
		*t = ipFamilyNone
	}
	return err
}

type peerType []string

func (t *peerType) UnmarshalJSON(b []byte) error {
	*t = strings.Split(strings.Trim(string(b), `"`), ",")
	return nil
}

type Peer struct {
	ID      string   `json:"peer_id,omitempty"`
	Owner   string   `json:"peer_owner"`
	Nick    string   `json:"peer_nick"`
	Name    string   `json:"peer_name"`
	Country string   `json:"peer_country"`
	Note    string   `json:"peer_note"`
	Family  ipFamily `json:"peer_family"`
	Type    peerType `json:"peer_type"`
	Created Time     `json:"peer_created"`
}

func (p *Peer) CanSupport(ip string) bool {
	addr := net.ParseIP(ip)
	if addr == nil {
		return false
	}
	if !addr.IsGlobalUnicast() {
		return false
	}

	switch p.Family {
	case ipFamilyV4:
		return addr.To4() != nil
	case ipFamilyV6:
		return addr.To16() != nil
	case ipFamilyNone:
		return false
	}

	return true
}