stanford-oval / genie-toolkit / 610

pending completion

Build # 610

Build Type

Pull #130

travis-ci-com

Committed by

web-flow

Commit Message

Fix tests

Pull Request Pull Request #130: Automate schema.org processsing

Run Details

1680 of 4186 branches covered (40.13%)

Branch coverage included in aggregate %.

49 of 173 new or added lines in 7 files covered. (28.32%)

3 existing lines in 3 files now uncovered.

4280 of 8423 relevant lines covered (50.81%)

559547.03 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

5.28

/tool/webqa-process-schemaorg.js

// -*- mode: js; indent-tabs-mode: nil; js-basic-offset: 4 -*-
//
// This file is part of Genie
//
// Copyright 2019 The Board of Trustees of the Leland Stanford Junior University
//
// Author: Giovanni Campagna <gcampagn@cs.stanford.edu>
//
// See COPYING for details
"use strict";

const assert = require('assert');
const POS = require("en-pos");
const Tp = require('thingpedia');
const ThingTalk = require('thingtalk');
const Type = ThingTalk.Type;
const Ast = ThingTalk.Ast;
const fs = require('fs');
const util = require('util');

const { clean, pluralize } = require('../lib/utils');
const StreamUtils = require('../lib/stream-utils');

const keepAnnotation = false;

function getId(id) {
    assert(id.startsWith('http://schema.org/'));
    return id.substring('http://schema.org/'.length);
}

function getIncludes(includes) {
    if (Array.isArray(includes))
        return includes.map((incl) => getId(incl['@id']));
    else
        return [getId(includes['@id'])];
}

const BUILTIN_TYPEMAP = {
    Time: Type.Time,
    Number: Type.Number,
    Float: Type.Number,
    Integer: Type.Number,
    Text: Type.String,
    Boolean: Type.Boolean,
    DateTime: Type.Date,
    Date: Type.Date,
    DataType: Type.Any,
    URL: Type.Entity('tt:url'),
    ImageObject: Type.Entity('tt:picture'),
    Barcode: Type.Entity('tt:picture'),

    Mass: Type.Measure('kg'),
    Energy: Type.Measure('kcal'),
    Distance: Type.Measure('m'),
    Duration: Type.Measure('ms'),

    GeoCoordinates: Type.Location,
    MonetaryAmount: Type.Currency,

    QuantitativeValue: Type.Any
};

const KEYWORDS = [
    'let', 'now', 'new', 'as', 'of', 'in', 'out', 'req', 'opt', 'notify', 'return',
    'join', 'edge', 'monitor', 'class', 'extends', 'mixin', 'this', 'import', 'null',
    'enum', 'aggregate', 'dataset', 'oninput', 'sort', 'asc', 'desc', 'bookkeeping',
    'compute', 'true', 'false'
];

const BLACKLISTED_TYPES = new Set([
    'QualitativeValue', 'PropertyValue', 'BedType', 'MedicalBusiness',

    // buggy, causes Audience to turn into an enum
    'Researcher',
]);

const BLACKLISTED_PROPERTIES = new Set([
    'sameAs', 'affiliation', 'mainEntityOfPage',
    'embedUrl',

    // FIXME we want to black-list aggregateRating.itemReviewed but not Review.itemReviewed...
    'itemReviewed',

     // This is used as the range of rating
    'bestRating', 'worstRating',

    // renamed to description during normalization
    'reviewBody',

    // this causes a loop in PriceSpecification, which turns PriceSpecification into an Entity and that sucks
    'eligibleTransactionVolume',
    // same thing, causes a loop in Offer which is bad
    'addOn',

    // not particularly useful, and kind of confusing
    'areaServed',

    // handled specially by normalization
    'priceCurrency'
]);

const STRUCTURED_HIERARCHIES = [
    'StructuredValue', 'Rating', 'Offer',

    // FIXME Review is too messy to represent as a structured value, either you lose info or you get cycles
    // 'Review'
];

const NON_STRUCT_TYPES = new Set([
]);

const PROPERTY_FORCE_ARRAY = new Set([
    'worksFor',

    'recipeCuisine',
    'recipeCategory',
]);

const PROPERTY_FORCE_NOT_ARRAY = new Set([
    'offers'
]);

const PROPERTY_TYPE_OVERRIDE = {
    'telephone': Type.Entity('tt:phone_number'),
    'email': Type.Entity('tt:email_address'),
    'image': Type.Entity('tt:picture'),
    'logo': Type.Entity('tt:picture'),
    'checkinTime': Type.Time,
    'checkoutTime': Type.Time,
    'price': Type.Currency,

    'weight': Type.Measure('ms'),
    'depth': Type.Measure('m'),
    'description': Type.String,
    'addressCountry': Type.Entity('tt:country'),
    'addressRegion': Type.Entity('tt:us_state'),

    // we want to prefer VideoObject to the default Clip
    'video': Type.Entity('org.schema:VideoObject'),

    // we want to prefer Organization to the default Person
    'publisher': Type.Entity('org.schema:Organization'),

    // weird number like things, but mostly text
    'recipeYield': Type.String
};

const PROPERTY_CANONICAL_OVERRIDE = {
    // thing
    url: {
        default: 'npp',
        npp: ['url', 'link']
    },
    name: {
        default: 'npp',
        pvp: ['called'],
        npp: ['name']
    },
    description: {
        default: 'npp',
        npp: ['description', 'summary']
    },

    // location
    'geo': { default:"npp", npp:['location', 'address'], pvp:["in", "around", "at", "on"] },
    'streetAddress': { default:"npp", npp:['street'] },
    'addressCountry': { default:"pvp", pvp:["in"], npp:["country"] },
    'addressRegion': { default:"pvp", pvp:["in"], npp:["state"] },
    'addressLocality': { default:"npp", npp:['city'] }
};

const MANUAL_PROPERTY_CANONICAL_OVERRIDE = {
    // restaurants
    'datePublished': { default:"pvp", pvp:["published on", "written on"], npp:["date published"] },
    'ratingValue': { default:"pvp", pvp:["rated #star"], npp:["rating"] },
    'reviewRating': { default:"npp", npp:["rating"] },
    'telephone': { default:"npp", npp:["telephone", "phone number"] },
    'servesCuisine': { default:"apv", apv:true, avp:["serves #cuisine", "serves #food", "offer #cuisine", "offer #food", "serves", "offers"],
        npp:["#cuisine", "#food"] },

    // hotels
    'amenityFeature': {
        default: 'npp',
        npp: ['amenity', 'amenity feature'],
        avp: ['offers', 'offer', 'has', 'have'],
    },
    'checkinTime': {
        default: 'npp',
        npp: ['checkin time', 'check in time', 'check-in time']
    },
    'checkoutTime': {
        default: 'npp',
        npp: ['checkout time', 'check out time', 'check-out time']
    },

    // linkedin
    alumniOf: {
        default: 'npi',
        npi: [
        // who is an alumnus of Stanford
        "alumni of", "alumnus of", "alumna of",
        // who is a Stanford alumnus
        "#alumnus", "#alumni", "#grad", "#gradudate"
        ],
        avp: [
        // who was educated at Stanford
        "was educated at", "is graduated from", "was studied at", "went to", "graduated from", "attended"
        ],
        pvp: [
        // what person educated at Stanford ...
        "educated at", "graduated from", "studied at", "attended"
        ],
        npp: [
        // who has alma mater ...
        'colleges', 'universities', "alma maters"
        ]
    },
    award: {
        default: 'avp',
        npi: [
            // who is a nobel prize winner
            'winner of', 'recipient of',
            '#winner', '#awardee', '#recipient', '#holder',
        ],
        avp: [
        "has the award", "has received the #award", "won the award for", "won the #award",
        "received the #award", "received the", "won the", "won", "holds the award for", "holds the #award"
        ],
        pvp: [
            "received"
        ],
        npp: ['awards']
    },
    affiliation: {
        default: 'npi',
        npi: [
            'affiliated with', 'affiliated to', 'member of'
        ],
        npp: ['affiliations']
    },
    worksFor: {
        default: 'avp',
        npi: [
            'employee of', '#employee'
        ],
        avp: ['works for', 'works at', 'worked at', 'worked for'],
        pvp: [
            'employed at', 'employed by',
        ],
        npp: ['jobs', 'employers']
    },
    addressLocality: {
        default: 'avp',
        npp: ['city'],
        avp: ['lives in', 'lived in'],
        pvp: ['from', 'in']
    },

    // recipes
    author: {
        default: 'pvp',
        avp: [
            'was written by', 'was submitted by',
        ],
        pvp: [
            'by', 'made by', 'written by', 'created by', 'authored by', 'uploaded by', 'submitted by'
        ],
        npp: ['author', 'creator']
    },
    publisher: {
        default: 'pvp',
        avp: [
            'was published by', 'was submitted by'
        ],
        pvp: [
            'by', 'made by', 'published by'
        ],
        npp: ['publisher']
    },

    prepTime: {
        default: 'npp',
        avp: ['takes #to_prepare', 'needs #to_prepare'],
        npp: ['prep time', 'preparation time', 'time to prep', 'time to prepare']
    },
    cookTime: {
        default: 'npp',
        avp: ['takes #to_cook', 'needs #to_cook'],
        npp: ['cook time', 'cooking time', 'time to cook']
    },
    totalTime: {
        default: 'avp',
        avp: ['takes', 'requires', 'needs', 'uses', 'consumes'],
        npp: ['total time', 'time in total', 'time to make']
    },
    recipeYield: {
        default: 'avp',
        avp: ['yields', 'feeds', 'produces', 'results in', 'is good for'],
        pvp: ['yielding'],
        npp: ['yield amount', 'yield size']
    },
    recipeCategory: {
        default: 'npp',
        npp: ['categories']
    },
    recipeIngredient: {
        default: 'npp',
        avp: ['contains', 'uses', 'has'],
        pvp: ['containing', 'using'],
        npp: ['ingredients']
    },
    recipeInstructions: {
        default: 'npp',
        npp: ['instructions']
    },
    recipeCuisines: {
        default: 'avp',
        apv: true,
        avp: ['belongs to the #cuisine'],
        npp: ['cuisines']
    },
    reviewBody: {
        default: 'npp',
        npp: ['body', 'text', 'content']
    },
    saturatedFatContent: {
        default: 'npp',
        npp: ['saturated fat content', 'saturated fat amount', 'saturated fat', 'trans far']
    },

    // product
    mpn: {
        default: 'npp',
        npp: ['manufacturer part number']
    }
};

const PROPERTIES_NO_FILTER = [
    'name', // no filter on name, if the id has ner support, we'll generate prim for it
    'priceRange',

    // ID properties or opaque strings
    'gtin13',
    'productID',
    'mpn'
];

const PROPERTIES_DROP_WITH_GEO = [
    'streetAddress', // street address and address locality should be handled by geo
    'addressLocality'
];

// HACK: certain structured types want to get the name & description property from Thing
const STRUCT_INCLUDE_THING_PROPERTIES = new Set([
    'LocationFeatureSpecification'
]);



function posTag(tokens) {
    return new POS.Tag(tokens)
        .initial() // initial dictionary and pattern based tagging
        .smooth() // further context based smoothing
        .tags;
}

function getItemType(typename, typeHierarchy) {
    // use conventions on the typename to convert an array type to its element type

    for (let suffix of ['List', 'Collection', 'Section', 'Catalog']) {
        if (typename.endsWith(suffix)) {
            const itemname = typename.substring(0, typename.length - suffix.length);
            if (itemname in typeHierarchy)
                return itemname;
            else
                return 'Thing';
        }
    }

    console.error(`ItemList subclass ${typename} does not have a recognized suffix`);
    return 'Thing';
}

const STRING_FILE_OVERRIDES = {
    'org.schema.Restaurant:Restaurant_name': 'com.yelp:restaurant_names',
    'org.schema.Person:Person_name': 'tt:person_full_name',
    'org.schema.Person:Person_alumniOf': 'tt:university_names',
    'org.schema.Person:Person_worksFor': 'tt:company_name',
    'org.schema.Hotel:Hotel_name': 'tt:hotel_name'
};

function recursiveAddStringValues(arg, fileId) {
    let type = arg.type;
    while (type.isArray)
        type = type.elem;

    if (type.isEntity && STRING_FILE_OVERRIDES[fileId]) {
        arg.annotations['string_values'] = Ast.Value.String(STRING_FILE_OVERRIDES[fileId]);
        return;
    }

    if (type.isString) {
        arg.annotations['string_values'] = Ast.Value.String(STRING_FILE_OVERRIDES[fileId] || fileId);
        return;
    }

    if (type.isCompound) {
        for (let field in type.fields) {
            if (field.indexOf('.') >= 0)
                continue;
            recursiveAddStringValues(type.fields[field], fileId + '_' + field);
        }
    }
}

class SchemaProcessor {
    constructor(args) {
        this._output = args.output;
        this._cache = args.cache_file;
        this._className = args.class_name;
        this._url = args.url;
        this._manual = args.manual;
        this._hasGeo = false;
        this._prefix = args.class_name ? `org.schema.${args.class_name}:` : `org.schema:`;
    }

    typeToThingTalk(typename, typeHierarchy, manualAnnotation) {
        if (typename in BUILTIN_TYPEMAP)
            return BUILTIN_TYPEMAP[typename];

        if (typeHierarchy[typename].isItemList)
            return Type.Array(this.typeToThingTalk(typeHierarchy[typename].itemType, typeHierarchy, manualAnnotation));
        if (typeHierarchy[typename].isEnum && typeHierarchy[typename].enum.length > 0)
            return Type.Enum(typeHierarchy[typename].enum);
        if (typeHierarchy[typename].representAsStruct)
            return this.makeCompoundType(typename, typeHierarchy[typename], typeHierarchy, manualAnnotation);

        return Type.Entity(this._prefix + typename);
    }

    getBestPropertyType(propname, property, typeHierarchy, manualAnnotation) {
        if (BLACKLISTED_PROPERTIES.has(propname))
            return [undefined, undefined];

        let best = undefined, bestScore = -Infinity;

        // if the property is defined as taking ItemList and something else, we make an array of that something else
        let isArray = property.types.some((type) => typeHierarchy[type] && typeHierarchy[type].isItemList);

        // if the property comment starts with "A " or "An ", we assume there can be multiple values
        // because if it starts with "The ", we assume it can only have one value
        // this is a pretty coarse heuristic, but it works sometimes...

        if (/^an? /i.test(property.comment))
            isArray = true;
        if (PROPERTY_FORCE_ARRAY.has(propname))
            isArray = true;
        if (PROPERTY_FORCE_NOT_ARRAY.has(propname))
            isArray = false;

        // prefer enum if possible
        // then specific data types
        // then fallback to a struct type if one is listed
        // then fallback to text if it's explicitly listed as one of the types
        // then fallback to an entity type

        for (let type of property.types) {
            let score;
            if (typeHierarchy[type] && typeHierarchy[type].isEnum)
                score = 5;
            else if (type === 'Text')
                score = 2;
            else if (type in BUILTIN_TYPEMAP)
                score = 4;
            else if (!typeHierarchy[type])
                score = -1;
            else if (typeHierarchy.isItemList) // ItemList and subclasses are useless
                score = 0;
            else if (typeHierarchy[type].representAsStruct)
                score = 3;
            else
                score = 1;

            if (score > bestScore) {
                best = type;
                bestScore = score;
            }
        }

        // if we didn't find a type we like, return nothing
        if (bestScore < 0)
            return [undefined, undefined];

        if (propname in PROPERTY_TYPE_OVERRIDE)
            return [best, PROPERTY_TYPE_OVERRIDE[propname]];

        // if we chose an item list as the best type, don't wrap into a further array
        if (typeHierarchy[best] && typeHierarchy[best].isItemList)
            isArray = false;

        // HACK
        if (best === 'QuantitativeValue') {
            if (/number/i.test(propname) || /level/i.test(propname) || /quantity/i.test(propname))
                return [best, Type.Number];
            if (/duration/i.test(propname))
                return [best, Type.Measure('ms')];

            console.error(`Cannot guess the correct type of ${propname} of type QuantitativeValue, assuming Number`);
            return [best, Type.Number];
        }

        let tttype = this.typeToThingTalk(best, typeHierarchy, manualAnnotation);
        if (!tttype)
            return [undefined, undefined];

        // an array of booleans or enums does not make much sense
        if (tttype.isBoolean || tttype.isEnum)
            isArray = false;

        if (isArray)
            tttype = Type.Array(tttype);
        return [best, tttype];
    }

    makeCompoundType(startingTypename, typedef, typeHierarchy) {
        const fields = {};

        // collect all properties of this type (incl. inherited ones)
        let allproperties = new Map;
        function recursiveCollectProperties(typename) {
            //console.error(typename);
            const typedef = typeHierarchy[typename];
            if (!typedef)
                return;
            // if something is a subclass of both a struct and non-struct,
            // we ignore the properties coming from the non-struct side
            // (unless the leaf type name we're starting from is explicitly
            // marking as going all the way up)
            if (!STRUCT_INCLUDE_THING_PROPERTIES.has(startingTypename) && !typeHierarchy[typename].isStructSubType)
                return;
            for (let propertyname in typedef.properties) {
                const propertydef = typedef.properties[propertyname];
                if (allproperties.has(propertyname))
                    continue;
                allproperties.set(propertyname, propertydef);
            }
            // stop at the base struct types (so we don't include Thing properties)
            if (!STRUCT_INCLUDE_THING_PROPERTIES.has(startingTypename) && STRUCTURED_HIERARCHIES.indexOf(typename) >= 0)
                return;

            for (let _extends of typeHierarchy[typename].extends)
                recursiveCollectProperties(_extends);
        }
        recursiveCollectProperties(startingTypename);

        let anyfield = false;
        for (let [propertyname, propertydef] of allproperties) {
            const [schemaOrgType, ttType] = this.getBestPropertyType(propertyname, propertydef, typeHierarchy);
            if (!ttType)
                continue;

            const canonical = this.makeArgCanonical(propertyname, ttType);
            const metadata = { canonical };
            const annotation = keepAnnotation ? {
                'org_schema_type': Ast.Value.String(schemaOrgType),
                'org_schema_comment': Ast.Value.String(propertydef.comment)
            } : {
                'org_schema_type': Ast.Value.String(schemaOrgType)
            };

            if (PROPERTIES_NO_FILTER.includes(propertyname)) {
                annotation['genie'] = new Ast.Value.Boolean(false);
            } else if (this._hasGeo && PROPERTIES_DROP_WITH_GEO.includes(propertyname)) {
                annotation['genie'] = new Ast.Value.Boolean(false);
                annotation['drop'] = new Ast.Value.Boolean(true);
            }

            fields[propertyname] = new Ast.ArgumentDef(null, undefined, propertyname, ttType, {
                nl: metadata,
                impl: annotation
            });
            anyfield = true;
        }
        if (!anyfield)
            throw new Error(`Struct type ${startingTypename} has no fields`);

        return Type.Compound(startingTypename, fields);
    }

    makeArgCanonical(name, ptype) {
        function cleanName(name) {
            if (name.endsWith(' value'))
                return name.substring(0, name.length - ' value'.length);
            return name;
        }

        if (name in PROPERTY_CANONICAL_OVERRIDE)
            return PROPERTY_CANONICAL_OVERRIDE[name];
        if (this._manual && name in MANUAL_PROPERTY_CANONICAL_OVERRIDE)
            return MANUAL_PROPERTY_CANONICAL_OVERRIDE[name];

        let canonical = {};
        let npp;
        let plural = ptype && ptype.isArray;
        name = clean(name);
        if (!name.includes('.')) {
            npp = plural ? pluralize(cleanName(name)) : cleanName(name);
        } else {
            const components = name.split('.');
            const last = components[components.length - 1];
            npp = plural ? pluralize(last) : last;
        }

        if (npp.endsWith(' content') && ptype.isMeasure) {
            npp = npp.substring(0, npp.length - ' content'.length);
            canonical = {
                default: 'npp',
                avp: ['contains #' + npp.replace(/ /g, '_')],
                npp: [npp + ' content', npp, npp + ' amount']
            };
            return canonical;
        }

        if (npp.startsWith('has ')) {
            npp = npp.substring('has '.length);
        } else if (npp.startsWith('is ')) {
            npp = npp.substring('is '.length);
            let tags = posTag(npp.split(' '));

            if (['NN', 'NNS', 'NNP', 'NNPS'].includes(tags[tags.length - 1]) || npp.endsWith(' of')) {
                canonical["npi"] = [npp];
                canonical["default"] = "npi";
            }
            else if (['VBN', 'JJ', 'JJR'].includes(tags[0])) {
                canonical["pvp"] = [npp];
                canonical["default"] = "pvp";
            }

        } else {
            let tags = posTag(npp.split(' '));
            if (['VBP', 'VBZ'].includes(tags[0])) {
                if (tags.length === 2 && ['NN', 'NNS', 'NNP', 'NNPS'].includes(tags[1])) {
                    canonical["avp"] = [npp.replace(' ', ' #')];
                    npp = npp.split(' ')[1];
                } else {
                    canonical["avp"] = [npp];
                }
                canonical["default"] = "avp";
            } else if (npp.endsWith(' of')) {
                canonical["npi"] = [npp];
                canonical["default"] = "npi";
            } else if (['VBN', 'JJ', 'JJR'].includes(tags[0]) && !['NN', 'NNS', 'NNP', 'NNPS'].includes(tags[tags.length - 1])) {
                // this one is actually somewhat problematic
                // e.g., all non-words are recognized as JJ, including issn, dateline, funder
                canonical["pvp"] = [npp];
                canonical["default"] = "pvp";
            }

        }

        canonical["npp"] = [npp];
        if (!("default" in canonical))
            canonical["default"] = "npp";

        return canonical;
    }

    async run() {
        let schemajsonld;
        if (await util.promisify(fs.exists)(this._cache)) {
            schemajsonld = await util.promisify(fs.readFile)(this._cache, { encoding: 'utf8' });
        } else {
            schemajsonld = await Tp.Helpers.Http.get(this._url);
            await util.promisify(fs.writeFile)(this._cache, schemajsonld);
        }

        // type_name -> {
        //    extends: [type_name],
        //    properties: { name -> { types: [type], comment: ... } },
        //    comment: ...
        // }
        const typeHierarchy = {};
        function ensureType(typename) {
            if (typeHierarchy[typename])
                return;
            typeHierarchy[typename] = {
                extends: [],
                properties: {},
                comment: ''
            };
        }
        function isSubClass(typename, subtypeof) {
            for (let _extend of typeHierarchy[typename].extends) {
                if (_extend === subtypeof)
                    return true;

                if (!typeHierarchy[_extend])
                    continue;
                if (isSubClass(_extend, subtypeof))
                    return true;
            }
            return false;
        }

        const enums = {};
        function ensureEnum(enumname) {
            if (enums[enumname])
                return;
            enums[enumname] = [];
        }

        for (let triple of JSON.parse(schemajsonld)['@graph']) {
            try {
                if (getId(triple['@id']) in BUILTIN_TYPEMAP)
                    continue;

                if (BLACKLISTED_TYPES.has(getId(triple['@id'])))
                    continue;

                if (triple['@type'].startsWith('http://schema.org/')) {
                    // an enum declaration
                    const enumtype = getId(triple['@type']);
                    const enumvalue = getId(triple['@id']);
                    ensureEnum(enumtype);
                    enums[enumtype].push(enumvalue);
                    continue;
                }

                switch (triple['@type']) {
                case 'rdf:Property': {
                    // ignore deprecated stuff
                    if (triple['http://schema.org/supersededBy'])
                        continue;


                    const domains = getIncludes(triple['http://schema.org/domainIncludes']);
                    const ranges = getIncludes(triple['http://schema.org/rangeIncludes']);
                    const name = getId(triple['@id']);
                    const comment = triple['rdfs:comment'];

                    if (BLACKLISTED_PROPERTIES.has(name))
                        continue;

                    for (let domain of domains) {
                        if (domain in BUILTIN_TYPEMAP)
                            continue;
                        if (BLACKLISTED_TYPES.has(domain))
                            continue;

                        ensureType(domain);
                        typeHierarchy[domain].properties[name] = {
                            types: ranges,
                            comment
                        };
                    }
                    break;
                }
                case 'rdfs:Class': {
                    const name = getId(triple['@id']);
                    const comment = triple['rdfs:comment'];
                    const _extends = getIncludes(triple['rdfs:subClassOf'] || []);
                    ensureType(name);
                    typeHierarchy[name].extends = _extends.filter((ex) => !BLACKLISTED_TYPES.has(ex));
                    if (typeHierarchy[name].extends.length === 0 && name !== 'Thing')
                        typeHierarchy[name].extends = ['Thing'];
                    typeHierarchy[name].comment = comment;
                    break;
                }

                default:
                    throw new Error(`don't know how to handle a triple of type ${triple['@type']}`); //'
                }
            } catch(e) {
                console.error('Triple failed');
                console.error(triple);
                throw e;
            }
        }


        for (let type in typeHierarchy) {
            typeHierarchy[type].isAction = isSubClass(type, 'Action');
            typeHierarchy[type].isEnum = !!enums[type] || isSubClass(type, 'Enumeration');
            if (typeHierarchy[type].isEnum)
                typeHierarchy[type].enum = enums[type] || [];

            typeHierarchy[type].isItemList = isSubClass(type, 'ItemList');
            if (typeHierarchy[type].isItemList)
                typeHierarchy[type].itemType = getItemType(type, typeHierarchy);

            if (STRUCTURED_HIERARCHIES.indexOf(type) >= 0) {
                typeHierarchy[type].isStructSubType = true;
                typeHierarchy[type].representAsStruct = true;
            } else {
                for (let structBase of STRUCTURED_HIERARCHIES) {
                    if (isSubClass(type, structBase)) {
                        typeHierarchy[type].isStructSubType = true;
                        typeHierarchy[type].representAsStruct = true;
                        break;
                    }
                }
            }

            if (NON_STRUCT_TYPES.has(type)) {
                typeHierarchy[type].isStructSubType = false;
                typeHierarchy[type].representAsStruct = false;
            }
        }

        function findCycle(typename, lookfor, visited, cycle = []) {
            if (visited.has(typename)) {
                if (typename === lookfor)
                    console.error('Found cycle for ' + typename, cycle, visited);
                return typename === lookfor;
            }
            visited.add(typename);

            for (let propname in typeHierarchy[typename].properties) {
                let propdef = typeHierarchy[typename].properties[propname];
                for (let type of propdef.types) {
                    if (type in BUILTIN_TYPEMAP)
                        continue;
                    if (!typeHierarchy[type] || !typeHierarchy[type].representAsStruct)
                        continue;
                    cycle.push(propname);
                    if (findCycle(type, lookfor, visited, cycle))
                        return true;
                    cycle.pop();
                }
            }
            return false;
        }

        // check all types - if they form a cycle, we cannot represent them as structs
        for (let typename in typeHierarchy) {
            if (typeHierarchy[typename].isEnum)
                continue;
            if (!typeHierarchy[typename].representAsStruct)
                continue;
            if (findCycle(typename, typename, new Set))
                typeHierarchy[typename].representAsStruct = false;
        }

        // check all types - all parents of non-struct types must also be non-struct types,
        // recursively
        function recursiveMakeNonStruct(typename) {
            typeHierarchy[typename].representAsStruct = false;
            for (let _extend of typeHierarchy[typename].extends) {
                if (!typeHierarchy[_extend])
                    continue;
                recursiveMakeNonStruct(_extend);
            }
        }

        for (let typename in typeHierarchy) {
            if (typeHierarchy[typename].isEnum)
                continue;
            if (typeHierarchy[typename].representAsStruct)
                continue;
            recursiveMakeNonStruct(typename);
        }

        //console.log(JSON.stringify(typeHierarchy, undefined, 2));

        const order = new Set;

        function toposort(typename) {
            if (typeHierarchy[typename].isAction || typeHierarchy[typename].isEnum ||
                typeHierarchy[typename].representAsStruct)
                return;

            for (let _extend of typeHierarchy[typename].extends) {
                if (!typeHierarchy[_extend])
                    continue;
                toposort(_extend);
            }

            order.add(typename);
        }
        for (let type in typeHierarchy) {
            if (order.has(type))
                continue;
            toposort(type);
        }

        const queries = {};
        for (let typename of order) {
            const typedef = typeHierarchy[typename];

            // do not generate a class for ItemList and subclasses
            if (typename === 'ItemList' || typedef.isItemList)
                continue;

            const args = [
                new Ast.ArgumentDef(null, Ast.ArgDirection.OUT, 'id', Type.Entity(this._prefix + typename), {
                    nl: {},
                    impl: {
                        'unique': new Ast.Value.Boolean(true),
                        'genie': new Ast.Value.Boolean(false) // no filter on id, if it has ner support, we'll generate prim for it
                    }
                })
            ];
            recursiveAddStringValues(args[0], this._prefix + typename + '_name');
            if (typename !== 'Thing') {
                // override name so we can apply a custom string_values annotation
                const arg = new Ast.ArgumentDef(null, Ast.ArgDirection.OUT, 'name', Type.String, {
                    nl: {},
                    impl: {
                        'org_schema_type': new Ast.Value.String('Text'),
                        'genie': new Ast.Value.Boolean(false) // no filter on name, if it has ner support, we'll generate prim for it
                    }
                });
                recursiveAddStringValues(arg, this._prefix + typename + '_name');
                args.push(arg);
            }

            this._hasGeo = 'geo' in typedef.properties;
            for (let propertyname in typedef.properties) {
                const propertydef = typedef.properties[propertyname];
                const [schemaOrgType, type] = this.getBestPropertyType(propertyname, propertydef, typeHierarchy);
                if (!type)
                    continue;

                if (KEYWORDS.includes(propertyname))
                    propertyname = '_' + propertyname;

                const canonical = this.makeArgCanonical(propertyname, type);
                const metadata = { canonical };
                const annotation = keepAnnotation ? {
                    'org_schema_type': Ast.Value.String(schemaOrgType),
                    'org_schema_comment': Ast.Value.String(propertydef.comment)
                } : {
                    'org_schema_type': Ast.Value.String(schemaOrgType)
                };

                if (PROPERTIES_NO_FILTER.includes(propertyname))
                    annotation['genie'] = new Ast.Value.Boolean(false);

                const arg = new Ast.ArgumentDef(null, Ast.ArgDirection.OUT, propertyname, type, {
                    nl: metadata,
                    impl: annotation
                });
                recursiveAddStringValues(arg, this._prefix + typename + '_' + propertyname);

                args.push(arg);
            }

            if (KEYWORDS.includes(typename))
                typename = '_' + typename;
            queries[typename] = new Ast.FunctionDef(null, 'query', null /* class */, typename,
                typedef.extends, {
                    is_list: true,
                    is_monitorable: false,
                }, args, {
                    nl: {
                        'canonical': clean(typename),
                        'confirmation': clean(typename),
                    },
                    impl: keepAnnotation ? {
                        'org_schema_comment': Ast.Value.String(typedef.comment),
                        'confirm': Ast.Value.Boolean(false)
                    } : {
                        'confirm': Ast.Value.Boolean(false)
                    }
                });
        }

        const imports = [
            new Ast.ImportStmt.Mixin(null, ['loader'], 'org.thingpedia.v2', []),
            new Ast.ImportStmt.Mixin(null, ['config'], 'org.thingpedia.config.none', [])
        ];

        const classdef = new Ast.ClassDef(null,
            `org.schema${this._className ? '.' + this._className : ''}`,
            [], { queries, imports }, {
            nl: {
                name: `${this._className ? this._className + ' in ' : ''}Schema.org`,
                description: 'Scraped data from websites that support schema.org'
            },
            impl: {}
        }, {
            is_abstract: false
        });

        this._output.end(classdef.prettyprint());
        await StreamUtils.waitFinish(this._output);
    }

}


module.exports = {
    initArgparse(subparsers) {
        const parser = subparsers.addParser('webqa-process-schemaorg', {
            addHelp: true,
            description: "Process a schema.org JSON+LD definition into a Thingpedia class."
        });
        parser.addArgument(['-o', '--output'], {
            required: true,
            type: fs.createWriteStream
        });
        parser.addArgument(['--cache-file'], {
            required: false,
            defaultValue: './schema.jsonld',
            help: 'Path to a cache file containing the schema.org definitions.'
        });
        parser.addArgument(['--url'], {
            required: false,
            defaultValue: 'https://schema.org/version/3.9/schema.jsonld',
            help: 'The schema.org URL to retrieve the definitions from.'
        });
        parser.addArgument('--manual', {
            nargs: 0,
            action: 'storeTrue',
            help: 'Enable manual annotations.',
            defaultValue: false
        });
        parser.addArgument('--class-name', {
            required: false,
            help: 'The name of the generated class, this will also affect the entity names'
        });
    },

    async execute(args) {
        const schemaProcessor = new SchemaProcessor(args);
        schemaProcessor.run();
    }
};

stanford-oval / genie-toolkit / 610

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous