// amps_sow_and_subscribe_with_oof.cpp
//
// This program shows how to retrieve the current state of a SOW database and
// subscribe to updates in a single operation. Doing things with this
// method ensures that the program will not miss messages. The subscription
// begins exactly at the point where the query ends, so no updates are
// lost.
//
// The program flow is simple:
//
// * Connect to AMPS
// * Logon
// * Query messages from the SOW topic with a simple filter. The subscription
//   requests OOF messages.
// * Output the messages to the console. The program continues
//   to receive update messages after the SOW query finishes. Notice
//   that the program handles OOF messages by checking the command returned
//   in the message.
//
// This sample doesn't include error handling, high availability, or
// connection retry logic.
//
// This file is a part of the AMPS C++ Client distribution samples.
////////////////////////////////////////////////////////////////////////////
//
// Copyright (c) 2010-2026 60East Technologies Inc., All Rights Reserved.
//
// This computer software is owned by 60East Technologies Inc. and is
// protected by U.S. copyright laws and other laws and by international
// treaties.  This computer software is furnished by 60East Technologies
// Inc. pursuant to a written license agreement and may be used, copied,
// transmitted, and stored only in accordance with the terms of such
// license agreement and with the inclusion of the above copyright notice.
// This computer software or any other copies thereof may not be provided
// or otherwise made available to any other person.
//
// U.S. Government Restricted Rights.  This computer software: (a) was
// developed at private expense and is in all respects the proprietary
// information of 60East Technologies Inc.; (b) was not developed with
// government funds; (c) is a trade secret of 60East Technologies Inc.
// for all purposes of the Freedom of Information Act; and (d) is a
// commercial item and thus, pursuant to Section 12.212 of the Federal
// Acquisition Regulations (FAR) and DFAR Supplement Section 227.7202,
// Government’s use, duplication or disclosure of the computer software
// is subject to the restrictions set forth by 60East Technologies Inc..
//
////////////////////////////////////////////////////////////////////////////
#include <amps/ampsplusplus.hpp>
#include <iostream>
#include <string>

// Callback function for handling messages from AMPS.
//
// * message is the message received from AMPS
//
// * user_data is provided by the subscriber as a way to
//   pass extra information into the message handler.



bool processSOWMessage(const AMPS::Message& message)
{

  if (message.getCommandEnum() == AMPS::Message::Command::GroupBegin)
  {
    std::cout << "Receiving messages from the SOW." << std::endl;
  }
  else if (message.getCommandEnum() == AMPS::Message::Command::GroupEnd)
  {
    std::cout << "Done receiving messages from SOW." << std::endl;
    return true;
  }
  else
  {

    std::cout << "SOW message: " << message.getData() << std::endl;
  }
  return false;
}

void processSubscribeMessage(const AMPS::Message& message)
{
  if (message.getCommand() == "oof")
  {
    std::cout << "OOF : " << message.getReason()
              << " : " <<  message.getData() << std::endl;
  }
  else
  {
    std::cout << "Received message: " << message.getData() << std::endl;
  }

}

// main
//
// Create a client, connect to the server, query the SOW,
// and then wait for the callback function to process
// the messages.

int main (void)
{

  // Address for the AMPS server.

  const char* uri = "tcp://sample_user@:127.0.0.1:9027/amps/json";

  // Create a client with the name "exampleSowAndSubscribe"

  AMPS::Client ampsClient("exampleSowAndSubscribeWithOOF");

  try
  {
    // connect to the server and log on

    ampsClient.connect(uri);
    ampsClient.logon();


    std::cout << std::endl << "-----" << std::endl
              << "Submitting query for messageNumber values that are "
              << "multiples of 10."
              << std::endl << std::endl;


    // query for messages from the messages-sow topic

    // After the query is done, the program receives new messages
    // on a subscription. When a previously received message no longer
    // matches the subscription, the program receives an Out of Focus
    // message.

    bool sowDone = false;

    std::cerr << "about to subscribe..." << std::endl;

    AMPS::MessageStream stream = ampsClient.sowAndSubscribe(
                                   "messages-sow",
                                   0,      // Timeout
                                   "/messageNumber % 10 = 0 AND (/OptionalField IS NULL OR /OptionalField <> 'ignore_me')", // Filter
                                   0,       // Batch Size
                                   true);  // OOF enabled
    for (auto message = stream.begin(); message != stream.end(); ++message)
    {
      if (sowDone == false)
      {
        sowDone = processSOWMessage(*message);
      }
      else
      {
        processSubscribeMessage(*message);
      }
    }

  }
  catch (const AMPS::AMPSException& e)
  {
    std::cerr << e.what() << std::endl;
    exit(1);
  }

  ampsClient.disconnect();
  return 0;
}


