data.ts

/**
 * Copyright (c) 2018 mol* contributors, licensed under MIT, See LICENSE file for more info.
 *
 * @author David Sehnal <david.sehnal@gmail.com>
 */

import { PluginStateTransform } from '../base';
import { PluginStateObjects as SO } from '../objects';
import { Task } from 'mol-task';
import CIF from 'mol-io/reader/cif'
import { PluginContext } from 'mol-plugin/context';
import { ParamDefinition as PD } from 'mol-util/param-definition';

export { Download }
namespace Download { export interface Params { url: string, isBinary?: boolean, label?: string } }
const Download = PluginStateTransform.Create<SO.DataRoot, SO.Data.String | SO.Data.Binary, Download.Params>({
    name: 'download',
    display: {
        name: 'Download',
        description: 'Download string or binary data from the specified URL'
    },
    from: [SO.DataRoot],
    to: [SO.Data.String, SO.Data.Binary],
    params: {
        default: () => ({
            url: 'https://www.ebi.ac.uk/pdbe/static/entry/1cbs_updated.cif'
        }),
        controls: () => ({
            url: PD.Text('URL', 'Resource URL. Must be the same domain or support CORS.', ''),
            isBinary: PD.Boolean('Binary', 'If true, download data as binary (string otherwise)', false)
        })
    },
    apply({ params: p }, globalCtx: PluginContext) {
        return Task.create('Download', async ctx => {
            // TODO: track progress
            const data = await globalCtx.fetch(p.url, p.isBinary ? 'binary' : 'string');
            return p.isBinary
                ? new SO.Data.Binary({ label: p.label ? p.label : p.url }, data as Uint8Array)
                : new SO.Data.String({ label: p.label ? p.label : p.url }, data as string);
        });
    }
});

export { ParseCif }
namespace ParseCif { export interface Params { } }
const ParseCif = PluginStateTransform.Create<SO.Data.String | SO.Data.Binary, SO.Data.Cif, ParseCif.Params>({
    name: 'parse-cif',
    display: {
        name: 'Parse CIF',
        description: 'Parse CIF from String or Binary data'
    },
    from: [SO.Data.String, SO.Data.Binary],
    to: [SO.Data.Cif],
    apply({ a }) {
        return Task.create('Parse CIF', async ctx => {
            const parsed = await (SO.Data.String.is(a) ? CIF.parse(a.data) : CIF.parseBinary(a.data)).runInContext(ctx);
            if (parsed.isError) throw new Error(parsed.message);
            return new SO.Data.Cif({ label: 'CIF File' }, parsed.result);
        });
    }
});