overte/assignment-client/src/main.cpp

//
//  main.cpp
//  assignment-client
//
//  Created by Stephen Birarda on 8/22/13.
//  Copyright (c) 2013 High Fidelity, Inc. All rights reserved.
//

#include <arpa/inet.h>
#include <sys/time.h>

#include <Assignment.h>
#include <AudioMixer.h>
#include <NodeList.h>
#include <PacketHeaders.h>
#include <SharedUtil.h>

const int ASSIGNMENT_REQUEST_INTERVAL_USECS = 1 * 1000 * 1000;

int main(int argc, char* const argv[]) {
    
    // create a NodeList as an unassigned client
    NodeList* nodeList = NodeList::createInstance(NODE_TYPE_UNASSIGNED);
    nodeList->getNodeSocket()->setBlocking(false);
    
    timeval lastRequest = {};
    
    unsigned char packetData[MAX_PACKET_SIZE];
    ssize_t receivedBytes = 0;
    
    // loop the parameters to see if we were passed a pool
    int parameter = -1;
    const char ALLOWED_PARAMETERS[] = "p::";
    const char POOL_PARAMETER_CHAR = 'p';
    
    char* assignmentPool = NULL;
    
    while ((parameter = getopt(argc, argv, ALLOWED_PARAMETERS)) != -1) {
        if (parameter == POOL_PARAMETER_CHAR) {
            // copy the passed assignment pool
            int poolLength = strlen(optarg);
            assignmentPool = new char[poolLength + sizeof(char)];
            strcpy(assignmentPool, optarg);
        }
    }
    
    // create a request assignment, accept all assignments, pass the desired pool (if it exists)
    Assignment requestAssignment(Assignment::Request, Assignment::All, assignmentPool);
    
    while (true) {
        if (usecTimestampNow() - usecTimestamp(&lastRequest) >= ASSIGNMENT_REQUEST_INTERVAL_USECS) {
            gettimeofday(&lastRequest, NULL);
            
            // send an assignment request to the Nodelist
            qDebug("Sending assignment request.\n");
            nodeList->sendAssignment(requestAssignment);
        }        
        
        while (nodeList->getNodeSocket()->receive(packetData, &receivedBytes)) {
            if (packetData[0] == PACKET_TYPE_DEPLOY_ASSIGNMENT && packetVersionMatch(packetData)) {
                
                // construct the deployed assignment from the packet data
                Assignment deployedAssignment(packetData, receivedBytes);
                
                qDebug() << "Received an assignment - " << deployedAssignment << "\n";
                
                // switch our nodelist DOMAIN_IP to the ip receieved in the assignment
                if (deployedAssignment.getDomainSocket()->sa_family == AF_INET) {
                    in_addr domainSocketAddr = ((sockaddr_in*) deployedAssignment.getDomainSocket())->sin_addr;
                    nodeList->setDomainIP(inet_ntoa(domainSocketAddr));
                    
                    qDebug() << "Changed domain IP to " << inet_ntoa(domainSocketAddr);
                }
                
                // run the AudioMixer, it's the only possible assignment for now
                AudioMixer::run();
                
                // reset our NodeList by switching back to unassigned and clearing the list
                nodeList->setOwnerType(NODE_TYPE_UNASSIGNED);
                nodeList->clear();
            }
        }
    }
}