package software.aws.mcs.examples;

/*-
 * #%L
 * AWS SigV4 Auth Java Driver 4.x Examples
 * %%
 * Copyright (C) 2020 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 * %%
 * Permission is hereby granted, free of charge, to any person obtaining a copy of this
 * software and associated documentation files (the "Software"), to deal in the Software
 * without restriction, including without limitation the rights to use, copy, modify,
 * merge, publish, distribute, sublicense, and/or sell copies of the Software, and to
 * permit persons to whom the Software is furnished to do so.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED,
 * INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A
 * PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
 * HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
 * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
 * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 * #L%
 */

import com.datastax.oss.driver.api.core.CqlSession;
import com.datastax.oss.driver.api.core.cql.*;

import software.aws.mcs.auth.SigV4AuthProvider;

import java.net.InetSocketAddress;
import java.util.Collections;
import java.util.List;

public class OrderFetcher {
    public final static String TABLE_FORMAT = "%-25s%s\n";

    public static void main(String[] args) {
        if (args.length != 3) {
            System.err.println("Usage: OrderFetcher <region> <endpoint> <customer ID>");
            System.exit(1);
        }

        // Both of these can be speficied in the configuration, but
        // are being done programmatically here for flexibility as
        // example code.
        SigV4AuthProvider provider = new SigV4AuthProvider(args[0]);
        List<InetSocketAddress> contactPoints = Collections.singletonList(new InetSocketAddress(args[1], 9142));

        try (CqlSession session = CqlSession.builder().addContactPoints(contactPoints).withAuthProvider(provider).withLocalDatacenter(args[0]).build()) {
            // Use a prepared query for quoting
            PreparedStatement prepared = session.prepare("select * from acme.orders where customer_id = ?");

            // We use execute to send a query to Cassandra. This returns a ResultSet, which is essentially a collection
            // of Row objects.
            ResultSet rs = session.execute(prepared.bind(args[2]));

            // Print the header
            System.out.printf(TABLE_FORMAT, "Date", "Order Id");

            for (Row row : rs) {
                System.out.printf(TABLE_FORMAT, row.getInstant("order_timestamp"), row.getUuid("order_id"));
            }
        }

    }
}