/*
 *
 *    Copyright (c) 2013-2017 Nest Labs, Inc.
 *    All rights reserved.
 *
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *
 *        http://www.apache.org/licenses/LICENSE-2.0
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */

/**
 *    @file
 *      Implementation of DeviceDescOptions object, which handles parsing of command line options
 *      that specify descriptive information about the simulated "device" used in test applications.
 *
 */


#include "ToolCommon.h"
#include <Weave/Support/CodeUtils.h>
#include <Weave/Profiles/WeaveProfiles.h>
#include <Weave/Profiles/common/CommonProfile.h>
#include <Weave/Profiles/vendor/nestlabs/device-description/NestProductIdentifiers.hpp>
#include "DeviceDescOptions.h"

using namespace nl::Weave::Profiles::DeviceDescription;

DeviceDescOptions gDeviceDescOptions;

DeviceDescOptions::DeviceDescOptions()
{
    static OptionDef optionDefs[] =
    {
        { "serial-num",       kArgumentRequired, kToolCommonOpt_DeviceSerialNum        },
        { "vendor-id",        kArgumentRequired, kToolCommonOpt_DeviceVendorId         },
        { "product-id",       kArgumentRequired, kToolCommonOpt_DeviceProductId        },
        { "product-rev",      kArgumentRequired, kToolCommonOpt_DeviceProductRevision  },
        { "software-version", kArgumentRequired, kToolCommonOpt_DeviceSoftwareVersion  },
        { }
    };
    OptionDefs = optionDefs;

    HelpGroupName = "DEVICE DESCRIPTION OPTIONS";

    OptionHelp =
        "  --serial-num <string>\n"
        "       Device serial number. Defaults to \"mock-device\".\n"
        "\n"
        "  --vendor-id <int>\n"
        "       Device vendor id.  Defaults to 0x235A (Nest Labs)\n"
        "\n"
        "  --product-id <int>\n"
        "       Device product id. Defaults to 5 (Nest Protect).\n"
        "\n"
        "  --product-rev <int>\n"
        "       Device product revision. Defaults to 1.\n"
        "\n"
        "  --software-version <string>\n"
        "       Device software version string. Defaults to \"mock-device/1.0\".\n"
        "\n";

    // Setup Defaults.
    BaseDeviceDesc.Clear();
    BaseDeviceDesc.VendorId = kWeaveVendor_NestLabs;
    BaseDeviceDesc.ProductId = nl::Weave::Profiles::Vendor::Nestlabs::DeviceDescription::kNestWeaveProduct_Topaz;
    BaseDeviceDesc.ProductRevision = 1;
    BaseDeviceDesc.ManufacturingDate.Year = 2013;
    BaseDeviceDesc.ManufacturingDate.Month = 1;
    BaseDeviceDesc.ManufacturingDate.Day = 1;
    memset(BaseDeviceDesc.Primary802154MACAddress, 0x11, sizeof(BaseDeviceDesc.Primary802154MACAddress));
    memset(BaseDeviceDesc.PrimaryWiFiMACAddress, 0x22, sizeof(BaseDeviceDesc.PrimaryWiFiMACAddress));
    strcpy(BaseDeviceDesc.RendezvousWiFiESSID, "MOCK-1111");
    strcpy(BaseDeviceDesc.SerialNumber, "mock-device");
    strcpy(BaseDeviceDesc.SoftwareVersion, "mock-device/1.0");
    BaseDeviceDesc.DeviceFeatures =
        WeaveDeviceDescriptor::kFeature_HomeAlarmLinkCapable |
        WeaveDeviceDescriptor::kFeature_LinePowered;
    // NOTE: For security reasons, pairing codes should only ever appear in device descriptors that are
    // encoded into QR codes. options.BaseDeviceDesc contains the device descriptor fields that get sent over the
    // network (e.g. in an IdentifyDevice exchange).  Therefore the PairingCode field should never be
    // set here.
}

void DeviceDescOptions::GetDeviceDesc(WeaveDeviceDescriptor& deviceDesc)
{
    deviceDesc = BaseDeviceDesc;
    deviceDesc.DeviceId = FabricState.LocalNodeId;
    deviceDesc.FabricId = FabricState.FabricId;
    memset(deviceDesc.PairingCode, 0, sizeof(deviceDesc.PairingCode));
}

bool DeviceDescOptions::HandleOption(const char *progName, OptionSet *optSet, int id, const char *name, const char *arg)
{
    DeviceDescOptions& options = *static_cast<DeviceDescOptions *>(optSet);

    switch (id)
    {
    case kToolCommonOpt_DeviceSerialNum:
        if (strlen(arg) > WeaveDeviceDescriptor::kMaxSerialNumberLength)
        {
            PrintArgError("%s: Invalid value specified for device serial number (value too long): %s\n", progName, arg);
            return false;
        }
        strncpy(options.BaseDeviceDesc.SerialNumber, arg, sizeof(options.BaseDeviceDesc.SerialNumber));
        break;
    case kToolCommonOpt_DeviceVendorId:
        if (!ParseInt(arg, options.BaseDeviceDesc.VendorId) || options.BaseDeviceDesc.VendorId == 0 || options.BaseDeviceDesc.VendorId >= 0xFFF0)
        {
            PrintArgError("%s: Invalid value specified for device vendor ID: %s\n", progName, arg);
            return false;
        }
        break;
    case kToolCommonOpt_DeviceProductId:
        if (!ParseInt(arg, options.BaseDeviceDesc.ProductId) || options.BaseDeviceDesc.ProductId == 0 || options.BaseDeviceDesc.ProductId == 0xFFFF)
        {
            PrintArgError("%s: Invalid value specified for device product ID: %s\n", progName, arg);
            return false;
        }
        break;
    case kToolCommonOpt_DeviceProductRevision:
        if (!ParseInt(arg, options.BaseDeviceDesc.ProductRevision))
        {
            PrintArgError("%s: Invalid value specified for device product revision: %s\n", progName, arg);
            return false;
        }
        break;
    case kToolCommonOpt_DeviceSoftwareVersion:
        if (strlen(arg) > WeaveDeviceDescriptor::kMaxSoftwareVersionLength)
        {
            PrintArgError("%s: Invalid value specified for device software version (value too long): %s\n", progName, arg);
            return false;
        }
        strncpy(options.BaseDeviceDesc.SoftwareVersion, arg, sizeof(options.BaseDeviceDesc.SoftwareVersion));
        break;
    default:
        PrintArgError("%s: INTERNAL ERROR: Unhandled option: %s\n", progName, name);
        return false;
    }

    return true;
}