Composite Keys don't generate ref statements

[RiyeUK]

With a table defined in the schema as so:

export const warehouseOffices = pgTable(
    'warehouse_offices',
    {
        officeId: integer('office_id')
            .notNull()
            .references(() => offices.id),
        warehouseId: integer('warehouse_id')
            .notNull()
            .references(() => warehouses.id),
    },
    (table) => {
        return {
            pk: primaryKey({ columns: [table.officeId, table.warehouseId] }),
        };
    }
);

The generated .dbml doesn't include any refs for this table:

table warehouse_offices {
  office_id integer [not null]
  warehouse_id integer [not null]

  indexes {
    (office_id, warehouse_id) [pk]
  }
}

I need to add these lines to the resulting .dbml:

ref: warehouse_offices.warehouse_id - warehouses.id
ref: warehouse_offices.office_id - offices.id

[L-Mario564]

What version of the ORM are you using? Do you know if this worked in a previous version?

[stdrnddev]

Hey @L-Mario564 i had this issue too Wanted it fast so i changed your index.cjs to use this

var Columns = Symbol.for("drizzle:Columns");
const pkColumns = wrapColumns(Object.values(index.table[Columns]), this.buildQueryConfig.escapeName);

Also the references are missing the schema's name which you can do it like this

sourceTable: `${relation.sourceTable[Schema]}.${relation.sourceTable[TableName]}`,
foreignTable: `${relation.sourceTable[Schema]}.${relation.referencedTableName}`,

your escapeSpaces function will introduce quotes to {schema}.{tableName} after adding the schema so i changed the regex to /\s/

didn't open a PR because i just tested for my own situation but it may help you to handle this

currently using "drizzle-orm": "^0.30.4" this version for drizzle-orm