package softbus

/*
#include <stdlib.h>
#include "libcsoftbus.h"
*/
import "C"
import (
	"errors"
	"fmt"
	"unsafe"
)

const (

	// RETVAL retual val
	RETVAL = 0x7fff
	// PullPush mode
	PullPush = 1
	// ReqRep mode
	ReqRep = 2
	// Pair mode
	Pair = 3
	// PubSub mode
	PubSub = 4
	// Survey mode
	Survey = 5
	// Bus mode
	Bus = 6
)

// Socket struct
type Socket struct {
	csocket unsafe.Pointer
}

// OpenSocket open
func OpenSocket(mod int) *Socket {
	if libsoftbus == nil {
		return nil
	}

	s := C.wrap_fn_socket_open(libsoftbus, C.int(mod))
	if s == nil {
		return nil
	}

	return &Socket{
		csocket: s,
	}
}

// Close socket
func (s *Socket) Close() int {
	if libsoftbus != nil || s.csocket != nil {
		r := C.wrap_fn_socket_close(libsoftbus, s.csocket)
		return int(r)
	}
	return RETVAL
}

// Bind socket
func (s *Socket) Bind(port int) int {
	if libsoftbus == nil {
		return RETVAL
	}

	r := C.wrap_fn_socket_bind(libsoftbus, s.csocket, C.int(port))
	return int(r)
}

// Listen socket
func (s *Socket) Listen() int {
	if libsoftbus == nil {
		return RETVAL
	}
	r := C.wrap_fn_socket_listen(libsoftbus, s.csocket)
	return int(r)
}

// Connect socket
func (s *Socket) Connect(port int) int {
	if libsoftbus == nil {
		return RETVAL
	}

	r := C.wrap_fn_socket_connect(libsoftbus, s.csocket, C.int(port))
	return int(r)
}

// Send data
func (s *Socket) Send(data []byte) int {
	if libsoftbus == nil {
		return RETVAL
	}

	r := C.wrap_fn_socket_send(libsoftbus, s.csocket, unsafe.Pointer(&data[0]), C.int(len(data)))
	return int(r)
}

// Recv data
func (s *Socket) Recv() ([]byte, error) {
	if libsoftbus == nil {
		return nil, errors.New("libsoftbus is nil")
	}

	var rd unsafe.Pointer
	var rl C.int
	r := C.wrap_fn_socket_recv(libsoftbus, s.csocket, &rd, &rl)
	if r != 0 {
		return nil, fmt.Errorf("Recv Data Failed errCode: %d", int(r))
	}
	data := C.GoBytes(rd, rl)
	C.wrap_fn_socket_buf_free(libsoftbus, rd)
	return data, nil
}

// Port get port/key
func (s *Socket) Port() int {
	if libsoftbus == nil {
		return -1
	}

	r := C.wrap_fn_socket_port(libsoftbus, s.csocket)

	return int(r)
}