package deliver

import (
	"time"

	"nanomsg.org/go-mangos"
)

// maxRecvSize max recv size
var (
	maxRecvSize  = 33 * 1024 * 1024
	surveyorTime = 0
	timeout      = time.Duration(25)
)

func setTimeOut(tm int) {
	timeout = time.Duration(tm)
}

func optDefault() map[string]interface{} {

	options := make(map[string]interface{})

	options[mangos.OptionMaxRecvSize] = maxRecvSize
	options[mangos.OptionWriteQLen] = 0
	options[mangos.OptionReadQLen] = 0
	options[mangos.OptionRecvDeadline] = timeout * time.Millisecond
	options[mangos.OptionSendDeadline] = timeout * time.Millisecond
	// options[mangos.OptionRaw] = true

	return options
}

func optRaw(options map[string]interface{}) map[string]interface{} {

	options[mangos.OptionRaw] = true
	return options
}

func optOther(p protocol, options map[string]interface{}, o ...interface{}) map[string]interface{} {
	// len 1 or 0, sub topic
	if p == SUB {
		if len(o) == 1 {
			v := o[0]
			if t, ok := v.(string); ok {
				options[mangos.OptionSubscribe] = []byte(t)
			}
		} else {
			options[mangos.OptionSubscribe] = []byte("")
		}
	}

	// len 1 or 0, surveyor time
	if p == SURVEYOR {
		if len(o) == 1 {
			v := o[0]
			if t, ok := v.(int); ok {
				surveyorTime = t / 2
				if surveyorTime == 0 {
					surveyorTime = 1
				}
			}
		}
		options[mangos.OptionSurveyTime] = time.Duration(surveyorTime) * time.Second
	}
	return options
}

func setSocketOptions(sock mangos.Socket, options map[string]interface{}) error {

	for k, v := range options {
		if err := sock.SetOption(k, v); err != nil {
			return err
		}
	}
	return nil
}