main.cpp

#include <v4l2vnc/VNCServer.h>
#include <v4l2vnc/V4L2VNCServer.h>
#include <v4l2vnc/Logging.h>

#include <signal.h>

#include <iostream>


using namespace v4l2vnc;


void printHelpAndExit()
{
    std::string_view msg =
"[h] [-i input_device] [-p port] [-l listen_address] [-W capture_width] [-H capture_height] [-n server_name] [-v verbosity]\n"
"   -v verbosity        - Must be one of 'trace', 'debug', 'info', 'warning', 'error', 'critical'\n";
    std::cout << msg;
    exit( 0 );
}


int intOrQuite( const char* source, const char* ps )
{
    int ret;
    if ( auto ec = std::from_chars( source, source + strlen( source ), ret ); ec.ec != std::errc{} )
    {
        spdlog::critical( ps );
        exit( -1 );
    }
    return ret;
}

// global for interruption handler
static std::shared_ptr<V4L2VNCServer> server;

void on_signal( int sig )
{
    if ( server && sig == SIGINT || sig == SIGTERM )
    {
        logger()->info( "Terminating gracefully..." );
        server->stop();
    }
}


int main( int argc,char** argv )
{
    V4L2VNCServer::Params params;
    params.bpp = v4l2vnc::VNCServer::BytesPerPixel::TrueColor;

    int c;
    while ( ( c = getopt( argc, argv, "i:p:l:W:H:n:v:h" ) ) > 0 )
    {
        switch ( c )
        {
            case 'i':
                params.inputDevice = optarg;
                break;
            case 'p':
                params.port = intOrQuite( optarg, "Invalid port" );
                break;
            case 'l':
                params.listen = strdup( optarg );
                break;
            case 'W':
                params.width = intOrQuite( optarg, "Invalid width" );
                break;
            case 'H':
                params.height = intOrQuite( optarg, "Invalid height" );
                break;
            case 'n':
                params.desktopName = strdup( optarg );
                break;
            case 'v':
                logger()->set_level( spdlog::level::from_str( optarg ) );
                break;
            case 'h':
                printHelpAndExit();
        }
    }

    server = std::make_shared<V4L2VNCServer>( params );
    if ( auto ec = server->init(); ec )
    {
        spdlog::critical( "Failed to init server: {}", ec.message() );
        return -1;
    }
    signal( SIGINT, on_signal );
    signal( SIGTERM, on_signal );
    server->run();
    return 0;
}