package resolver

import (
	"context"
	"fmt"
	"net"
	"strings"
)

import (
	"github.com/eyedeekay/sam3"
)

type Resolver struct {
	*sam3.SAMResolver
	allowedSuffixes []string
}

func (r Resolver) Resolve(ctx context.Context, name string) (context.Context, net.Addr, error) {
	return r.ResolveI2P(ctx, name)
}

func (r Resolver) ResolveI2P(ctx context.Context, name string) (context.Context, *sam3.I2PAddr, error) {
	if r.ValidateI2PAddr(name) {
		return ctx, nil, fmt.Errorf("Error, not an allowed suffix")
	}
	raddr, err := r.SAMResolver.Resolve(name)
	if err != nil {
		return ctx, nil, err
	}
	return ctx, &raddr, nil
}

func (r Resolver) ValidateI2PAddr(name string) bool {
	noi2p := true
	for _, suffix := range r.allowedSuffixes {
		if strings.HasSuffix(name, suffix) {
			if suffix == ".b32.i2p" {
				if len(name) != 52 {
					noi2p = true
					break
				}
			}
			noi2p = false
		}
	}
	return noi2p
}

func NewResolver() (*Resolver, error) {
	return NewResolverFromOptions()
}

func NewResolverFromOptions() (*Resolver, error) {
	var r Resolver
	r.allowedSuffixes = []string{".i2p", ".b32.i2p"}
	var err error
	r.SAMResolver, err = sam3.NewFullSAMResolver("127.0.0.1:7656")
	if err != nil {
		return nil, err
	}
	return &r, nil
}