package internet

import (
	"context"

	"v2ray.com/core/common/net"
)

type key int

const (
	streamSettingsKey key = iota
	dialerSrcKey
	transportSettingsKey
	securitySettingsKey
)

func ContextWithStreamSettings(ctx context.Context, streamSettings *StreamConfig) context.Context {
	return context.WithValue(ctx, streamSettingsKey, streamSettings)
}

func StreamSettingsFromContext(ctx context.Context) *StreamConfig {
	ss := ctx.Value(streamSettingsKey)
	if ss == nil {
		return nil
	}
	return ss.(*StreamConfig)
}

func ContextWithDialerSource(ctx context.Context, addr net.Address) context.Context {
	return context.WithValue(ctx, dialerSrcKey, addr)
}

func DialerSourceFromContext(ctx context.Context) net.Address {
	if addr, ok := ctx.Value(dialerSrcKey).(net.Address); ok {
		return addr
	}
	return net.AnyIP
}

func ContextWithTransportSettings(ctx context.Context, transportSettings interface{}) context.Context {
	return context.WithValue(ctx, transportSettingsKey, transportSettings)
}

func TransportSettingsFromContext(ctx context.Context) interface{} {
	return ctx.Value(transportSettingsKey)
}

func ContextWithSecuritySettings(ctx context.Context, securitySettings interface{}) context.Context {
	return context.WithValue(ctx, securitySettingsKey, securitySettings)
}

func SecuritySettingsFromContext(ctx context.Context) interface{} {
	return ctx.Value(securitySettingsKey)
}