#include <winsock2.h>
// #include <Windows.h>
#include <openssl/ssl.h>
#include <openssl/err.h>
#include <openssl/applink.c>

#define zero {}
#define func static

#include "generated\generated_main.cpp"

int main()
{
    WSADATA wsa_data = zero;
    int err = 0;
    err = WSAStartup(MAKEWORD(2, 2), &wsa_data);
    if(err != 0) {
        error("BAD\n");
    }

    SSL_library_init();
    SSL_load_error_strings();
    OpenSSL_add_all_algorithms();

    SSL_CTX* ctx = SSL_CTX_new(TLS_server_method());
    if(!ctx) {
        error("BAD ctx\n");
    }

    if(SSL_CTX_use_certificate_file(ctx, "localhost.crt", SSL_FILETYPE_PEM) <= 0) {
        ERR_print_errors_fp(stderr);
        error("bad cert");
    }

    if(SSL_CTX_use_PrivateKey_file(ctx, "localhost.key", SSL_FILETYPE_PEM) <= 0) {
        ERR_print_errors_fp(stderr);
        error("bad key");
    }

    SOCKET server_fd = socket(AF_INET, SOCK_STREAM, 0);
    if(server_fd == INVALID_SOCKET) {
        error("failed to create socket");
    }

    sockaddr_in server_addr = zero;
    server_addr.sin_family = AF_INET;
    server_addr.sin_addr.s_addr = INADDR_ANY;
    server_addr.sin_port = htons(443);

    err = bind(server_fd, (sockaddr*)&server_addr, sizeof(server_addr));
    if(err == SOCKET_ERROR) {
        error("bad bind");
    }

    err = listen(server_fd, SOMAXCONN);
    if(err == SOCKET_ERROR) {
        error("bad listen");
    }

    sockaddr_in client_addr = zero;
    int client_len = sizeof(client_addr);
    printf("Waiting for connection...\n");
    SOCKET client_fd = accept(server_fd, (sockaddr*)&client_addr, &client_len);
    if(client_fd == INVALID_SOCKET) {
        error("bad accept");
    }
    printf("FOOBAR\n");

    SSL* ssl = SSL_new(ctx);
    SSL_set_fd(ssl, (int)client_fd);

    err = SSL_accept(ssl);
    if(err <= 0) {
        // err = SSL_get_error(ssl, err);
        // switch (err) {
        //  case SSL_ERROR_WANT_READ:
        //  case SSL_ERROR_WANT_WRITE:
        //  {
        //      printf("try again\n");
        //  } break;
        //  case SSL_ERROR_ZERO_RETURN: {
        //      printf("tcp closed\n");
        //  } break;
        //  default: {
        //      printf("fatal %i\n", err);
        //  } break;
        // }
        printf("bad ssl_accept\n");
    }

    char* reply = "it's a cheat code";
    // SSL_write(ssl, reply, (int)strlen(reply));
    while(true) {
        char buffer[1024] = zero;
        int bytes_read = SSL_read(ssl, buffer, sizeof(buffer));
        if(bytes_read > 0) {
            printf("received: %s\n", buffer);
        }
        else {
            ERR_print_errors_fp(stderr);
        }
    }

    // Print OpenSSL version
    printf("OpenSSL version: %s\n", OpenSSL_version(OPENSSL_VERSION));

    getchar();
}

func void error(char* str)
{
    printf("%s\n", str);
    getchar();
    exit(1);
}