var _demographic_data = "*Demographic Data";
var _site_information = "*Site Information";
var _intern_data      = "*Intern Data";

function onOpen() {
  var ui = SpreadsheetApp.getUi();

  ui.createMenu('Demographic Manager')
    .addItem('Initialize Spreadsheet', 'spread_init')
    .addItem('Regenerate Reporting Spreadsheets', 'regenerate')
    .addToUi()
}

function spread_init() {
  var ui = SpreadsheetApp.getUi();
  // This is a destructive operation, so prompt the user.
  var response = ui.alert(
    'Initialize spreadsheet',
    'Warning: if you click yes, the spreadsheet will delete data inside itself. Make a backup before running this operation. Continue?',
    ui.ButtonSet.YES_NO_CANCEL
  );

  if (response == ui.Button.NO) {
    ui.alert("Canceled the initialization of the spreadsheet.")
    return;
  } else if (response == ui.Button.CANCEL) {
    ui.alert("Canceled the initialization of the spreadsheet.")
    return;
  }

  var spreadsheet = SpreadsheetApp.getActiveSpreadsheet()
  var sheetsToRemove = spreadsheet.getSheets()
  for (var sheet = 0; sheet < sheetsToRemove.length; sheet++) {
    try {
      spreadsheet.deleteSheet(sheetsToRemove[sheet]);
    } catch { // ran out of spreadsheets to nuke, so rename it to the first thing we need
      spreadsheet.getSheets()[0].setName(_demographic_data)
    }
  }

  // take care of demographic data layout
  var sheet = spreadsheet.getSheets()[0]
  sheet.getRange('A1').activate();
  sheet.getCurrentCell().setValue('Site Name');
  sheet.getRange('B1').activate();
  sheet.getCurrentCell().setValue('Count');
  sheet.getRange('C1').activate();
  sheet.getCurrentCell().setValue('Age');
  sheet.getRange('D1').activate();
  sheet.getCurrentCell().setValue('Gender');
  sheet.getRange('E1').activate();
  sheet.getCurrentCell().setValue('Ethnicity');
  sheet.getRange('F1').activate();
  sheet.getCurrentCell().setValue('Race');

  var newSheet = spreadsheet.insertSheet();
  newSheet.setName(_site_information)
  newSheet.getRange('A1').activate();
  newSheet.getCurrentCell().setValue('Site Name');
  newSheet.getRange('B1').activate();
  newSheet.getCurrentCell().setValue('Date');
  newSheet.getRange('C1').activate();
  newSheet.getCurrentCell().setValue('Those Present');
  newSheet.getRange('D1').activate();
  newSheet.getCurrentCell().setValue('Data Leads');
  newSheet.getRange('E1').activate();
  newSheet.getCurrentCell().setValue('Who Collected?');

  newSheet = spreadsheet.insertSheet();
  newSheet.setName(_intern_data)
  newSheet.getRange('A1').activate();
  newSheet.getCurrentCell().setValue('First Name');
  newSheet.getRange('B1').activate();
  newSheet.getCurrentCell().setValue('Last Name');
  newSheet.getRange('C1').activate();
  newSheet.getCurrentCell().setValue('Gender');
  newSheet.getRange('D1').activate();
  newSheet.getCurrentCell().setValue('Ethnicity');
  newSheet.getRange('E1').activate();
  newSheet.getCurrentCell().setValue('Race');
}

function get_titled_spreadsheet_contents(sheet_name) {
  var spreadsheet = SpreadsheetApp.getActiveSpreadsheet().getSheetByName(sheet_name);
  var rows = spreadsheet.getDataRange().getValues();
  rows.shift(); /* remove the column headers from the data */

  return rows;
}

/*
 * @param {string} site_name
 */
function get_demographic_records_for_site(site_name) {
  var rows = get_titled_spreadsheet_contents(_demographic_data);
  var output = [];

  for(row of rows) {
    if(row[0] === site_name) {
      output = output.concat(Array([1]).fill({
        "site_name": row[0],
        "age": row[2],
        "gender": row[3],
        "ethnicity": row[4],
        "race": row[4],
      }));
    }
  }

  return output;
}

function get_sites() {
  var rows = get_titled_spreadsheet_contents(_site_information);
  var output = [];

  for(row of rows) {
    output.push({
      "site_name": row[0],
      "date": row[1],
      "present": row[2],
      "data_leads": row[3],
      "compiled_by": row[4],
      "attached_records": [],
    })
  }

  return output;
}

function get_intern_data() {
  var rows = get_titled_spreadsheet_contents(_intern_data);
  var output = {};

  for(row of rows) {
    var no_last_name = row[1] === "";
    /* if there's no last name, then just use the first name
     * as the key */
    var key = row[0].concat(no_last_name ? "" : " ".concat(row[1]))
    output[key] = {
      "first_name": row[0],
      "last_name": row[1],
      "calculated_name": key,
      "age": "Intern",
      "gender": row[2],
      "ethnicity": row[3],
      "race": row[4],
    };
  }

  return output;
}

function collate_data() {
  var sites = get_sites();
  var intern_data = get_intern_data();

  for(site of sites) {
    var name = site["site_name"];
    var data = get_demographic_records_for_site(name);

    site["attached_records"].push(...data);
    var those_present = site["present"].split(',').map((x) => x.trim());

    for(person of those_present) {
      site["attached_records"].push(
        intern_data[person]
      )
    }
    var data_leads = site["data_leads"].split(',').map((x) => x.trim());
  }
}