/**
 * Copyright (C) 2004-2016, GoodData(R) Corporation. All rights reserved.
 * This source code is licensed under the BSD-style license found in the
 * LICENSE.txt file in the root directory of this source tree.
 */
package com.gooddata;

import com.gooddata.account.Account;
import com.gooddata.account.AccountService;
import com.gooddata.dataset.DatasetManifest;
import com.gooddata.dataset.DatasetService;
import com.gooddata.gdc.DataStoreService;
import com.gooddata.md.Attribute;
import com.gooddata.md.Fact;
import com.gooddata.md.MetadataService;
import com.gooddata.md.Metric;
import com.gooddata.md.report.AttributeInGrid;
import com.gooddata.md.report.GridReportDefinitionContent;
import com.gooddata.md.report.MetricElement;
import com.gooddata.md.report.Report;
import com.gooddata.md.report.ReportDefinition;
import com.gooddata.model.ModelDiff;
import com.gooddata.model.ModelService;
import com.gooddata.project.Project;
import com.gooddata.project.ProjectService;
import com.gooddata.report.ReportExportFormat;
import com.gooddata.report.ReportService;

import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.util.Collection;

import static com.gooddata.md.Restriction.identifier;
import static com.gooddata.md.report.MetricGroup.METRIC_GROUP;
import static java.util.Arrays.asList;

public class Example {

    public static void main(String... args) throws FileNotFoundException {
        final GoodData gd = new GoodData("roman@gooddata.com", "Roman1");


        final AccountService accountService = gd.getAccountService();
        final Account current = accountService.getCurrent();
        System.out.println(current.getId());


        final ProjectService projectService = gd.getProjectService();

        final Collection<Project> projects = projectService.getProjects();
        System.out.println(projects);

        final Project project = projectService.createProject(new Project("sparkling3", "pgroup2")).get();
        System.out.println(project.getUri());


        final ModelService modelService = gd.getModelService();

        final ModelDiff projectModelDiff = modelService.getProjectModelDiff(project,
                new InputStreamReader(Example.class.getResourceAsStream("/person.json"))).get();
        modelService.updateProjectModel(project, projectModelDiff).get();


        final MetadataService md = gd.getMetadataService();

        String fact = md.getObjUri(project, Fact.class, identifier("fact.person.shoesize"));
        Metric m = md.createObj(project, new Metric("Avg shoe size", "SELECT AVG([" + fact + "])", "#,##0"));

        Attribute attr = md.getObj(project, Attribute.class, identifier("attr.person.department"));

        ReportDefinition definition = GridReportDefinitionContent.create(
                "Department avg shoe size",
                asList(METRIC_GROUP),
                asList(new AttributeInGrid(attr.getDefaultDisplayForm().getUri())),
                asList(new MetricElement(m.getUri(), "Avg shoe size"))
        );
        definition = md.createObj(project, definition);
        Report report = md.createObj(project, new Report(definition.getTitle(), definition));


        final DatasetService datasetService = gd.getDatasetService();

        final DatasetManifest manifest = datasetService.getDatasetManifest(project, "dataset.person");
        datasetService.loadDataset(project, manifest, Example.class.getResourceAsStream("/person.csv")).get();


        final ReportService reportService = gd.getReportService();

        reportService.exportReport(definition, ReportExportFormat.CSV, new FileOutputStream("report.csv"));


        DataStoreService dataStoreService = gd.getDataStoreService();

        dataStoreService.upload("/dir/file.txt", new FileInputStream("file.txt"));
        InputStream stream = dataStoreService.download("/dir/file.txt");
        dataStoreService.delete("/dir/file.txt");

        gd.logout();
    }

}