package hysteria

import (
	"net"
	"os"
	"syscall"

	"github.com/sagernet/quic-go"
	"github.com/sagernet/sing-box/common/baderror"
	"github.com/sagernet/sing/common"
)

type PacketConnWrapper struct {
	net.PacketConn
}

func (c *PacketConnWrapper) SetReadBuffer(bytes int) error {
	return common.MustCast[*net.UDPConn](c.PacketConn).SetReadBuffer(bytes)
}

func (c *PacketConnWrapper) SetWriteBuffer(bytes int) error {
	return common.MustCast[*net.UDPConn](c.PacketConn).SetWriteBuffer(bytes)
}

func (c *PacketConnWrapper) SyscallConn() (syscall.RawConn, error) {
	return common.MustCast[*net.UDPConn](c.PacketConn).SyscallConn()
}

func (c *PacketConnWrapper) File() (f *os.File, err error) {
	return common.MustCast[*net.UDPConn](c.PacketConn).File()
}

func (c *PacketConnWrapper) Upstream() any {
	return c.PacketConn
}

type StreamWrapper struct {
	Conn quic.Connection
	quic.Stream
}

func (s *StreamWrapper) Read(p []byte) (n int, err error) {
	n, err = s.Stream.Read(p)
	return n, baderror.WrapQUIC(err)
}

func (s *StreamWrapper) Write(p []byte) (n int, err error) {
	n, err = s.Stream.Write(p)
	return n, baderror.WrapQUIC(err)
}

func (s *StreamWrapper) LocalAddr() net.Addr {
	return s.Conn.LocalAddr()
}

func (s *StreamWrapper) RemoteAddr() net.Addr {
	return s.Conn.RemoteAddr()
}

func (s *StreamWrapper) Upstream() any {
	return s.Stream
}

func (s *StreamWrapper) Close() error {
	s.CancelRead(0)
	s.Stream.Close()
	return nil
}