From 39795b3a4e6af3638780601225ea78bb5fc45e0a Mon Sep 17 00:00:00 2001
From: David Robertson <davidr@element.io>
Date: Wed, 15 Feb 2023 13:51:37 +0000
Subject: [PATCH] Make it easier to use DataGrip w/ Synapse's schema (#14982)

Also tweak the schema dump script:

- add a note explaining myself how to use it
-Explicitly call `poetry run`, because not everyone uses direnv :(
---
 changelog.d/14982.misc                        |   1 +
 contrib/datagrip/README.md                    |  28 ++++++++++++++++++
 contrib/datagrip/common.sql                   |   1 +
 contrib/datagrip/datagrip-aware-of-schema.png | Bin 0 -> 13610 bytes
 contrib/datagrip/main.sql                     |   1 +
 contrib/datagrip/schema_version.sql           |   1 +
 contrib/datagrip/state.sql                    |   1 +
 scripts-dev/make_full_schema.sh               |  20 +++++++++++--
 8 files changed, 50 insertions(+), 3 deletions(-)
 create mode 100644 changelog.d/14982.misc
 create mode 100644 contrib/datagrip/README.md
 create mode 120000 contrib/datagrip/common.sql
 create mode 100644 contrib/datagrip/datagrip-aware-of-schema.png
 create mode 120000 contrib/datagrip/main.sql
 create mode 120000 contrib/datagrip/schema_version.sql
 create mode 120000 contrib/datagrip/state.sql

diff --git a/changelog.d/14982.misc b/changelog.d/14982.misc
new file mode 100644
index 000000000..9aaa7ce26
--- /dev/null
+++ b/changelog.d/14982.misc
@@ -0,0 +1 @@
+Add a schema dump symlinks inside `contrib`, to make it easier for IDEs to interrogate Synapse's database schema.
diff --git a/contrib/datagrip/README.md b/contrib/datagrip/README.md
new file mode 100644
index 000000000..bbe4f3a5a
--- /dev/null
+++ b/contrib/datagrip/README.md
@@ -0,0 +1,28 @@
+# Schema symlinks
+
+This directory contains symlinks to the latest dump of the postgres full schema. This is useful to have, as it allows IDEs to understand our schema and provide autocomplete, linters, inspections, etc.
+
+In particular, the DataGrip functionality in IntelliJ's products seems to only consider files called `*.sql` when defining a schema from DDL; `*.sql.postgres` will be ignored. To get around this we symlink those files to ones ending in `.sql`. We've chosen to ignore the `.sql.sqlite` schema dumps here, as they're not intended for production use (and are much quicker to test against).
+
+## Example
+![](datagrip-aware-of-schema.png)
+
+## Caveats
+
+- Doesn't include temporary tables created ad-hoc by Synapse.
+- Postgres only. IDEs will likely be confused by SQLite-specific queries.
+- Will not include migrations created after the latest schema dump.
+- Symlinks might confuse checkouts on Windows systems.
+
+## Instructions
+
+### Jetbrains IDEs with DataGrip plugin
+
+- View -> Tool Windows -> Database
+- `+` Icon -> DDL Data Source
+- Pick a name, e.g. `Synapse schema dump`
+- Under sources, click `+`.
+- Add an entry with Path pointing to this directory, and dialect set to PostgreSQL.
+- OK, and OK.
+- IDE should now be aware of the schema.
+- Try control-clicking on a table name in a bit of SQL e.g. in `_get_forgotten_rooms_for_user_txn`.
\ No newline at end of file
diff --git a/contrib/datagrip/common.sql b/contrib/datagrip/common.sql
new file mode 120000
index 000000000..28c5aa8a1
--- /dev/null
+++ b/contrib/datagrip/common.sql
@@ -0,0 +1 @@
+../../synapse/storage/schema/common/full_schemas/72/full.sql.postgres
\ No newline at end of file
diff --git a/contrib/datagrip/datagrip-aware-of-schema.png b/contrib/datagrip/datagrip-aware-of-schema.png
new file mode 100644
index 0000000000000000000000000000000000000000..653642da9130d192dcec00eff13cc2698f311601
GIT binary patch
literal 13610
zcmY+r2|U!__di~FmljegOIebL48vH*8fA+tDQ0YgF~%-q9m~j;WXm>oLa2~2vM-}-
z@v=qstuSPdvBVgD)93y9ejfk-JRaw9U-!=I+;i@^=RVJUz3zEM8t7><pTB<o#EBEk
z+FB1DpEz*}IC0{n)>#I6OZ@fwr|899Pj#5*eGJOl)4|R4<`V}O)CmoY!%cC?n;Iya
zn-Fn`<V^_~d1<J;1mvdXqbC`@Sr6!3oPW8D-0VDktTCt)XjeBEl&A+1gF?A_*t>ad
zFp!i_oM1R%p!-B!Mpar=R8&@0*4Wq>0)Yq%3kwPg-q(C6A|j&m=#h|+khF@FgoK2E
zfPkoo2p9|&7Z(>15q<RN(Gz2sq!L72QCdz_T3j9sQIt?nmBw1#R}~eJQI!(`$w{e5
znqvGkG#|>+IYm`^d?Ctk5B&!+(hvzF*MLW|U@2wEduCpC+TtV+Nf1OeCkUo%o~|Gc
zdi^X*910dwL&&J<`a42IL?jUR#e9+4qH>QNGn<}24>D1gd5U_5^)`PfCZc8Ill@f6
zBflxhL;tlk%qh@O0wnUjy8XdppD<TQ;`>)Fk7eUM3d9vmG*zX10s{PVYcpzUV@!<<
zLWu8SlJ_OSZz8SSy|Lc*5Etj%uU-%rG{hgRtA#ChcYvpQe*mkvd1y!l`g`AhXjffd
zr=c!mZYkx5!=Y`G44y*#;d*Z~{qCtri-^iRHkU*@<OU<8W6SGgG$3tgh@6qDjGBsu
zagxk~rzDglNLtt4Oe^m-+Q$GYt{YlSEO>_0Z+DRtmwDipR2qS>iYv{isI2h7V5GG$
z5QvnC%^NflQWkA*8(C8Dktq8F3sVwD8p;<|G<-p;g+2>{Ky-EV%oJf^#u6Y;bJgU+
zieij}qPAp>o1~8BLkU=dosLe1Hkf2C_R{5^h?w4eZS{=Tm;hh*hlUO&p?NP+9<mQD
zpZQwHyI_?KAvhN!`+MLz3yFxV;&7LY#1{^k9+D+U$udht8)u(zdxan)4H0Q%oEs$k
zeYLWrgvG;$Z)*#ZA4~Z?5!dp4XZgBT4~Bb$!QYpYGJaVt0g->8Y4W11DB90LP6wVv
z_*_#IUX$VN?P7J`!aCj#om2ECr!+n-E(Gq5&_KS*NRN$H7guq~d;bC%mWhoC4%D--
zf2yr*?}m|*lGcE$Jdc0-Aup!3ED?n;&aZ6SC^pJ4(Moi5(X|VBfz%@=K2tJ8rrAhI
z=;_!>fs74wjYFP%twp)|zt+(+LVMnGdzoVu8LEr>NNaV>Avipcwe3iGB&7&>_c7+l
zi4!+ZXg^SW;yb>Y6=Z*PiL(RE5_ZyhEtBxt&Vx^4BCORuw)K$>YYZk*D;oM+^Dw^M
z@}XGOqX!?gV|+q1;{E}GUq$p-oTts%zcU=Xb@c0Ltm!Ls8CtarJ+9LAGkI3~^u~`c
zwX&e?tSl^g8atgfHD-&nlbrzvcdm9fY*3mgoeF&o3Nz7p5TznPlMic`q)2|#eR}6%
zFPb6$1f$NL0{Xw`+!#BF&kes#5FAdA4|$cbjwj0jLvz-<b1HSqUi%+&I&|c(E6SYf
zxRzx62js?vZakd$=@}P4r!9*ktre|38gjLqZbn*bme%H4V;akL2eue_My`?lVhtbX
zCO>)Sf!==xyf@IYXaE&C2GxPg$PPB|P2V-5yg?=iSkB!3*3T90{Cw{G%3;RjN8E{J
z?C*C(Q5rs}E2j%scs+Sa9~V=vrxr$y1_v@v*&ID}+jd78Ay_65A{(yDS3f!&${F|W
z@EFHhy2in>oAJh9dM|@#N^<ZWaf|LtR3F)~#o_Up=m_wNS-Kb~sGqHB#iRZHL>36=
zUXP~2&3!UPz4{I_Q;?Mz+h9?B5n2C^4}B#OpHg<&*Gy#WhQh;fiG7HR(LkHz%2-?=
z%ITxX2avN7<HHG<N^HcwpQ;15LY&Ko>o#}NW7tixCf}MH2Z!C)vbdR*xL3Buj-K*|
zAC#|n+Y4BDBhu+M^nSWTZp(-h#2<QeLBKg9`oLTVrDis6$4?y}L&GWtz=QZm4W%@-
zCW{<No^jWWKUouv<9)MtuM6uYrTm$6EUQgANJ;8AGb@~>Rj(Y1s%`8&%A=**C-u8+
zACmt8BI=7r<6^w|ZsakJ<^2BPY3Tk_*lFrIxW_5Q7(&bS)nPq8G|(AaoH<9KT^S+m
zt_M^*eKL$)7SgvF1I^Sc&iOWxnsR)5$W+SQenPqVV?!`ecb0E$_f?MTo+Ih<NDYA`
z+3}h>0?%&B?<EE{d?tRk96&sdwI+5?X!jxX2$Q%^$+WW|{n%J<Vw2qD<HEf&h$2vN
za*o$7bUkWK*_g5Z3r1|wRyD(~8FRx8kKJ#;newP83j=D30ZVPfYb*}cvVdp3KQ#;F
zQ%lgLrp$vr$lDRRS3J;ZyD98|u|feQg$7btTIF2J{B))ALC%fpMi;-8{km3Sla7Z;
zF}1ctrxQ|NvB{ItgWTD|(mpBT8dUH`ae%Sn+bQi1RK(<j+x#ivb`&P?CTL;~@;+iW
zaul)Em=h14py~A?b}LGfpV*VXI7G_Eu`>r0E*ggV@MaQ6)|y;N28D~*J-zC>@{+}H
z4P_0O<>M?lJ=-GUcQO61)X~~9Lf7I>;uijUMh9hU{_vY^C*AoH%bi>2oU2U^2fWL6
z@2r1|4jB9DgZ1yh(dUxDWwVcYs`auXb>_`l0ZpBn+=u-A*XedE%=@3T<DNL+p6|dH
zJ(J}S&D6R#wPnGgD`SBKomlIcjzRY^x!$_MiVGp`ysmik$V^HNVv(G&qXT_`ZIIje
znb{#8R7Byhgub{IS21O|2~0Fj<kU1!<Vtqab>0Up$(8kKZf6o#qEwdbQxyezkgtyt
z4xWl?b%V{;l6nzhK^sL5rue}7Y`t2|X|6s(+eHJL(6B`NwCzsNxY=0i(LWe!J*&NC
z0ZYl0!l}y-8_x70?|EEb2)zQZUIX2~^cQIUkNyRmORzHlz4<1AN%bUpY${@HW-#sJ
zz>bKDKQyOP=Q`t_y29bNw6!=ZKU`=hv7W{M2&Oku#$}HeMSS`d*Xtbg97O^d@$UZc
zJ>2wdsKp<|2;_)kaU<11@|B3OCH_L3_2V)GN!pElN1sw7pdz~yX*5%Ljd3sNy8^7x
z9lpajNy;9YgKS!x_*|3a7R?sU$03)mR0Vv03pUy>3jCAIt!%T*Ubu6xV@B~dJpR-}
z9PE1(IJ+&Ua1o5pa^E}%T}Sje;v-q=o2Y9MEHl}iML6Wx-W&O0+CkCaj!6A}c$**X
zpcGsGE@<4Y_RChymea>ynn{6U0rl)aZ;dZab*~I94e0S!8AB=Gmi&3ST?%l>F5+6g
z;7HBYqowaS)GWUs&f2t0pHw*{ww$wTlXlR$<i5Lm8+75n^6l{NS?b?7VAm4KbJokU
zg`g;$rBgl2@h30|Qd@ml3IEA*2Io{^@m-~7t`KCDBd63my&i#*VN2Z^-2JDAl-)zO
zVbyRzwX@?VwXGXh@S?n*0@nlfi-zdZi{euMgj+1{ELJb9<sCWjR42!!@AL0OCeDS)
z{ZS0_P!4HF4e>rZ_LHdq?GqSB$kjgIKYVuskvZ^gV|6OK<ec4~Cf)}xBN3^?D<f__
z69W51JvF7+Q<Nslt;oUsD^B%GQB`;z`dAp~wp`yw-9mVfD7nys<i*jCt+moV+JiGC
z@u=nl>B=t@du{Wi8EY^Bfjw%vVLEA=1VT>*%{`|FF#Kma@tysTo6-F#81<LvF%9}A
zo~57BYOHKly8PKQ1^*9}om)3k_(4Tf<<%szp*~7=$umBV_P8<Mr`O&09r2kW(*h=x
z>^T#WrF=ivSh^WhymnzL&z_L!PB-&JS=ktm=OYy`cBY~+t8_S=?r&3IGPSVN<GZ3B
zpXJ`3_e95?*(<YTGElZ0O7t)nRCXpc*=0N(?9-Xqb7t<!MG2e&@GS6g@IJM0uE&D+
z34VN9l&9!Y*@}Pt+FZ<^FkkGx@1&#&?7y6-1DyT-_`22D7N7qjhwDa$UP+wuHJHro
z+s(R0-eGymQL!YI6L^m>)D~UTM!Yw=Rzv`(Vu4-*^kyV?`c0dHL@eX>ycJ;U=j)T=
z+@YGV!oNB#z+(XNtbb}?G?1jOES6GVzDk>R?Q<X7N3(wM)bRd0okj=gA{SOnQF!v6
zk^;O?+1i10btA)WACsI7reK5s%Z)WQ3W`4m-{s~etaIBed@$9%*O6`~g~~GMPeD8k
zKeGoSl$1TrAsjX>%x`2Ytk^-5P<?yeIuwpi0m>wE_~>bTg#tp)EOdz(k=y@LocppO
ze1_`4_qzHkELKHtMQ`aDy2bVl={C>)rDoA_cc{{5OJ=-BIO;4Y^uN@ne*T}@Us!}c
zeZv3Oc;AW>^dE{!N4ZQCKFT=)jq$W0QUZ5!ZaG9&oH^}c9t*zVFUbkyo=UVPw!%Z%
zGRezxh|F3<J06@<dp!Js@Bs%F2F}!%4zQw`P3`e>6VvbRHa%q*t7-Jm+|g(!F4*oj
z#$=>=kLx7_2N}n3oz>CFl(x@>4>wWsQ!g(2I~_OUd^l4I&@x_ghA`9F64eXYNUozH
z1@w9j5kCBQo+Gv6`6>SmcFV_De{}Ik4Sm&Eah?|f7^(}SURAF-S5n9`wAwW@puvcV
zz;xQnoX3AZN;0VI+JeLxR~l+?BPfQ8p-Ud0GD_lrzWDIdUqwiPf^yeG4wW2xvGiWQ
zdK<r~c;>%rgTRE(owSCJcu*R+L;l_J{l{8#V8+S3nx>w7a{_=jW45WJuR3)tS=C45
zXM~!!CgC`jkuD{Z8`n@9=L?Z$$qXtoU*}fJO06f5T?NSp2d_*9+}8R#qa+fShrsId
zYtli7C~)ctfYWNfs3qo#6E3l6bt=2#?I<W?<zTmG<kmyW>1ih{wt_Py|4_&W-xoN3
z_I5Ky5H?+DVJJB6iI4HhKp_JukJM{D4UU~Rq=Q?3EmNpqgzZ(HaT2-hO5-qQb@ixX
z1bi&sH|;jl6y$~?>8?#<aHI>sx8tUG#{cxD{A^W=^!4KnAim&L_f{LsbLS52{vDMi
zrBBqraBxSj%tu_%s<SQ!zHbS{lBM`fp>iUIJfXmU?-U3<KfN&|;-_4)I#q!ocV(Oc
z7V3J5d|x1QnALz1yS>Zm`<*)2fKGACzz4Qz0+;211Md+oen0<f^l+ADjdifKsmKN+
z*62TWn9N9Gc?n0fAR6m%7cU_eO712Eq+9ze!9cWtF);AH9W@VFLGOdynyzPzN>75j
zk=u9ho;)p~C?yt|E%Ft+CSBMtg*<KlU<TLmHN}l-FA9X@VF00Dk|S@};^2t&_XQNd
z5(H2<^zcsJ8%V$Mp6rg)Kl_g~Pe?mSkeKj`oBpJCI}@M;xVM(=>K0pJRE&mpq}r|c
zcM|Tb9t$fzv2*?^1g`lFLjJtlmw{>K?L<DsxRZDfGPG9fs=;&eq!s5W;4fWDLV`lQ
z{>9k;h4QV6_?C#6jT4_SyTsLP>*KYoI4VQn?k(~Z%}P!Lo~@FzI}*pLYrwnvy*;jt
zG2KUCIa1ikxMb9$(zrZ(q8>6?<}4T|+@mV(6tv<&?H>>N$v3oWi=7S_2OlEUXE&io
zhg%mU8Dp3irxit4tL#Dz`_h@?BQ09Y;DP0lHzxSIl6?@h+w1ignu)C8J~wsX*#-J$
zk-dtvg>NaBJvz63XvTDz=(`cgy+v^!HC`O;#e6KTD!c7SEae&(vZ-`pK&KB?5M>N!
zf`2NKyFN3Q;T$Tsro4~;9FHmNrM@FdK1XbhmyiW}?{cQvLTE)Eq48RF;jnY>RXt<Q
zhuwv>nnqmxdh6|vnm5-zg<RL1lMK0<Y4ILjM}W`s6`E#>HNAeY;$&e_=lE(zRVD*f
zX|Fpnid&rYUq?>2+_!r{Ilb78o!r1pQ|o7F>uFHRP#`n=XY|wKQ6Gr@-Mx)Jdnw>?
zls<T<Ozrqut%}%B+(E8}!JR6d17+PHzhA`<toD!K-^&o?zV3XccM_kG8ZXAMQ-uN5
z<}L9N2Q$};nPR3=;_RcM!2>V(Uu7^r4UB&sGnhN6aLN}9I<)QJ4k9X8#9m?L6d3E9
z=Uida6?~7ZIGul)dw+0Bd-QoLzjff4tuWcrUJ6%rGmVdYR1ldG#|yt3!@o0dfs8e|
zMsq(Gqw&s3Tj5&~X)=}FXDN(rhd(gd@>oBZyp>lIW-Gc>1Y&Hdn4zn~4>mi9k}{v>
z)Viv}|6cOPTk7w7*4(=-Tfw=x!LmRmT!R{%dW$Eyf+LNbDUe}ma~0GgP=1{4J{Q8O
z5!7U^`OwO%&E40K5WNyhvoGped#Ox&^<_~#&#)%O`T;&JYR-U!KOn!c!uqa!(m_l#
zLoW#Y`FYD6A@xQikULc8s%JMVtQ`k03-YD)fFz`zmS>)%cHYN9%S0j#i6ujo&PEj_
zBao(whO&<zT?<0UgR|ah3g0@M1D^VvwPBu&gUh@GU6@NNR#QPi&%$^X<kxEWPHg;y
znBPLI!txxVrI>O#=*VPiCiVIk7q|nhC2_{mLZA(RhFCnrZ?-EFGcEHz=-}YacMihR
z*skhldM~AD|J&EJyh{tqeF=5R7iWUWsd);&znUD#*Z=BmhTf}E`#ts}lj3RWz+FXN
zxPf%ohc05HL1`!*v@Cxgvc`^c)|P7{VSpPbVAoq{lg=_A=+2^YEe1xmfxH2LEFX!w
z3clfG>d-K%55IT{A5p89p*qN(C7;Xn{XFX2E6~F15|gj&_FUXl^rwpxHqV1h6-kO_
zi|MG<)PjA~2uPs^S!2!FXCig|8t4M$>RXF}J&z=P^H>QPF&GnZLw}$l;&0C{e8&oW
zEs9*uYKrM{oj}$68CH{#hP8?^fAAVTs9(|dJB&P8YHV>Jq&ducYIDno%MWopXKLM)
zCT8?D$67&aX<}bl!&gqy@5;q#WwTt6zApah3SsB_m?1ti6?HcW)djiM1wo~Cnc1eD
zK8I8SV0&?-*OOb$uOAX7HXapHAuox#aQY?}|DHy_`BY^Nz50e;6$X;XZ_sJfshRz9
z+t0G}hux#d4h-%LsV=5Z-)8Q{n}PNU|Ngx>8C#;S#r8?#4^+{{dX+e~X*e>4lHV6%
zM$4Ko@IS0Ft@_cYc&I|G_L>;$FDF%lP@-c4iL0!PYg;G+==~yiwGqkk5PxC(PlC~I
ziE=9>0^a*W^#@Gmw{@{B>Y*2@ZCt<V`VF?prRvdTk4qOM0#99FF8Nr@7q*%+_-%Y<
zBh)Y6x)=fSNOoEeEBNTkngrsVvf|~6woKrsl9_TDhD0SQA6v+;3+c=JS$UFVa@Xi?
zkDUH4rzKCd_geMWjH0~5$LnzM4c{Gd)b9MArvCB91hM~p9ItZ)Kbyhn<h&<ri-cha
z2aBIk`cF_Fwb|Ab&AvMk@SOVvO#wG`G)2uXnyH`>I$A16gU)xT4=8AU=dap$sJh?i
zI#&HsU^G5t^&QA{UEF2EeFL8n4We$|Qs#i$IXh^>`|7f5ZJ|N;3nhUZ%{}}Sac_Nk
zI?*)Xg+RgnGf(I6TBzwy@2WQ*;Z&2(#k<#nMjFjA+6x7hB6@Cc`59SS#M6VHOwc9G
z7cGrvUNwC*-fj;LcZ`>IT2=EWw;>gs-OA3;Y)RAGeIBS|v9Y80<vlx6Ex$mU?J0{{
zcSA0)zvCvx(s6?^h$%ZKsWrB1HoV5uNBKFKMdjRM{IQ9Otb@KDS~O}zl^<_7)-iif
zXl(pJWFw*(B#(9;Q_D`$D=7!r)%rb+?ePDET{@H&uoD23mltb9t0B;f2T&Pr=o9tG
z4{nvCj!>~Cu*x~u58`#Xi?Ewh$@Yp{8K{}-hWz8%47C1s%))}M9^1$P_dmKSajU%2
zG|SZu<_Rx32~f1pcb#pmwTqh&d6r*t#%w0OioEY1@s?s)--0T#7O$W0%?d#`zRk{7
z8U^odZ$EE&E`M>*ILhm(bH`6)Xe_627JMPldTcoXHF5y_Lz^$zf?PnIXNz@xWv4*B
zbOCR~m<8*<G+eVB#a+0-KxllQpO36<4%5iL*As)QwC95H)Z+SfTkI|&Uyi(dLFjgn
z69C?2&3)g*?A*1DFu|98=0Xy5&QMLyO!%5eXq@zW<NZn^Y);&qwE8U0_yBbXnpah^
z8!JWt9i-~dr!Gt;s8-`R+gm((UEh+fRoDL*DmcPk$POu)xA=n+Q8|CmIM(MLlnBEg
zLbgWas$SC|c4;2J0lDn5-@eKr>12>uae<}Pm^Dx6+J>_gHUi6p;c95Fpk1Pmo)WwH
zW+)4>(2c$zskCfT5k1?JjWdqEBEU8(_S@lf#_G%PQyI-6AB$DD@`quz`lm<y-Lo2H
zz4>Nt)N{>_pGOZ*{?e2;r;?#dG0h8FJO1Uo7D}mP%Tbor#so{>1<B79ygK91;{D&j
zxNKxN>u%_(+*~VBI&48=^l%mZ;(%!wi{n)Cc#iZ=Mc`vY2Ngrp$&P&b-o+!>6P;jw
zHpOL=a&wDl3`d5k^cQH@MD@+fJAG2=LcWn3gAcw)f9UM$t@4_XPIwa3f3lBqy@#uR
zBH&*BKEwBFv||kf&d9T%LCR0v;KOh9jMzW^d=$Szf2Vfl5OOzc|1un0*WT}M-E?Y-
z&&KHS&Eeqs`(37Th3^9WrB*~OFD~q(@R})tcPx{SaGnAwzv37y?C6?(diEJfZWvm|
zv29yOL8)TtIh91(^mYrUq))2uFQ}z_Wxu>jdb>~XXJ1A6J+)8_)#2k#{mkbipTg%2
zl5NR2^+a|WZnW&DZRv;N&`RTVWquCer?p~j%va5cE>o!6owUW|UcXyGFI?7A5P_Fx
zRGePBu3WT_t<p`GaY)$6FM0`aX*kUTdDQdGaans(jTE<O+!{}=To+L16V&7chvDv?
zf1mL}INy_#)o@T3&clvZ!lLlWff5&)zn;qs%UY^AYYlh2#JsiV_9WUcxb$rrk?S5v
z<jaNlA1C5-UP6T5JW!>awJrq3F$1h!Cm%erB7$Z?EsE`+!5OCCi;I=_m|mnw@(UVm
z-oItnE5-5q^M#|Cj<e7tenI_$*$UlQ4vFq0vV;vZgt!U@E+>)isY728S6Ki<{ugOP
zMPrbrA)bbW-hB(Z7<@Q*3K$hN!qe<Htu9`}GbkT_asx%VLKFVpAAi1s>%7E^`$rD#
z1CbTaUn`C(15*?2o1XMYbD7A+a|)y|?_y)>ydy*v?hIWk{uLJu2IN{_z6DV`F!NfW
z(I=4bEd97nxBJ2wZxu%$7~l#LHH1}+CQtE|v36bv+!HnpGDwx_<vW8KL*O+zZ2#c^
zmKBS-6I_$dKkf<@J>LZml(w$I&!^$xV&L)A*WJH%eBACVS2CgKCKCslEqj>S7-WJL
zKI1ce&b0)P;5u7rU7$q8Z*$kde*>Z($jPjVQ#4Cxr7o4xtdpqP`%R=mcPpNUGYHQw
z3>*=;I88ytbEs{9_AyRVVU{Wd66gR4)!*)z>mvT5cYv$lQs+5RbK}V`zCmu#OWd+j
z=m6yUJ{@85H>ulm(cN3|6O3Ntk!_eq2hLp>s~)fjMW45_@;vx9OH~Gz$()o>WCeBI
z7;F=%JlTfni3+S$ARXoBk;%4Qn6Jdu%%JhWX``nPzdsW+c+k1!pZN5h(h-nIs&b?0
zxsLnOpeVTAeq;X@@cB59sT#Bq*uZ(xWakeD3aduoI@~XsDnDHE3;q;EdU)g+2uVVe
zICb^(G%7bP3^iqiX9TxW*AzA{egFMV411ST=Baev^1=3dBd~>n;6I`Yyfh6Vy|bW1
z7smn82VlyNjP|Ove{PVb2gd<@xZJL-HgMRFyoFU+*0*25q>{SbdDWqvMd>9oePvbG
z1M+l-EKfsMGTb<><{umlH6(ksyRbvn?$Ak+2T|#j#Q==BcKaq%e^s-5_(;oXq{Kzu
zt)Km)liqV-Yi6`TOP()rRNGAFrr`Zaj&ZRz!EUv)Pi#)g5<TreoPt8*hIw{2ui>{|
zy5qI*yjNHmPBU>ZZkq7DG<3WdRhneQZXt8p#=A{GB0iN?yw_2`F(vC}m->10&Z0)G
z#|Cs`S^}q-?K4Abnozy*l>v4JKi5pOUO0nt5Fn|xK&&(SWitpN=hno&J`}fNUntyZ
zURaM5U|p(UHBLSLU7`wL97FhSn9ZN5JOMOg6z4EaGGcwBjqiLjA6DtSS%eS!m3)`t
zXcyE><uBHo8Y%C-uJhL2TaNy4j7OsP^W$Wko5uqKo1#Mt@ThS2uJoVpV^O7y4<U|r
zLt6T@mWoW#ib-Y=1K@50x9osHrN>kWb|4@2w~i)!{XKlhw7Sz+PyOwLtD*8;*Lqci
zfcMt07d1?vjwn>p!0Zkg+sFFw3fo>l=tf&RTb9TRqwM-V(iqYJrePc~fiI|YP%?Hb
z)Or;R*=SP|Awkt>>{)VGWC>2&&@-(lx;wf<bE&BODan`|suyE_gfRNc*mbHcBaZ7=
z?Y6lE_j>jomN(%?_v9xY<nEIN`)V)_|7^FQul{)iHzhY!zbu%p3rP>pe1_qmbd7IQ
zn(y5WX|LjonH_64dT(Cgfyusf4nqqxV-Df51rJPBS8|2hBdYZ;!f=i1<HU{~*YapL
z%5(DGbMmc#>96h;_j;?R1<x`L7`@E4C5#Ke;yPS(Ak+HnSyPRHe0hgvMOT8>Ko#MR
zR{Pv&sFLKlYvwvG>O)wyz{*;ecLuCy%zy(kvzb#shy?Ku{pL6pU<Cs{ieO8T<QzPk
znB)@T_2kms;>57ev^4*er}!9!A}3RxY{NY;u(LG2|DOHi4)cwNX45m><M_2WN}z=u
zYNi4*jiBWGDU`ef#6^@()g*qXO>4>+8_FAuyn<IL3*7v@eAsd0Im#-HU>>(~JCz*z
zlAH<Lhpuq}7dY2>MYd8eW(*an^wfBK+&5~AtDLHiWu={?gON{~p_vWMS#d5)ECRI}
z61L{6_}Tt2ofz;PHI^+PQGXEbCGXw^fog7i>w8Y#>4rc%j99;7M@Vc>@Ic8BO_&R)
zzx{KYUviLIe!aYo!gcVuTzCAlcOJ|9N8}u*kFlc9xhMB@c7#K^x1eJz6VRj`dVgE_
zGl7AWjpnTbyW0NX^RYl|u5K_I{&4nF@e?P}bKfc^M+XLoQON!D0PZ#P+_?xu)gOb8
zxsIg?DExcUv$*l(rxzRbxajUpp3F1ol(;KV_pr&l@S8OtA?b&w>^Y4;G{;lNeCL}s
z*n01L437r7KL#NF<$#+T!<5D4?!H^$$c5X<I~zwJ#{yRpS_V9Vk!fV$6N0mBv07}8
z`!M#@ZZ>lgI3w|P8=^{N#-!{DBR&|#4crzNSpn#897~a5)@daso@r40lKklvlXfch
z=!$ZO+@b%WWnFt&k{8QZ@58fMas(z&1a?)>VMR#CBbg=n_yMld<?4Xi@~v+>3%}mJ
zTt3)314Gae;S*Tx((EoSdMLZJmBuMUXCpcJN6Dx~)|~)0WPn>q?=H~W7=UKTe~Mya
zNrTaev#04gS#i=W@&7M!PXH3^`KA2l=>N=c0sduL>g<WjlAX%6jYHaBT=RflBhRrC
zZV&gI)kL2$&Oz_6nWArpt+G23Zg{WZ;RGgP-1We?IpA=6J#BJ8S&}Z2kv%MGhgfX?
z1XXKL4A`xW&Jpv9V!sk6rl7_dEnZ*cot(OaP*5ZL8{P9SlvJK+_&15r*^Dq3>+liX
z!$>aRdox|8C_B)EOzNSu)z&t0hCHiL^eG+;Kkx$>559}$w0{~H*l?Fq$2F2K!`i6|
z(u_=9@=hz5N*%kj0C2%~>3I0>hC5uI)6>!Oh%C*g`(9R9rHV;cOT8Hx7+V6W1i08j
zlvoP}0;wXMfm>lK&o;fj3*PRe=K@^D{mlg!LAM8*!(7`z{PlAI<z0{|r=TR3?e?^D
z%3=Ur&@U@aB<BhXXD35#L2YfI-Cif-Mbz~DU-dsWE(fE`N+~fv;w{9TUO9J)+5aYW
z{n>Uj8<M<2C=qojMMG8qQ^|K1Z&XCD7^;3lRaC{2Hx|XM@UzH}XZD(`kwadM0X#ae
zc5mg5ez!8Wqs8xNu5n$o3wJ4)=`*ojvfknCpnyB&e56++79R-Bkm2?N!#?^6;ESJ=
z38>!-_T(dP&oYx=&%Pt^U<Iy=DIe&Q@Pn)6vGVIE%z2&7;o#s>2ry0#e+^n0bX+H2
zKUGB6VYhl@1<sbB9c%rakePWnxEDc|>tO(`M0xF#Xn?t8oq|>hw-};5M_*Zscy%$X
zo^+v;sN7MrccrKJQd;l0{41jc!anf@%5#JLZpR<%Cimr#QKgw*wN76^=n?OiRp*LI
z*>=qRzRj+sUs*z{sIIv;&rOD;zcWi(Sq)iYb88!V?(_5ItJ@<x+RgR8pId!U?4VD+
zbsA5u@LVJHV9%K~T;rG15l7AVMi=s4?ub;6b~re5g~z{(5w!Bq&mSJNqAuJje`W~9
zSc-g;$w4NKoZgGRLVl-^(C?N~F!>(Mxd>e3?=9lqd&NJ`(H?ZHN*Rz>M=RZZQWys7
zcr66@ny1q@U-u`eF^rFC=$tsl^@AK4u}zTv<j#!PD2l+0j{M`{4X_5cHi<X7k5c5U
za-_%R<?Bz3WG`W<lB83K<EY~F!&8^P`h4qS{%|hFbbak$7p(d2blljz_AE_H$<|x#
zQc87o=Y1G}y2l7&X=$ZFe;F94Tm7+v!cM6XuBG62J6AaFbj#yKznYuYNu!6nnC9YS
z|0t-aqhnyMPRx}}Tn({&711N}pP=pCU(QmoA5^cTWX<~G@It%Y3C~1_Z7*wG0wcY{
z_I}TvEf$9E(W-uSLYZ|xIdx|U^!_3K5zEwmBcgEi3TLHFT+Td)%)B}r-}q8J+TG9Q
z1sDJAN9DMn&UYV+PRHCFVxG?nx!Do;1Q6Ah=pm$6TwSO|*yK^);>lEx`8y$~Rk%1H
z?T7ysG_5o(;2uD=VJvpPv*4x}gb53c^!kZ3U)u#WMcKCb{xfttmb5*llykZwy7Ap#
zS<ZB{Y0Y1$q)~mz=uv#28Kd5OcqE`y@vU6ZPG9|`J0@(Xm=HY`QYT(8w;$DTu1JCO
zdy2&_EjCj0D8@tLk$!jcmfyJ9cO~}`&)r%Vxgk;kn?R>tm?nN{_sMO%yP&+AzO=BH
zExx{3F^Ia`=V6N9Os){@y^Y)Y6UE+yHpRbZuk8M_-q!|EYP=4!)Thq7GX9ot$J3qg
z)WPvY#cgGmO+}^$Ht>M!zY|c*wU*(ihBrU3z7JZqVj?_+)ueMB8mC|JtAb?x#CkgM
zqNkaUV1@LYB1U@jYbDy>3dcRE8~(F*)RLn$MoBlx&CgU;78z>ZqPAcZbv#biW;lsA
za^vUZl?V+@aE+7P$E-7t&0|-#|HeAb-OiK$2j72D{Pe%D4Sm514y=M)q=V(4TJN3C
zOP-6?X5>JRcR?QhY0LVD-D*MpXV?mo+K!6i9_Mh=<RaC^d9Tp%9u75;|D)&}PSzE6
zaC>$4VH@tg$!!e%a5Kv3CBbmHBA=I$F2UjR;{H#Mg3!JeV|e@Z209KusrWwAw043W
ze~r7Up)7UC>k;{p31g=A2ZU|l?zrc4Dw*MaO3*n_nLnO0{}4N>2qeQ=KM*H1!k#aM
z@33PV&kb2i21*sLq(cTij!37~KD$oWs~ZE7-|%lsCC}5+j!)o6pa4x+ct~stJ9-{?
zcl_s(7(z^V%sl4Xs+;~#BC4>(PF~TNKYB2B6BP07<T5<gbg^`SBV)L0YcOa(n9OoZ
zOo7lOMjxKEm;#rgJB$(LdAXy*Q`yXTyQG+y$#D(%w$k8Xsf6I^L#)15$)5GuI0P5)
zza>?X22BU})@cK`y)AWwYatOOFypFyKSJZ+L6*=kzshRVh2mhc+yX~3qeUHsFj&iV
zzn=(t%?>+TU#8_`4XvxuRn{z`#C9|yLjK$Z6$NP4Ow!0&i~ZLc&b|BfYUXVP1K?)E
z-C`=V)Z0>6c_{?-zVw%sr5yfvW_mwDBR<=`!JlBOH8GXJ^8`pB*1s3{aaXmrQv1SZ
zE!p{3!wJ0XdBLO8=VsME`8-?1PK>|4boLzIPpx#%D?6;U&#a!#c5;^bTpu)0novo|
zR3Ql6We$m`BnzEq44g^Z3tZe=cTSjc#>{|cGw9WvG@rdhbPm=oo4G7*p)?%j6u7(Z
zGu=Otrk_>)1!GYuY`O01TSxi1LD^Cc=>MR|(>Ff7#5kz3RyHKJ?7JpydgJ<A{WX<-
zm@&<%&tln)zjY?|Nj7ur_Z_BJw2+}YN}<t1Z@SP^V{fv(N_jzOMN&ii!5(3U$_!#>
z8@`jIqiTjP#WL{&!w0J>0?LCEIVM@v2_14Cz*Vr%Ax&%gm&*E4={|5A;A<Vo<K~Z!
z^(-Mtco;4w-w_5JNxfjk$m4>(^^IS(I)|dc{y!#_m*e+wpO^C`7ln!!dEJ#Cfhur>
zOfnVcdc@<M4jeVrDpSo44psvrMp8TKODV>npp2<Kh4^2PHV!Gv8h6Sq8c%e?H*jb}
z0m!WulFxGGDn1#P!Eqny>0?We-qD-yXwDe?5RTBGwr;I3jNTY+xTg?Jp2{$pAY{s_
zaRlme^30b?0&(21*xRg;*{WNEoYF)BAwZ>^{*3OZ0&A~Orvz5Lbgvb{!43WU(lk{S
z@S43U)+OT=B|D)gLc<9>#d#LR0I0Hp5lm-9H!;(+R;YlbvcJ#U*n16k++$b2?K+ht
z{JNjgK_HfF6fv>f6{u`D4Ooqd0xA%)gf8fK;Z*5)+SHt)5l<+DQ+OUb$A9~a!Qrxc
zIih3v>vKG!3OUwK%2-gf?@g<uAG;UXjHz-}+{h|@&}G3|c~`1l8Dj7F8FN=#1Y!B~
z&&2vtg)XA}Fktso?@HToc7`gQ1>9tro{c*P3MD7MF)=ek(8Y7&IO)gJt@H~JxNd!l
z0RV!@0fbCe^6;>~2A+~~8Tfh3MxzGtbq<8khdBuovsz6zBSdo`;Y6-P(Aq2H*aiHe
z$J!kO%QGO^DezH?a*nOTX9xxZ#GpFo;%|3}xhU0nqXZ$%nDo7eY-~0+{~mhwgZ~=s
z@hpmv=@iLC$m9X`t)OV|Q8T6v<o_9?N6ciw4<ezwblvC&>XOWoFue%!>3sTSCnV`b
zGp3i=o!x1$ZSmFcP<kMj3GWeuQtCb-&)II5Bnd!h8{t<&EnnZf(0wMI?+zJ!rppQ%
z3K9_ywKKWBVsHkiy&!3?_h`%J4ZXh$lKGrk3Tn=R*dw7(GE;?fD<?62iF^w9))xhl
zpEh=LldQTZ9%}D%6?}=D{AKHN?9!t&uH)f)nF4c++i~UVkB=@&&T9-1t)WK`$MW7d
zRb9#E5UI*DerhN&h*p^K4qy$x!$V}~VqU4t6RUme8TCcZ?IeXYFVg}~R<{|mx#J2r
zf1eq)>BCG0yDz>D%XhSbCrp^uUG<!EypWy}JHJZg$|=4nfAzZ9)T~JA2j*~{V{Bbe
ziN+0s0ry6VX?ZJkziJ@CedXbv2*_O_57A^^OsFcye)Pw;e!fxd)8}olaD5GKH&?49
z5jU22!!CpB<|y%U(!j7>_U6*+dFB)Z<kK5$r&g~RV0hdwqK=EZU1?#VgG=~QP?;NA
z!29y3@csBKYhBO5zW3nwO;UosJ1<(_&b!rj0hIqV;#EtcN-l{}`UcP`HWo~N2eD+^
z9aVJIclCu`Z4A(gQd2zQ>%x3)9YtKNulo_2+F$(~yJTRQ?h9`Uq<rzed*Yzad3kbv
zq7_BNHI>RWRtE{f7Q8=gK7Dn$)_UXmvRi}>xZGV22j>RE7AUDoDAx(@K#qJu$+m0_
zno|)SH&;@z`qO>g7m8}>!k~kKjt@9C7AKY&Ye0uWaVc%SuJb?agZ7+L+KD<O(1L{j
zvbhp58(YEjPBQlQzyY~%v0_3+g21)26skAuc2*-2adF}gD)3N;<d2-Xx;ZCIBg$JI
z*T3yX?&>xOQh?4`ln~Gb0VFM?EP2wGUR65}l)pXrS@_;5*DGH^U1err6Z2`=w@2eV
zw1U4KaJ0;G*y_uD_S8i;Dy5_BXjWZir+6_goC$jgrC8mu=<d&kcW5x5+jH22RCiF=
zWw1>4;T2r#sl~938#7IaSIRRt`YQqyZ$<{3rVlGmNsGe7XRWzfv~MF%a-o{b_NcrJ
zxIZ|*^GaX;;lPJhQ4bvSro-;_F-5z+?yYCgo(#|RfmMvZ88FlUam#hP2oS{{3~1^o
z^G0am176_ZGT~XyNd%z^=jm<JEN$>&JhS3vp>SDgtZY{JlJE4NVAop~chfVQu4@$U
z&QCK)@e7&$%4o$bT&2DH`t|MS>MGtx($l+`n8QRDut92L+_2UQwo?FS3b<p9jyZo*
zVE(^JWB(>J{1Z%mdpEVgGGjh6A{ScuJJjLu`PS9;t-&wD66VS@O<q{9JHBxfU#ZR6
zf8`$U2M$u*JCJ`6K_n>#d%;}O$gR#R+6wy(W|{SDf4pN4Uu*Sde(OO8`*}sXGrPCX
zudr{)Y{)q)D4;~eau&D#ICDR0ywdl5=A;iO%1gFN@iYMK#e6vBnj9SR0*o`?I<ydB
zm}X|Ex4GaiBq@_K$VxdBi|ZS9r^4Oj=^P3}-=34((Ad8BG=_7t&_l1tuM5vV$v@WG
zw!7!IFSMkvaE0bIq|eHm!pUuw^L9wnt)P#aael(=yx>=oPG<1WZ!Ys4IUOZT&Kw4Z
z$w5JHo8;eQF=`Hc6H{>`gG3jTi61lj<~sDTitSsA<-YPDrxjyMK|<y=eCM5Yo~`wD
z<!ir<a7Y#1ZNne|m>tcl=u~z}u~&+OBezspoNM~^;xW^MU@6aEw1x7$j2y_B`GjvS
zZjG)jTOWe80t0UkXvUiDW_3BP$%TmCssB=cWQ-c~wS2$awXf)YnACxkc+uws5KXV#
zbE^H0&lnt5!g-0pdRc+6k*B;Voypa=ms-nHUrPK#kKIo?sw(#g6s1U|#7Xa>=n);+
z>HUKd(qa0rjtA+`uZ6UFh}pcsa3?AXIa_o1A9-rgs5xxG2N8D4=%x{`zBzS_iW=(#
zQ@k|Y`ksDoZ^qzZg>Ks9pAT-icV(kGUvQ2|KFIx<`Rq))*~irYZxrY)1=y8%VM2>8
z0#W6F7kvTrF+5iSgqOS86w9FPT6Gum4n1v{rs{5#WrrbpAwYHX2+m6w#<od}4}L~5
zNf9KC&wAVd-#EmuR!P;1(TJ5FC*|}X_7bt&dblyZ_Wsg(=1mM9F6i!5UPOEMnBgF)
z6=(|FNOF-Yn`@0l(GSPm?LI9pbmk7`{9Xx4Gw+MY+~P<Q;EhZ{JefuL<4e73j|JS_
z;N9pJ;(d!+hinGm;>sJZnN4&X4<OH_#^f^f6=|D;V(GafROaXZ_Vp@dM15J7F@dO9
zp>5JqWO;lgM`n=ycu0(WALYT*l8q|17)EGQM8m5~7RgUcT+<huiG%WAo)$dlJLO8@
zfFP<TFW@V(5xa$>RYHo6H4$q@nO93AjnP~FBu8kq;)egIUHyB#$KDlZ%BXYiS6(kw
zb7^Nw4gW++=q&bf!01L=5-HtqN<d-N8=w>zV|K^a!2jfR()R)Me7+F8EO6<6iJt%W
z;lke*dNiife;4-OOu)Yh$A2@6|Chb=FPqg^_z_DzRL1fj1z`jFXG|xw)%6~HRI>{C
Ef05{9b^rhX

literal 0
HcmV?d00001

diff --git a/contrib/datagrip/main.sql b/contrib/datagrip/main.sql
new file mode 120000
index 000000000..eec0a2fb6
--- /dev/null
+++ b/contrib/datagrip/main.sql
@@ -0,0 +1 @@
+../../synapse/storage/schema/main/full_schemas/72/full.sql.postgres
\ No newline at end of file
diff --git a/contrib/datagrip/schema_version.sql b/contrib/datagrip/schema_version.sql
new file mode 120000
index 000000000..e1b0985d7
--- /dev/null
+++ b/contrib/datagrip/schema_version.sql
@@ -0,0 +1 @@
+../../synapse/storage/schema/common/schema_version.sql
\ No newline at end of file
diff --git a/contrib/datagrip/state.sql b/contrib/datagrip/state.sql
new file mode 120000
index 000000000..4de4fbbdf
--- /dev/null
+++ b/contrib/datagrip/state.sql
@@ -0,0 +1 @@
+../../synapse/storage/schema/state/full_schemas/72/full.sql.postgres
\ No newline at end of file
diff --git a/scripts-dev/make_full_schema.sh b/scripts-dev/make_full_schema.sh
index e2bc1640b..473f54772 100755
--- a/scripts-dev/make_full_schema.sh
+++ b/scripts-dev/make_full_schema.sh
@@ -19,7 +19,8 @@ usage() {
   echo "-c"
   echo "  CI mode. Prints every command that the script runs."
   echo "-o <path>"
-  echo "  Directory to output full schema files to."
+  echo "  Directory to output full schema files to. You probably want to use"
+  echo "  '-o synapse/storage/schema'"
   echo "-n <schema number>"
   echo "  Schema number for the new snapshot. Used to set the location of files within "
   echo "  the output directory, mimicking that of synapse/storage/schemas."
@@ -27,6 +28,11 @@ usage() {
   echo "-h"
   echo "  Display this help text."
   echo ""
+  echo ""
+  echo "You probably want to invoke this with something like"
+  echo "  docker run --rm -e POSTGRES_PASSWORD=postgres -e POSTGRES_USER=postgres -e POSTGRES_DB=synapse -p 5432:5432 postgres:11-alpine"
+  echo "  echo postgres | scripts-dev/make_full_schema.sh -p postgres -n MY_SCHEMA_NUMBER -o synapse/storage/schema"
+  echo ""
   echo "  NB: make sure to run this against the *oldest* supported version of postgres,"
   echo "  or else pg_dump might output non-backwards-compatible syntax."
 }
@@ -189,7 +195,7 @@ python -m synapse.app.homeserver --generate-keys -c "$SQLITE_CONFIG"
 
 # Make sure the SQLite3 database is using the latest schema and has no pending background update.
 echo "Running db background jobs..."
-synapse/_scripts/update_synapse_database.py --database-config "$SQLITE_CONFIG" --run-background-updates
+poetry run python synapse/_scripts/update_synapse_database.py --database-config "$SQLITE_CONFIG" --run-background-updates
 
 # Create the PostgreSQL database.
 echo "Creating postgres databases..."
@@ -198,7 +204,7 @@ createdb --lc-collate=C --lc-ctype=C --template=template0 "$POSTGRES_MAIN_DB_NAM
 createdb --lc-collate=C --lc-ctype=C --template=template0 "$POSTGRES_STATE_DB_NAME"
 
 echo "Running db background jobs..."
-synapse/_scripts/update_synapse_database.py --database-config "$POSTGRES_CONFIG" --run-background-updates
+poetry run python synapse/_scripts/update_synapse_database.py --database-config "$POSTGRES_CONFIG" --run-background-updates
 
 
 echo "Dropping unwanted db tables..."
@@ -293,4 +299,12 @@ pg_dump --format=plain --data-only --inserts --no-tablespaces --no-acl --no-owne
 pg_dump --format=plain --schema-only         --no-tablespaces --no-acl --no-owner "$POSTGRES_STATE_DB_NAME"  | cleanup_pg_schema  > "$OUTPUT_DIR/state/full_schemas/$SCHEMA_NUMBER/full.sql.postgres"
 pg_dump --format=plain --data-only --inserts --no-tablespaces --no-acl --no-owner "$POSTGRES_STATE_DB_NAME"  | cleanup_pg_schema >> "$OUTPUT_DIR/state/full_schemas/$SCHEMA_NUMBER/full.sql.postgres"
 
+if [[ "$OUTPUT_DIR" == *synapse/storage/schema ]]; then
+  echo "Updating contrib/datagrip symlinks..."
+  ln -sf "../../synapse/storage/schema/common/full_schemas/$SCHEMA_NUMBER/full.sql.postgres" "contrib/datagrip/common.sql"
+  ln -sf "../../synapse/storage/schema/main/full_schemas/$SCHEMA_NUMBER/full.sql.postgres"   "contrib/datagrip/main.sql"
+  ln -sf "../../synapse/storage/schema/state/full_schemas/$SCHEMA_NUMBER/full.sql.postgres"  "contrib/datagrip/state.sql"
+else
+  echo "Not updating contrib/datagrip symlinks (unknown output directory)"
+fi
 echo "Done! Files dumped to: $OUTPUT_DIR"