import io.github.spencerpark.ijava.IJava;
import io.github.spencerpark.jupyter.kernel.magic.common.Shell;
IJava.getKernelInstance().getMagics().registerMagics(Shell.class);
%sh asd

%%loadFromPOM
<dependencies>
  <dependency>
    <groupId>org.slf4j</groupId>
    <artifactId>slf4j-simple</artifactId>
    <version>1.7.25</version>
  </dependency>
  <dependency>
    <groupId>com.aerospike</groupId>
    <artifactId>aerospike-client</artifactId>
    <version>6.1.4</version>
  </dependency>
  <dependency>
    <groupId>com.aerospike</groupId>
    <artifactId>aerospike-document-api</artifactId>
    <version>1.2.0</version>
  </dependency>
  <dependency>
    <groupId>commons-io</groupId>
    <artifactId>commons-io</artifactId>
    <version>2.6</version>
  </dependency>
  <dependency>
    <groupId>com.googlecode.json-simple</groupId>
    <artifactId>json-simple</artifactId>
    <version>1.1.1</version>
</dependency>
</dependencies>

import com.aerospike.client.AerospikeClient;
import com.aerospike.client.Bin;
import com.aerospike.client.Host;
import com.aerospike.client.Key;
import com.aerospike.client.Record;
import com.aerospike.client.policy.WritePolicy;
import com.aerospike.documentapi.*;
import com.aerospike.documentapi.JsonConverters;

import com.fasterxml.jackson.databind.JsonNode;
import com.jayway.jsonpath.JsonPath;
import java.util.List;
import java.util.Map;
import java.nio.charset.StandardCharsets;
import org.apache.commons.io.FileUtils;
import org.json.simple.JSONArray;
import org.json.simple.JSONObject;
import org.json.simple.parser.JSONParser;

Host[] hosts = new Host[] {
    new Host("127.0.0.1", 3000)
};

WritePolicy writePolicy = new WritePolicy();
writePolicy.sendKey = true;

AerospikeClient client = new AerospikeClient(null, hosts);
System.out.println("Initialized Aerospike client and connected to the cluster.");

AerospikeDocumentClient docClient = new AerospikeDocumentClient(client);
System.out.println("Initialized document client from the Aerospike client.");

String companiesString = FileUtils.readFileToString(new File("./space-companies.json"), StandardCharsets.UTF_8);

System.out.println("Document parsed");

System.out.println("Writing record...\n");

Key key = new Key("demo", "space-doc", "companies");

JsonNode spaceNode = JsonConverters.convertStringToJsonNode(companiesString);

docClient.put(writePolicy, key, "company_bin", spaceNode);

Record record = client.get(null, key);
System.out.format("Companies: %2.100s\n\n", record.bins.get("company_bin"));

System.out.println("Writing record complete");

System.out.println("Getting company data...\n");
Key key = new Key("demo", "space-doc", "companies");

Object pixxel = docClient.get(key, "company_bin", "$.companies[*].company[?(@.companyName == 'Pixxel')]");

System.out.format("%s\n\n", pixxel);
System.out.println("Company data returned");

System.out.println("Getting company data...\n");
Key key = new Key("demo", "space-doc", "companies");

Object pixxel = docClient.get(key, "company_bin", "$.companies[*].company[?(@.companyName == 'Pixxel')].['companyName','about','email','webSite']");

System.out.format("%s\n\n", pixxel);
System.out.println("Company data returned");

System.out.println("Getting companies data...\n");
Key key = new Key("demo", "space-doc", "companies");

List companyData = (List) docClient.get(key, "company_bin", "$.companies[*].company[?(@.testimonials.length() > 2 && @.paymentType == 'free')].['companyName','about','email','webSite']");

for(int i = 0; i < companyData.size(); i++){
    System.out.format("%s\n\n", companyData.get(i));
}
System.out.println("Company data returned");

// Create a new employee testimonial for CERN
Map empTest = new TreeMap<String, String>();

empTest.put("content", "My summer internship at CERN was absolutely incredible. The amount of knowledge I have gained while working with some of the best scientists in the world is sure to propel my career forward. I hope to have the opportunity to return to CERN in the future");
empTest.put("person", "Rebecca");
empTest.put("position", "Summer intern at CERN");

System.out.println("Testimonial created");

System.out.println("Adding new testimonial...\n");
Key key = new Key("demo", "space-doc", "companies");

// Add the new testimonial
docClient.append(key, "company_bin", "$.companies[*].company[?(@.companyName=='CERN')].testimonials", empTest);

// Read back the testimonials to see the new entry
List test = (List) docClient.get(key, "company_bin", "$.companies[*].company[?(@.companyName=='CERN')]..testimonials");

List testimonials = (List) test.get(0);
for (int i = 0; i < testimonials.size(); i++){
    System.out.format("%s\n\n", testimonials.get(i));
}

System.out.println("Testimonial added");

// Get number of companies in the document
Object companies = docClient.get(key, "company_bin", "$.companies.length()");
System.out.println(companies + " Companies in the list\n");

// Delete all companies with a payment type NOT equal to 'free'
docClient.delete(key, "company_bin", "$.companies[?(@.company.paymentType!='free')]");
System.out.println("Companies deleted\n");

// Get new count of companies in the document
companies = docClient.get(key, "company_bin", "$.companies.length()");
System.out.println(companies + " Companies in the list");

// Remove the data
client.truncate(null, "demo", "space-doc", null);
System.out.println("Data removed");

client.close();