/*
 Copyright (C) 2020 Fredrik Öhrström

 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"rtlsdr.h"
#include"util.h"

// Include rtl-sdr which is licensed under GPL v2 or later
// so it happily relicenses with wmbusmeters under GPL v3 or later.

#include<rtl-sdr.h>

// Include libusb, we do not currently make use of this,
// but the intention is to have better detection of usb
// changes in the future. Perhaps even callbacks?

//#include<libusb-1.0/libusb.h>

using namespace std;

/*
struct StaticLibUSB
{
    libusb_context *ctx_;

    StaticLibUSB()
    {
        libusb_init(&ctx_);
        libusb_device **list;
        ssize_t count = libusb_get_device_list(ctx_, &list);

        for (ssize_t idx = 0; idx < count; ++idx)
        {
            libusb_device *device = list[idx];
            libusb_device_descriptor desc = {0};

            libusb_get_device_descriptor(device, &desc);

            printf("Vendor:Device = %04x:%04x\n", desc.idVendor, desc.idProduct);
        }

        libusb_free_device_list(list, true);
    }
    ~StaticLibUSB()
    {
        libusb_exit(ctx_);
    }
};
*/
//StaticLibUSB static_;

vector<string> listRtlSdrDevices()
{
    vector<string> devices;

    uint32_t n = rtlsdr_get_device_count();

    char mfct[256];
    char product[256];
    char serial[256];

    for (uint32_t i=0; i<n; ++i)
    {
        rtlsdr_get_device_usb_strings(i, mfct, product, serial);
        devices.push_back(serial);
    }
    return devices;
}

int indexFromRtlSdrSerial(std::string serialnr)
{
    uint32_t n = rtlsdr_get_device_count();

    char mfct[256];
    char product[256];
    char serial[256];

    for (uint32_t i=0; i<n; ++i)
    {
        rtlsdr_get_device_usb_strings(i, mfct, product, serial);
        if (serialnr == serial)
        {
            return i;
        }
    }

    // Something is wrong.
    return -1;
}

AccessCheck detectRTLSDR(string serialnr, Detected *detected)
{
    if (detected->specified_device.type != WMBusDeviceType::DEVICE_RTLWMBUS &&
        detected->specified_device.type != WMBusDeviceType::DEVICE_RTL433)
    {
        return AccessCheck::NoSuchDevice;
    }

    uint32_t n = rtlsdr_get_device_count();

    char mfct[256];
    char product[256];
    char serial[256];

    for (uint32_t i=0; i<n; ++i)
    {
        rtlsdr_get_device_usb_strings(i, mfct, product, serial);
        if (serialnr == "" || serialnr == serial)
        {
            LinkModeSet lms;
            lms.addLinkMode(LinkMode::C1);
            lms.addLinkMode(LinkMode::T1);
            detected->setAsFound(serialnr, detected->specified_device.type, 0, false, lms);
            return AccessCheck::AccessOK;
        }
    }

    // Something is wrong.
    return AccessCheck::NoSuchDevice;
}

/*
Find /dev/swradio0 1 2 3 etc
Works in Ubuntu where the dev device is automatically created.

    struct dirent **entries;
    vector<string> found_swradios;
    string devdir = "/dev/";

    int n = scandir(devdir.c_str(), &entries, NULL, sorty);
    if (n < 0)
    {
        perror("scandir");
        return found_swradios;
    }

    for (int i=0; i<n; ++i)
    {
        string name = entries[i]->d_name;

        if (name ==  ".." || name == ".")
        {
            free(entries[i]);
            continue;
        }
        // swradio0 swradio1 swradio2 ...
        if (name.length() > 7 &&
            !strncmp(name.c_str(), "swradio", 7) &&
            name[7] >= '0' &&
            name[7] <= '9')
        {
            string rtlsdr = devdir+name;
            found_swradios.push_back(rtlsdr);
        }
        free(entries[i]);
    }
    free(entries);

    return found_swradios;
*/