seed category implications from yaml files

Author: ScriptRaccoon

databases/catdat/scripts/seed.yaml.ts

@@ -7,6 +7,7 @@ import type {
 	ConfigYaml,
 	CategoryPropertyYaml,
 	LemmaYaml,
+	CategoryImplicationYaml,
 } from './yaml.types'
 
 const db = get_client()
@@ -27,9 +28,9 @@ function seed() {
 
 	seed_lemmas()
 	seed_category_properties()
+	seed_category_implications()
 	seed_categories()
 
-	// TODO: seed implications
 	// TODO: seed functors
 	// TODO: seed all the remaining tables
 }
@@ -307,3 +308,56 @@ function seed_lemmas() {
 		}
 	})()
 }
+
+function seed_category_implications() {
+	const implications_folder = path.join(data_folder, 'category-implications')
+
+	const implication_files = fs
+		.readdirSync(implications_folder)
+		.filter((file) => file.endsWith('.yaml'))
+		.sort()
+
+	const implication_insert = db.prepare(
+		`INSERT INTO category_implications (
+	        id, reason, is_equivalence
+		) VALUES (?, ?, ?)`,
+	)
+
+	const assumption_insert = db.prepare(
+		`INSERT INTO category_implication_assumptions (
+			implication_id, property_id
+		) VALUES (?, ?)`,
+	)
+
+	const conclusion_insert = db.prepare(
+		`INSERT INTO category_implication_conclusions (
+			implication_id, property_id
+		) VALUES (?, ?)`,
+	)
+
+	db.transaction(() => {
+		for (const implication_file of implication_files) {
+			const content = fs.readFileSync(
+				path.join(implications_folder, implication_file),
+				'utf8',
+			)
+			const implications = YAML.parse(content) as CategoryImplicationYaml[]
+
+			for (const implication of implications) {
+				implication_insert.run(
+					implication.id,
+					implication.reason,
+					Number(implication.is_equivalence),
+				)
+
+				for (const assumption of implication.assumptions) {
+					assumption_insert.run(implication.id, assumption)
+				}
+
+				for (const conclusion of implication.conclusions) {
+					conclusion_insert.run(implication.id, conclusion)
+				}
+			}
+		}
+	})()
+}

databases/catdat/scripts/yaml.types.ts

@@ -68,3 +68,11 @@ export type LemmaYaml = {
 	claim: string
 	proof: string
 }
+
+export type CategoryImplicationYaml = {
+	id: string
+	assumptions: string[]
+	conclusions: string[]
+	reason: string
+	is_equivalence: boolean
+}