import db


def add_flow_edge(slug: str, screen_id: int, from_node_id: int, from_port: str | None, to_node_id: int) -> int:
    conn = db.connect(slug)
    conn.execute(
        "INSERT INTO _flow_edges (screen_id, from_node_id, from_port, to_node_id) VALUES (?, ?, ?, ?)",
        (screen_id, from_node_id, from_port or "out", to_node_id),
    )
    edge_id = int(conn.execute("SELECT last_insert_rowid() AS id").fetchone()["id"])
    conn.commit()
    conn.close()
    return edge_id