Function load

crates/devsql/src/providers/imports.rs:39–92

(conn: &Connection, repo_path: &Path)

"#;

pub fn load(conn: &Connection, repo_path: &Path) -> Result<()> {
    ensure_table(conn)?;

    let files = walk_source_files(repo_path);
    conn.execute_batch("BEGIN")?;
    let mut stmt = conn.prepare(INSERT_SQL)?;
    let mut parser = TsParser::new();

    for file in files {
        if file.size > MAX_FILE_SIZE {
            continue;
        }

        let language = detect_language(&file.extension);
        let Some(lang_kind) = TsLanguageKind::from_name(language) else {
            continue;
        };

        // Currently only TypeScript/JavaScript imports are implemented.
        if !matches!(
            lang_kind,
            TsLanguageKind::TypeScript | TsLanguageKind::Tsx | TsLanguageKind::JavaScript | TsLanguageKind::Jsx
        ) {
            continue;
        }

        let abs_path = repo_path.join(&file.path);
        let Ok(contents) = std::fs::read_to_string(&abs_path) else {
            continue;
        };

        let line_index = LineIndex::new(&contents);
        let Some(tree) = parser.parse(language, &contents) else {
            continue;
        };

        let rows = extract_imports_from_tree(
            lang_kind,
            &tree,
            &line_index,
            &contents,
            &file.path,
        );

        for row in rows {
            insert_row(&mut stmt, &row)?;
        }
    }

    drop(stmt);
    conn.execute_batch("COMMIT")?;
    create_indexes(conn)?;
    Ok(())
}

fn ensure_table(conn: &Connection) -> Result<()> {
    conn.execute_batch(CREATE_TABLE_SQL)?;