// Copyright (c) Jupyter Development Team.
// Distributed under the terms of the Modified BSD License.

import {
  showDialog, okButton, cancelButton, warnButton
} from '../common/dialog';

import {
  INotebookModel
} from './model';


// The message to display to the user when prompting to trust the notebook.
const TRUST_MESSAGE = '<p>A trusted Jupyter notebook may execute hidden malicious code when you open it.<br>Selecting trust will re-render this notebook in a trusted state.<br>For more information, see the <a href="http://ipython.org/ipython-doc/2/notebook/security.html">Jupyter security documentation</a>.</p>';


/**
 * Trust the notebook after prompting the user.
 *
 * @param model - The notebook model.
 *
 * @param host - The host node for the confirmation dialog (defaults to body).
 *
 * @returns a promise that resolves when the transaction is finished.
 *
 * #### Notes
 * No dialog will be presented if the notebook is already trusted.
 */
export
function trustNotebook(model: INotebookModel, host?: HTMLElement): Promise<void> {
  if (!model) {
    return Promise.resolve(void 0);
  }
  // Do nothing if already trusted.
  let cells = model.cells;
  let trusted = true;
  for (let i = 0; i < cells.length; i++) {
    let cell = cells.at(i);
    if (!cell.trusted) {
      trusted = false;
    }
  }
  if (trusted) {
    return showDialog({
      body: 'Notebook is already trusted',
      buttons: [okButton]
    }).then(() => void 0);
  }
  return showDialog({
    body: TRUST_MESSAGE,
    title: 'Trust this notebook?',
    buttons: [cancelButton, warnButton]
  }).then(result => {
    if (result.text === 'OK') {
      for (let i = 0; i < cells.length; i++) {
        let cell = cells.at(i);
        cell.trusted = true;
      }
    }
  });
}