seed properties from yaml files

Author: ScriptRaccoon

databases/catdat/scripts/seed.yaml.ts

@@ -2,7 +2,11 @@ import fs from 'node:fs'
 import path from 'node:path'
 import { get_client } from './shared'
 import YAML from 'yaml'
-import type { CategoryYaml, ConfigYaml } from './yaml.types'
+import type {
+	CategoryYaml,
+	ConfigYaml,
+	PropertyYaml as CategoryPropertyYaml,
+} from './yaml.types'
 
 const db = get_client()
 
@@ -20,6 +24,16 @@ function seed() {
 
 	seed_config()
 
+	const category_properties = get_category_properties()
+
+	for (const property of category_properties) {
+		seed_category_property(property)
+	}
+
+	for (const property of category_properties) {
+		seed_related_and_dual_properties(property)
+	}
+
 	const categories = get_categories()
 
 	for (const category of categories) {
@@ -226,3 +240,54 @@ function seed_related_and_dual_categories(category: CategoryYaml) {
 		).run(category.id, rel)
 	}
 }
+
+function get_category_properties() {
+	const properties: CategoryPropertyYaml[] = []
+
+	const property_files = fs
+		.readdirSync(path.join(data_folder, 'category-properties'))
+		.filter((file) => file.endsWith('.yaml'))
+		.sort()
+
+	for (const property_file of property_files) {
+		const property_file_content = fs.readFileSync(
+			path.join(data_folder, 'category-properties', property_file),
+			'utf8',
+		)
+
+		const property = YAML.parse(property_file_content)
+		properties.push(property)
+	}
+
+	return properties
+}
+
+function seed_category_property(property: CategoryPropertyYaml) {
+	db.prepare(
+		`INSERT INTO category_properties
+		(id, relation, description, nlab_link,
+		dual_property_id, invariant_under_equivalences)
+		VALUES (?, ?, ?, ?, ?, ?)`,
+	).run([
+		property.id,
+		property.relation,
+		property.description,
+		property.nlab_link || null,
+		null,
+		Number(property.invariant_under_equivalences),
+	])
+}
+
+function seed_related_and_dual_properties(property: CategoryPropertyYaml) {
+	if (property.dual_property_id) {
+		db.prepare(
+			`UPDATE category_properties SET dual_property_id = ? WHERE id = ?`,
+		).run(property.dual_property_id, property.id)
+	}
+
+	for (const rel of property.related_properties) {
+		db.prepare(
+			`INSERT INTO related_category_properties (property_id, related_property_id) VALUES (?, ?)`,
+		).run(property.id, rel)
+	}
+}

databases/catdat/scripts/yaml.types.ts

@@ -62,3 +62,13 @@ type MorphismEntry = {
 	description: string
 	reason: string
 }
+
+export type PropertyYaml = {
+	id: string
+	relation: string
+	description: string
+	nlab_link?: string | null
+	dual_property_id?: string | null
+	invariant_under_equivalences: boolean
+	related_properties: string[]
+}