/*
 * This file is part of the modbus_relay project.
 * Copyright 2025 Edward V. Emelianov <edward.emelianoff@gmail.com>.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <modbus/modbus.h>
#include <stdio.h>
#include <string.h>

#include "cmdlnopts.h"

static void getN(modbus_t *ctx){
    uint16_t nodeN;
    int s = modbus_get_slave(ctx);
    VDBG("Old node num: %d\n", s);
    if(modbus_set_slave(ctx, 0)) ERRX("Can't set modbus slave");
    printf("now slave id=%d\n", modbus_get_slave(ctx));
    if(modbus_read_registers(ctx, 0, 1, &nodeN) < 0) WARNX("Can't read node number");
    if(modbus_set_slave(ctx, s)) ERRX("Can't set modbus slave");
    else printf("NODENUM=%u\n", nodeN);
    printf("now slave id=%d\n", modbus_get_slave(ctx));
}

int main(int argc, char **argv){
    sl_init();
    parse_args(argc, argv);
    verblvl = GP.verbose + LOGLEVEL_WARN;
    if(verblvl >= LOGLEVEL_AMOUNT) verblvl = LOGLEVEL_AMOUNT - 1;
    if(GP.logfile) OPENLOG(GP.logfile, verblvl, 1);
    LOGMSG("Started");
    VMSG("Try to open %s @%d ... ", GP.device, GP.baudrate);
    modbus_t *ctx = modbus_new_rtu(GP.device, GP.baudrate, 'N', 8, 1);
    modbus_set_response_timeout(ctx, 0, 100000);
    if(modbus_set_slave(ctx, GP.nodenum)) ERRX("Can't set modbus slave");
    if(modbus_connect(ctx) < 0) ERR("Can't open device %s", GP.device);
    if(GP.relaytype < 0 || GP.relaytype > 1) ERRX("Type should be 0 or 1");
    VMSG("OK!\n");
    uint8_t dest8[8] = {0};
    int result = 0;
    int ninputs = 8, noutputs = 8;
    if(GP.relaytype == 1){ ninputs = 4; noutputs = 4;}
    if(GP.setall){
        if(GP.relaytype == 0){
            memset(dest8, 1, noutputs);
            result = modbus_write_bits(ctx, 0, noutputs, dest8);
        }else{
            result = modbus_write_bit(ctx, 0xff, 0xff00);
        }
        if(result < 0) WARNX("Can't set all relays");
    }else if(GP.resetall){
        if(GP.relaytype == 0){
            result = modbus_write_bits(ctx, 0, noutputs, dest8);
        }else{
            result = modbus_write_bit(ctx, 0xff, 0);
        }
        if(result < 0) WARNX("Can't clear all relays");
    }else{
        if(GP.resetrelay){
            int **p = GP.resetrelay;
            while(*p){
                int n = **p;
                if(n > 7 || n < 0) WARNX("Relay number should be in [0, 7]");
                else{
                    if(modbus_write_bit(ctx, n, 0) < 0) WARNX("Can't reset relay #%d", n);
                    else VMSG("RELAY%d=0\n", n);
                }
                ++p;
            }
        }
        if(GP.setrelay){
            int **p = GP.setrelay;
            while(*p){
                int n = **p;
                if(n > 7 || n < 0) WARNX("Relay number should be in [0, 7]");
                else{
                    if(modbus_write_bit(ctx, n, 1) < 0) WARNX("Can't set relay #%d", n);
                    else VMSG("RELAY%d=1\n", n);
                }
                ++p;
            }
        }
    }
    if(GP.getinput){
        if(modbus_read_input_bits(ctx, 0, ninputs, dest8) < 0) WARNX("Can't read inputs");
        else{
            int **p = GP.getinput;
            while(*p){
                int n = **p;
                if(n > ninputs-1 || n < 0) WARNX("Input number should be in [0, %d]", ninputs-1);
                else printf("INPUT%d=%u\n", n, dest8[n]);
                ++p;
            }
        }
    }
    if(GP.getrelay){
        if(modbus_read_bits(ctx, 0, noutputs, dest8) < 0) WARNX("Can't read relays");
        else{
            int **p = GP.getrelay;
            while(*p){
                int n = **p;
                if(n > noutputs-1 || n < 0) WARNX("Relay number should be in [0, %d]", noutputs-1);
                else printf("RELAY%d=%u\n", n, dest8[n]);
                ++p;
            }
        }
    }
    if(GP.readn){
        getN(ctx);
    }
    if(GP.setn){
        uint16_t nodeN = (uint16_t) GP.setn;
        if(GP.relaytype == 0){
            modbus_write_registers(ctx, 0, 1, &nodeN);
        }else{
            modbus_write_register(ctx, 0x4000, nodeN);
        }
        modbus_set_slave(ctx, nodeN);
        getN(ctx);
    }
    LOGMSG("End");
    VMSG("End\n");
    modbus_close(ctx);
    modbus_free(ctx);
    return 0;
}