#include <iostream>
#include <ThermoHubClient/ThermoHubClient.h>
using namespace std;
using namespace ThermoHubClient;

int main()
{
    DatabaseClient dbc("hub-connection-config.json"); // database connection  (local or remote) using a configuration file

    DatabaseClient dbc_default; // default connection (remote) to db.thermohub.net

    // Save ThermoDataSet 'aq17' to a database file aq17-thermofun.json
    dbc_default.saveDatabase("aq17");
    // Save a subset of ThermoDataSet 'aq17' to a database file aq17-subset-thermofun.json, using a list of elements
    dbc_default.saveDatabaseContainingElements("aq17", {"H", "O", "Na", "K", "Si", "Al", "Cl", "Zr", "Zz"});
    // Save a subset of ThermoDataSet 'mines16' to a database file mines16-subset-thermofun.json, 
    // using a list of elements, substances, and substance classes. 
    // Only the data that fulfils all selection criteria will be saved
    dbc.saveDatabaseSubset("mines16", {"H", "O", "Na", "K", "Si", "Al", "Cl", "Zr", "Zz"},
                           {"O2@", "H2@", "NaHSiO3@", "Al(OH)2+", "NaAl(OH)4@", "Al+3", "AlH3SiO4+2", "AlOH+2", "Al(OH)4-", "H2", "O2", "H2O"},
                           {"{\"1\":\"SC_GASFLUID\"}", "{\"3\":\"SC_AQSOLVENT\"}"});

    // Get ThermoDataSet 'mines16' as a JSON string
    // all 'save...' functions have a 'get...' function counterpart
    // getDatabase, getDatabaseContainingElements, getDatabaseSubset
    std::string jsonMines16 = dbc.getDatabase("mines16");

    auto tds = dbc.availableThermoDataSets();
    cout << "ThermoDataSets" << endl;
    for (auto t : tds)
        cout << t << endl;

    cout << "Elements" << endl;
    for (auto e : dbc.elementsInThermoDataSet("aq17"))
        cout << e << " ";
    cout << endl;
    
    cout << "Substances" << endl;
    for (auto e : dbc.substancesInThermoDataSet("aq17"))
        cout << e << " ";
    cout << endl;

    cout << "Reactions" << endl;
    for (auto e : dbc.reactionsInThermoDataSet("mines16"))
        cout << e << " ";
    cout << endl;

    cout << "Substances classes" << endl;
    for (auto e : dbc.substanceClassesInThermoDataSet("aq17"))
        cout << e << " ";
    cout << endl;

    cout << "Substances aggregate state" << endl;
    for (auto e : dbc.substanceAggregateStatesInThermoDataSet("aq17"))
        cout << e << " ";
    cout << endl;

    return 0;
}