//
// Copyright 2025 Staysail Systems, Inc. <info@staysail.tech>
//
// This software is supplied under the terms of the MIT License, a
// copy of which should be located in the distribution where this
// file was obtained (LICENSE.txt).  A copy of the license may also be
// found online at https://opensource.org/licenses/MIT.
//

#ifndef CORE_STREAM_H
#define CORE_STREAM_H

// This provides an abstraction for byte streams, allowing polymorphic
// use of them in rather flexible contexts.

#include "defs.h"

// Private property operations (these include the types.)
extern nng_err nni_stream_get(
    nng_stream *, const char *, void *, size_t *, nni_type);
extern nng_err nni_stream_set(
    nng_stream *, const char *, const void *, size_t, nni_type);

extern nng_err nni_stream_dialer_get(
    nng_stream_dialer *, const char *, void *, size_t *, nni_type);
extern nng_err nni_stream_dialer_set(
    nng_stream_dialer *, const char *, const void *, size_t, nni_type);
extern nng_err nni_stream_dialer_set_tls(
    nng_stream_dialer *, nng_tls_config *);
extern nng_err nni_stream_dialer_get_tls(
    nng_stream_dialer *, nng_tls_config **);

extern nng_err nni_stream_listener_get(
    nng_stream_listener *, const char *, void *, size_t *, nni_type);
extern nng_err nni_stream_listener_set(
    nng_stream_listener *, const char *, const void *, size_t, nni_type);
extern nng_err nni_stream_listener_set_tls(
    nng_stream_listener *, nng_tls_config *);
extern nng_err nni_stream_listener_get_tls(
    nng_stream_listener *, nng_tls_config **);

// This is the common implementation of a connected byte stream.  It should be
// the first element of any implementation.  Applications are not permitted to
// access it directly.
struct nng_stream {
	void (*s_free)(void *);
	void (*s_close)(void *);
	void (*s_stop)(void *);
	void (*s_recv)(void *, nng_aio *);
	void (*s_send)(void *, nng_aio *);
	nng_err (*s_get)(void *, const char *, void *, size_t *, nni_type);
	nng_err (*s_set)(void *, const char *, const void *, size_t, nni_type);
	nng_err (*s_peer_cert)(void *, nng_tls_cert **);
};

// Dialer implementation.  Stream dialers create streams.
struct nng_stream_dialer {
	void (*sd_free)(void *);
	void (*sd_close)(void *);
	void (*sd_stop)(void *);
	void (*sd_dial)(void *, nng_aio *);
	nng_err (*sd_get)(void *, const char *, void *, size_t *, nni_type);
	nng_err (*sd_set)(
	    void *, const char *, const void *, size_t, nni_type);
	nng_err (*sd_get_tls)(void *, nng_tls_config **);
	nng_err (*sd_set_tls)(void *, nng_tls_config *);
};

// Listener implementation.  Stream listeners accept connections and create
// streams.
struct nng_stream_listener {
	void (*sl_free)(void *);
	void (*sl_close)(void *);
	void (*sl_stop)(void *);
	nng_err (*sl_listen)(void *);
	void (*sl_accept)(void *, nng_aio *);
	nng_err (*sl_get)(void *, const char *, void *, size_t *, nni_type);
	nng_err (*sl_set)(
	    void *, const char *, const void *, size_t, nni_type);
	nng_err (*sl_get_tls)(void *, nng_tls_config **);
	nng_err (*sl_set_tls)(void *, nng_tls_config *);
	nng_err (*sl_set_security_descriptor)(void *, void *);
};

#endif // CORE_STREAM_H