librepages/src/pages/db.rs.html

1390 lines
57 KiB
HTML
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

<!DOCTYPE html><html lang="en"><head><meta charset="utf-8"><meta name="viewport" content="width=device-width, initial-scale=1.0"><meta name="generator" content="rustdoc"><meta name="description" content="Source of the Rust file `src/db.rs`."><meta name="keywords" content="rust, rustlang, rust-lang"><title>db.rs - source</title><link rel="preload" as="font" type="font/woff2" crossorigin href="../../SourceSerif4-Regular.ttf.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../FiraSans-Regular.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../FiraSans-Medium.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../SourceCodePro-Regular.ttf.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../SourceSerif4-Bold.ttf.woff2"><link rel="preload" as="font" type="font/woff2" crossorigin href="../../SourceCodePro-Semibold.ttf.woff2"><link rel="stylesheet" href="../../normalize.css"><link rel="stylesheet" href="../../rustdoc.css" id="mainThemeStyle"><link rel="stylesheet" href="../../ayu.css" disabled><link rel="stylesheet" href="../../dark.css" disabled><link rel="stylesheet" href="../../light.css" id="themeStyle"><script id="default-settings" ></script><script src="../../storage.js"></script><script defer src="../../source-script.js"></script><script defer src="../../source-files.js"></script><script defer src="../../main.js"></script><noscript><link rel="stylesheet" href="../../noscript.css"></noscript><link rel="alternate icon" type="image/png" href="../../favicon-16x16.png"><link rel="alternate icon" type="image/png" href="../../favicon-32x32.png"><link rel="icon" type="image/svg+xml" href="../../favicon.svg"></head><body class="rustdoc source"><!--[if lte IE 11]><div class="warning">This old browser is unsupported and will most likely display funky things.</div><![endif]--><nav class="mobile-topbar"><button class="sidebar-menu-toggle">&#9776;</button><a class="sidebar-logo" href="../../pages/index.html"><div class="logo-container"><img class="rust-logo" src="../../rust-logo.svg" alt="logo"></div></a><h2 class="location"></h2></nav><nav class="sidebar"><a class="sidebar-logo" href="../../pages/index.html"><div class="logo-container"><img class="rust-logo" src="../../rust-logo.svg" alt="logo"></div></a></nav><main><div class="width-limiter"><div class="sub-container"><a class="sub-logo-container" href="../../pages/index.html"><img class="rust-logo" src="../../rust-logo.svg" alt="logo"></a><nav class="sub"><form class="search-form"><div class="search-container"><span></span><input class="search-input" name="search" autocomplete="off" spellcheck="false" placeholder="Click or press S to search, ? for more options…" type="search"><div id="help-button" title="help" tabindex="-1"><button type="button">?</button></div><div id="settings-menu" tabindex="-1"><a href="../../settings.html" title="settings"><img width="22" height="22" alt="Change settings" src="../../wheel.svg"></a></div></div></form></nav></div><section id="main-content" class="content"><div class="example-wrap"><pre class="line-numbers"><span id="1">1</span>
<span id="2">2</span>
<span id="3">3</span>
<span id="4">4</span>
<span id="5">5</span>
<span id="6">6</span>
<span id="7">7</span>
<span id="8">8</span>
<span id="9">9</span>
<span id="10">10</span>
<span id="11">11</span>
<span id="12">12</span>
<span id="13">13</span>
<span id="14">14</span>
<span id="15">15</span>
<span id="16">16</span>
<span id="17">17</span>
<span id="18">18</span>
<span id="19">19</span>
<span id="20">20</span>
<span id="21">21</span>
<span id="22">22</span>
<span id="23">23</span>
<span id="24">24</span>
<span id="25">25</span>
<span id="26">26</span>
<span id="27">27</span>
<span id="28">28</span>
<span id="29">29</span>
<span id="30">30</span>
<span id="31">31</span>
<span id="32">32</span>
<span id="33">33</span>
<span id="34">34</span>
<span id="35">35</span>
<span id="36">36</span>
<span id="37">37</span>
<span id="38">38</span>
<span id="39">39</span>
<span id="40">40</span>
<span id="41">41</span>
<span id="42">42</span>
<span id="43">43</span>
<span id="44">44</span>
<span id="45">45</span>
<span id="46">46</span>
<span id="47">47</span>
<span id="48">48</span>
<span id="49">49</span>
<span id="50">50</span>
<span id="51">51</span>
<span id="52">52</span>
<span id="53">53</span>
<span id="54">54</span>
<span id="55">55</span>
<span id="56">56</span>
<span id="57">57</span>
<span id="58">58</span>
<span id="59">59</span>
<span id="60">60</span>
<span id="61">61</span>
<span id="62">62</span>
<span id="63">63</span>
<span id="64">64</span>
<span id="65">65</span>
<span id="66">66</span>
<span id="67">67</span>
<span id="68">68</span>
<span id="69">69</span>
<span id="70">70</span>
<span id="71">71</span>
<span id="72">72</span>
<span id="73">73</span>
<span id="74">74</span>
<span id="75">75</span>
<span id="76">76</span>
<span id="77">77</span>
<span id="78">78</span>
<span id="79">79</span>
<span id="80">80</span>
<span id="81">81</span>
<span id="82">82</span>
<span id="83">83</span>
<span id="84">84</span>
<span id="85">85</span>
<span id="86">86</span>
<span id="87">87</span>
<span id="88">88</span>
<span id="89">89</span>
<span id="90">90</span>
<span id="91">91</span>
<span id="92">92</span>
<span id="93">93</span>
<span id="94">94</span>
<span id="95">95</span>
<span id="96">96</span>
<span id="97">97</span>
<span id="98">98</span>
<span id="99">99</span>
<span id="100">100</span>
<span id="101">101</span>
<span id="102">102</span>
<span id="103">103</span>
<span id="104">104</span>
<span id="105">105</span>
<span id="106">106</span>
<span id="107">107</span>
<span id="108">108</span>
<span id="109">109</span>
<span id="110">110</span>
<span id="111">111</span>
<span id="112">112</span>
<span id="113">113</span>
<span id="114">114</span>
<span id="115">115</span>
<span id="116">116</span>
<span id="117">117</span>
<span id="118">118</span>
<span id="119">119</span>
<span id="120">120</span>
<span id="121">121</span>
<span id="122">122</span>
<span id="123">123</span>
<span id="124">124</span>
<span id="125">125</span>
<span id="126">126</span>
<span id="127">127</span>
<span id="128">128</span>
<span id="129">129</span>
<span id="130">130</span>
<span id="131">131</span>
<span id="132">132</span>
<span id="133">133</span>
<span id="134">134</span>
<span id="135">135</span>
<span id="136">136</span>
<span id="137">137</span>
<span id="138">138</span>
<span id="139">139</span>
<span id="140">140</span>
<span id="141">141</span>
<span id="142">142</span>
<span id="143">143</span>
<span id="144">144</span>
<span id="145">145</span>
<span id="146">146</span>
<span id="147">147</span>
<span id="148">148</span>
<span id="149">149</span>
<span id="150">150</span>
<span id="151">151</span>
<span id="152">152</span>
<span id="153">153</span>
<span id="154">154</span>
<span id="155">155</span>
<span id="156">156</span>
<span id="157">157</span>
<span id="158">158</span>
<span id="159">159</span>
<span id="160">160</span>
<span id="161">161</span>
<span id="162">162</span>
<span id="163">163</span>
<span id="164">164</span>
<span id="165">165</span>
<span id="166">166</span>
<span id="167">167</span>
<span id="168">168</span>
<span id="169">169</span>
<span id="170">170</span>
<span id="171">171</span>
<span id="172">172</span>
<span id="173">173</span>
<span id="174">174</span>
<span id="175">175</span>
<span id="176">176</span>
<span id="177">177</span>
<span id="178">178</span>
<span id="179">179</span>
<span id="180">180</span>
<span id="181">181</span>
<span id="182">182</span>
<span id="183">183</span>
<span id="184">184</span>
<span id="185">185</span>
<span id="186">186</span>
<span id="187">187</span>
<span id="188">188</span>
<span id="189">189</span>
<span id="190">190</span>
<span id="191">191</span>
<span id="192">192</span>
<span id="193">193</span>
<span id="194">194</span>
<span id="195">195</span>
<span id="196">196</span>
<span id="197">197</span>
<span id="198">198</span>
<span id="199">199</span>
<span id="200">200</span>
<span id="201">201</span>
<span id="202">202</span>
<span id="203">203</span>
<span id="204">204</span>
<span id="205">205</span>
<span id="206">206</span>
<span id="207">207</span>
<span id="208">208</span>
<span id="209">209</span>
<span id="210">210</span>
<span id="211">211</span>
<span id="212">212</span>
<span id="213">213</span>
<span id="214">214</span>
<span id="215">215</span>
<span id="216">216</span>
<span id="217">217</span>
<span id="218">218</span>
<span id="219">219</span>
<span id="220">220</span>
<span id="221">221</span>
<span id="222">222</span>
<span id="223">223</span>
<span id="224">224</span>
<span id="225">225</span>
<span id="226">226</span>
<span id="227">227</span>
<span id="228">228</span>
<span id="229">229</span>
<span id="230">230</span>
<span id="231">231</span>
<span id="232">232</span>
<span id="233">233</span>
<span id="234">234</span>
<span id="235">235</span>
<span id="236">236</span>
<span id="237">237</span>
<span id="238">238</span>
<span id="239">239</span>
<span id="240">240</span>
<span id="241">241</span>
<span id="242">242</span>
<span id="243">243</span>
<span id="244">244</span>
<span id="245">245</span>
<span id="246">246</span>
<span id="247">247</span>
<span id="248">248</span>
<span id="249">249</span>
<span id="250">250</span>
<span id="251">251</span>
<span id="252">252</span>
<span id="253">253</span>
<span id="254">254</span>
<span id="255">255</span>
<span id="256">256</span>
<span id="257">257</span>
<span id="258">258</span>
<span id="259">259</span>
<span id="260">260</span>
<span id="261">261</span>
<span id="262">262</span>
<span id="263">263</span>
<span id="264">264</span>
<span id="265">265</span>
<span id="266">266</span>
<span id="267">267</span>
<span id="268">268</span>
<span id="269">269</span>
<span id="270">270</span>
<span id="271">271</span>
<span id="272">272</span>
<span id="273">273</span>
<span id="274">274</span>
<span id="275">275</span>
<span id="276">276</span>
<span id="277">277</span>
<span id="278">278</span>
<span id="279">279</span>
<span id="280">280</span>
<span id="281">281</span>
<span id="282">282</span>
<span id="283">283</span>
<span id="284">284</span>
<span id="285">285</span>
<span id="286">286</span>
<span id="287">287</span>
<span id="288">288</span>
<span id="289">289</span>
<span id="290">290</span>
<span id="291">291</span>
<span id="292">292</span>
<span id="293">293</span>
<span id="294">294</span>
<span id="295">295</span>
<span id="296">296</span>
<span id="297">297</span>
<span id="298">298</span>
<span id="299">299</span>
<span id="300">300</span>
<span id="301">301</span>
<span id="302">302</span>
<span id="303">303</span>
<span id="304">304</span>
<span id="305">305</span>
<span id="306">306</span>
<span id="307">307</span>
<span id="308">308</span>
<span id="309">309</span>
<span id="310">310</span>
<span id="311">311</span>
<span id="312">312</span>
<span id="313">313</span>
<span id="314">314</span>
<span id="315">315</span>
<span id="316">316</span>
<span id="317">317</span>
<span id="318">318</span>
<span id="319">319</span>
<span id="320">320</span>
<span id="321">321</span>
<span id="322">322</span>
<span id="323">323</span>
<span id="324">324</span>
<span id="325">325</span>
<span id="326">326</span>
<span id="327">327</span>
<span id="328">328</span>
<span id="329">329</span>
<span id="330">330</span>
<span id="331">331</span>
<span id="332">332</span>
<span id="333">333</span>
<span id="334">334</span>
<span id="335">335</span>
<span id="336">336</span>
<span id="337">337</span>
<span id="338">338</span>
<span id="339">339</span>
<span id="340">340</span>
<span id="341">341</span>
<span id="342">342</span>
<span id="343">343</span>
<span id="344">344</span>
<span id="345">345</span>
<span id="346">346</span>
<span id="347">347</span>
<span id="348">348</span>
<span id="349">349</span>
<span id="350">350</span>
<span id="351">351</span>
<span id="352">352</span>
<span id="353">353</span>
<span id="354">354</span>
<span id="355">355</span>
<span id="356">356</span>
<span id="357">357</span>
<span id="358">358</span>
<span id="359">359</span>
<span id="360">360</span>
<span id="361">361</span>
<span id="362">362</span>
<span id="363">363</span>
<span id="364">364</span>
<span id="365">365</span>
<span id="366">366</span>
<span id="367">367</span>
<span id="368">368</span>
<span id="369">369</span>
<span id="370">370</span>
<span id="371">371</span>
<span id="372">372</span>
<span id="373">373</span>
<span id="374">374</span>
<span id="375">375</span>
<span id="376">376</span>
<span id="377">377</span>
<span id="378">378</span>
<span id="379">379</span>
<span id="380">380</span>
<span id="381">381</span>
<span id="382">382</span>
<span id="383">383</span>
<span id="384">384</span>
<span id="385">385</span>
<span id="386">386</span>
<span id="387">387</span>
<span id="388">388</span>
<span id="389">389</span>
<span id="390">390</span>
<span id="391">391</span>
<span id="392">392</span>
<span id="393">393</span>
<span id="394">394</span>
<span id="395">395</span>
<span id="396">396</span>
<span id="397">397</span>
<span id="398">398</span>
<span id="399">399</span>
<span id="400">400</span>
<span id="401">401</span>
<span id="402">402</span>
<span id="403">403</span>
<span id="404">404</span>
<span id="405">405</span>
<span id="406">406</span>
<span id="407">407</span>
<span id="408">408</span>
<span id="409">409</span>
<span id="410">410</span>
<span id="411">411</span>
<span id="412">412</span>
<span id="413">413</span>
<span id="414">414</span>
<span id="415">415</span>
<span id="416">416</span>
<span id="417">417</span>
<span id="418">418</span>
<span id="419">419</span>
<span id="420">420</span>
<span id="421">421</span>
<span id="422">422</span>
<span id="423">423</span>
<span id="424">424</span>
<span id="425">425</span>
<span id="426">426</span>
<span id="427">427</span>
<span id="428">428</span>
<span id="429">429</span>
<span id="430">430</span>
<span id="431">431</span>
<span id="432">432</span>
<span id="433">433</span>
<span id="434">434</span>
<span id="435">435</span>
<span id="436">436</span>
<span id="437">437</span>
<span id="438">438</span>
<span id="439">439</span>
<span id="440">440</span>
<span id="441">441</span>
<span id="442">442</span>
<span id="443">443</span>
<span id="444">444</span>
<span id="445">445</span>
<span id="446">446</span>
<span id="447">447</span>
<span id="448">448</span>
<span id="449">449</span>
<span id="450">450</span>
<span id="451">451</span>
<span id="452">452</span>
<span id="453">453</span>
<span id="454">454</span>
<span id="455">455</span>
<span id="456">456</span>
<span id="457">457</span>
<span id="458">458</span>
<span id="459">459</span>
<span id="460">460</span>
<span id="461">461</span>
<span id="462">462</span>
<span id="463">463</span>
<span id="464">464</span>
<span id="465">465</span>
<span id="466">466</span>
<span id="467">467</span>
<span id="468">468</span>
<span id="469">469</span>
<span id="470">470</span>
<span id="471">471</span>
<span id="472">472</span>
<span id="473">473</span>
<span id="474">474</span>
<span id="475">475</span>
<span id="476">476</span>
<span id="477">477</span>
<span id="478">478</span>
<span id="479">479</span>
<span id="480">480</span>
<span id="481">481</span>
<span id="482">482</span>
<span id="483">483</span>
<span id="484">484</span>
<span id="485">485</span>
<span id="486">486</span>
<span id="487">487</span>
<span id="488">488</span>
<span id="489">489</span>
<span id="490">490</span>
<span id="491">491</span>
<span id="492">492</span>
<span id="493">493</span>
<span id="494">494</span>
<span id="495">495</span>
<span id="496">496</span>
<span id="497">497</span>
<span id="498">498</span>
<span id="499">499</span>
<span id="500">500</span>
<span id="501">501</span>
<span id="502">502</span>
<span id="503">503</span>
<span id="504">504</span>
<span id="505">505</span>
<span id="506">506</span>
<span id="507">507</span>
<span id="508">508</span>
<span id="509">509</span>
<span id="510">510</span>
<span id="511">511</span>
<span id="512">512</span>
<span id="513">513</span>
<span id="514">514</span>
<span id="515">515</span>
<span id="516">516</span>
<span id="517">517</span>
<span id="518">518</span>
<span id="519">519</span>
<span id="520">520</span>
<span id="521">521</span>
<span id="522">522</span>
<span id="523">523</span>
<span id="524">524</span>
<span id="525">525</span>
<span id="526">526</span>
<span id="527">527</span>
<span id="528">528</span>
<span id="529">529</span>
<span id="530">530</span>
<span id="531">531</span>
<span id="532">532</span>
<span id="533">533</span>
<span id="534">534</span>
<span id="535">535</span>
<span id="536">536</span>
<span id="537">537</span>
<span id="538">538</span>
<span id="539">539</span>
<span id="540">540</span>
<span id="541">541</span>
<span id="542">542</span>
<span id="543">543</span>
<span id="544">544</span>
<span id="545">545</span>
<span id="546">546</span>
<span id="547">547</span>
<span id="548">548</span>
<span id="549">549</span>
<span id="550">550</span>
<span id="551">551</span>
<span id="552">552</span>
<span id="553">553</span>
<span id="554">554</span>
<span id="555">555</span>
<span id="556">556</span>
<span id="557">557</span>
<span id="558">558</span>
<span id="559">559</span>
<span id="560">560</span>
<span id="561">561</span>
<span id="562">562</span>
<span id="563">563</span>
<span id="564">564</span>
<span id="565">565</span>
<span id="566">566</span>
<span id="567">567</span>
<span id="568">568</span>
<span id="569">569</span>
<span id="570">570</span>
<span id="571">571</span>
<span id="572">572</span>
<span id="573">573</span>
<span id="574">574</span>
<span id="575">575</span>
<span id="576">576</span>
<span id="577">577</span>
<span id="578">578</span>
<span id="579">579</span>
<span id="580">580</span>
<span id="581">581</span>
<span id="582">582</span>
<span id="583">583</span>
<span id="584">584</span>
<span id="585">585</span>
<span id="586">586</span>
<span id="587">587</span>
<span id="588">588</span>
<span id="589">589</span>
<span id="590">590</span>
<span id="591">591</span>
<span id="592">592</span>
<span id="593">593</span>
<span id="594">594</span>
<span id="595">595</span>
<span id="596">596</span>
<span id="597">597</span>
<span id="598">598</span>
<span id="599">599</span>
<span id="600">600</span>
<span id="601">601</span>
<span id="602">602</span>
<span id="603">603</span>
<span id="604">604</span>
<span id="605">605</span>
<span id="606">606</span>
<span id="607">607</span>
<span id="608">608</span>
<span id="609">609</span>
<span id="610">610</span>
<span id="611">611</span>
<span id="612">612</span>
<span id="613">613</span>
<span id="614">614</span>
<span id="615">615</span>
<span id="616">616</span>
<span id="617">617</span>
<span id="618">618</span>
<span id="619">619</span>
<span id="620">620</span>
<span id="621">621</span>
<span id="622">622</span>
<span id="623">623</span>
<span id="624">624</span>
<span id="625">625</span>
<span id="626">626</span>
<span id="627">627</span>
<span id="628">628</span>
<span id="629">629</span>
<span id="630">630</span>
<span id="631">631</span>
<span id="632">632</span>
<span id="633">633</span>
<span id="634">634</span>
<span id="635">635</span>
<span id="636">636</span>
<span id="637">637</span>
<span id="638">638</span>
<span id="639">639</span>
<span id="640">640</span>
<span id="641">641</span>
<span id="642">642</span>
<span id="643">643</span>
<span id="644">644</span>
<span id="645">645</span>
<span id="646">646</span>
<span id="647">647</span>
<span id="648">648</span>
<span id="649">649</span>
<span id="650">650</span>
<span id="651">651</span>
<span id="652">652</span>
<span id="653">653</span>
<span id="654">654</span>
<span id="655">655</span>
<span id="656">656</span>
<span id="657">657</span>
<span id="658">658</span>
<span id="659">659</span>
<span id="660">660</span>
<span id="661">661</span>
<span id="662">662</span>
<span id="663">663</span>
<span id="664">664</span>
<span id="665">665</span>
<span id="666">666</span>
<span id="667">667</span>
<span id="668">668</span>
<span id="669">669</span>
<span id="670">670</span>
<span id="671">671</span>
<span id="672">672</span>
<span id="673">673</span>
<span id="674">674</span>
<span id="675">675</span>
<span id="676">676</span>
<span id="677">677</span>
<span id="678">678</span>
<span id="679">679</span>
<span id="680">680</span>
<span id="681">681</span>
<span id="682">682</span>
<span id="683">683</span>
<span id="684">684</span>
<span id="685">685</span>
<span id="686">686</span>
<span id="687">687</span>
<span id="688">688</span>
<span id="689">689</span>
<span id="690">690</span>
<span id="691">691</span>
<span id="692">692</span>
<span id="693">693</span>
<span id="694">694</span>
</pre><pre class="rust"><code><span class="comment">/*
* Copyright (C) 2022 Aravinth Manivannan &lt;realaravinth@batsense.net&gt;
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as
* published by the Free Software Foundation, either version 3 of the
* License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see &lt;https://www.gnu.org/licenses/&gt;.
*/
</span><span class="kw">use </span>std::str::FromStr;
<span class="kw">use </span>serde::{Deserialize, Serialize};
<span class="kw">use </span>sqlx::postgres::PgPoolOptions;
<span class="kw">use </span>sqlx::types::time::OffsetDateTime;
<span class="comment">//use sqlx::types::Json;
</span><span class="kw">use </span>sqlx::ConnectOptions;
<span class="kw">use </span>sqlx::PgPool;
<span class="kw">use </span>url::quirks::hostname;
<span class="kw">use </span><span class="kw">crate</span>::errors::<span class="kw-2">*</span>;
<span class="doccomment">/// Connect to databse
</span><span class="kw">pub enum </span>ConnectionOptions {
<span class="doccomment">/// fresh connection
</span>Fresh(Fresh),
<span class="doccomment">/// existing connection
</span>Existing(Conn),
}
<span class="doccomment">/// Use an existing database pool
</span><span class="kw">pub struct </span>Conn(<span class="kw">pub </span>PgPool);
<span class="kw">pub struct </span>Fresh {
<span class="kw">pub </span>pool_options: PgPoolOptions,
<span class="kw">pub </span>disable_logging: bool,
<span class="kw">pub </span>url: String,
}
<span class="kw">impl </span>ConnectionOptions {
<span class="kw">async fn </span>connect(<span class="self">self</span>) -&gt; ServiceResult&lt;Database&gt; {
<span class="kw">let </span>pool = <span class="kw">match </span><span class="self">self </span>{
<span class="self">Self</span>::Fresh(fresh) =&gt; {
<span class="kw">let </span><span class="kw-2">mut </span>connect_options =
sqlx::postgres::PgConnectOptions::from_str(<span class="kw-2">&amp;</span>fresh.url).unwrap();
<span class="kw">if </span>fresh.disable_logging {
connect_options.disable_statement_logging();
}
sqlx::postgres::PgConnectOptions::from_str(<span class="kw-2">&amp;</span>fresh.url)
.unwrap()
.disable_statement_logging();
fresh
.pool_options
.connect_with(connect_options)
.<span class="kw">await
</span>.unwrap()
<span class="comment">//.map_err(|e| ServiceError::ServiceError(Box::new(e)))?
</span>}
<span class="self">Self</span>::Existing(conn) =&gt; conn.<span class="number">0</span>,
};
<span class="prelude-val">Ok</span>(Database { pool })
}
}
<span class="attribute">#[derive(Clone)]
</span><span class="kw">pub struct </span>Database {
<span class="kw">pub </span>pool: PgPool,
}
<span class="kw">impl </span>Database {
<span class="kw">pub async fn </span>migrate(<span class="kw-2">&amp;</span><span class="self">self</span>) -&gt; ServiceResult&lt;()&gt; {
<span class="macro">sqlx::migrate!</span>(<span class="string">&quot;./migrations/&quot;</span>)
.run(<span class="kw-2">&amp;</span><span class="self">self</span>.pool)
.<span class="kw">await
</span>.unwrap();
<span class="comment">//.map_err(|e| ServiceError::ServiceError(Box::new(e)))?;
</span><span class="prelude-val">Ok</span>(())
}
<span class="kw">pub async fn </span>ping(<span class="kw-2">&amp;</span><span class="self">self</span>) -&gt; bool {
<span class="kw">use </span>sqlx::Connection;
<span class="kw">if let </span><span class="prelude-val">Ok</span>(<span class="kw-2">mut </span>con) = <span class="self">self</span>.pool.acquire().<span class="kw">await </span>{
con.ping().<span class="kw">await</span>.is_ok()
} <span class="kw">else </span>{
<span class="bool-val">false
</span>}
}
<span class="doccomment">/// register a new user
</span><span class="kw">pub async fn </span>register(<span class="kw-2">&amp;</span><span class="self">self</span>, p: <span class="kw-2">&amp;</span>Register&lt;<span class="lifetime">&#39;_</span>&gt;) -&gt; ServiceResult&lt;()&gt; {
<span class="macro">sqlx::query!</span>(
<span class="string">&quot;INSERT INTO librepages_users
(name , password, email) VALUES ($1, $2, $3)&quot;</span>,
<span class="kw-2">&amp;</span>p.username,
<span class="kw-2">&amp;</span>p.hash,
<span class="kw-2">&amp;</span>p.email,
)
.execute(<span class="kw-2">&amp;</span><span class="self">self</span>.pool)
.<span class="kw">await
</span>.map_err(map_register_err)<span class="question-mark">?</span>;
<span class="prelude-val">Ok</span>(())
}
<span class="doccomment">/// delete a user
</span><span class="kw">pub async fn </span>delete_user(<span class="kw-2">&amp;</span><span class="self">self</span>, username: <span class="kw-2">&amp;</span>str) -&gt; ServiceResult&lt;()&gt; {
<span class="macro">sqlx::query!</span>(<span class="string">&quot;DELETE FROM librepages_users WHERE name = ($1)&quot;</span>, username)
.execute(<span class="kw-2">&amp;</span><span class="self">self</span>.pool)
.<span class="kw">await
</span>.map_err(|e| map_row_not_found_err(e, ServiceError::AccountNotFound))<span class="question-mark">?</span>;
<span class="prelude-val">Ok</span>(())
}
<span class="doccomment">/// check if username exists
</span><span class="kw">pub async fn </span>username_exists(<span class="kw-2">&amp;</span><span class="self">self</span>, username: <span class="kw-2">&amp;</span>str) -&gt; ServiceResult&lt;bool&gt; {
<span class="kw">let </span>res = <span class="macro">sqlx::query!</span>(
<span class="string">&quot;SELECT EXISTS (SELECT 1 from librepages_users WHERE name = $1)&quot;</span>,
username,
)
.fetch_one(<span class="kw-2">&amp;</span><span class="self">self</span>.pool)
.<span class="kw">await
</span>.map_err(map_register_err)<span class="question-mark">?</span>;
<span class="kw">let </span><span class="kw-2">mut </span>resp = <span class="bool-val">false</span>;
<span class="kw">if let </span><span class="prelude-val">Some</span>(x) = res.exists {
resp = x;
}
<span class="prelude-val">Ok</span>(resp)
}
<span class="doccomment">/// get user email
</span><span class="kw">pub async fn </span>get_email(<span class="kw-2">&amp;</span><span class="self">self</span>, username: <span class="kw-2">&amp;</span>str) -&gt; ServiceResult&lt;String&gt; {
<span class="kw">struct </span>Email {
email: String,
}
<span class="kw">let </span>res = <span class="macro">sqlx::query_as!</span>(
Email,
<span class="string">&quot;SELECT email FROM librepages_users WHERE name = $1&quot;</span>,
username
)
.fetch_one(<span class="kw-2">&amp;</span><span class="self">self</span>.pool)
.<span class="kw">await
</span>.map_err(|e| map_row_not_found_err(e, ServiceError::AccountNotFound))<span class="question-mark">?</span>;
<span class="prelude-val">Ok</span>(res.email)
}
<span class="doccomment">/// check if email exists
</span><span class="kw">pub async fn </span>email_exists(<span class="kw-2">&amp;</span><span class="self">self</span>, email: <span class="kw-2">&amp;</span>str) -&gt; ServiceResult&lt;bool&gt; {
<span class="kw">let </span>res = <span class="macro">sqlx::query!</span>(
<span class="string">&quot;SELECT EXISTS (SELECT 1 from librepages_users WHERE email = $1)&quot;</span>,
email
)
.fetch_one(<span class="kw-2">&amp;</span><span class="self">self</span>.pool)
.<span class="kw">await
</span>.map_err(map_register_err)<span class="question-mark">?</span>;
<span class="kw">let </span><span class="kw-2">mut </span>resp = <span class="bool-val">false</span>;
<span class="kw">if let </span><span class="prelude-val">Some</span>(x) = res.exists {
resp = x;
}
<span class="prelude-val">Ok</span>(resp)
}
<span class="doccomment">/// update a user&#39;s email
</span><span class="kw">pub async fn </span>update_email(<span class="kw-2">&amp;</span><span class="self">self</span>, p: <span class="kw-2">&amp;</span>UpdateEmail&lt;<span class="lifetime">&#39;_</span>&gt;) -&gt; ServiceResult&lt;()&gt; {
<span class="macro">sqlx::query!</span>(
<span class="string">&quot;UPDATE librepages_users set email = $1
WHERE name = $2&quot;</span>,
<span class="kw-2">&amp;</span>p.new_email,
<span class="kw-2">&amp;</span>p.username,
)
.execute(<span class="kw-2">&amp;</span><span class="self">self</span>.pool)
.<span class="kw">await
</span>.map_err(|e| map_row_not_found_err(e, ServiceError::AccountNotFound))<span class="question-mark">?</span>;
<span class="prelude-val">Ok</span>(())
}
<span class="doccomment">/// get a user&#39;s password
</span><span class="kw">pub async fn </span>get_password(<span class="kw-2">&amp;</span><span class="self">self</span>, l: <span class="kw-2">&amp;</span>Login&lt;<span class="lifetime">&#39;_</span>&gt;) -&gt; ServiceResult&lt;NameHash&gt; {
<span class="kw">struct </span>Password {
name: String,
password: String,
}
<span class="kw">let </span>rec = <span class="kw">match </span>l {
Login::Username(u) =&gt; <span class="macro">sqlx::query_as!</span>(
Password,
<span class="string">r#&quot;SELECT name, password FROM librepages_users WHERE name = ($1)&quot;#</span>,
u,
)
.fetch_one(<span class="kw-2">&amp;</span><span class="self">self</span>.pool)
.<span class="kw">await
</span>.map_err(|e| map_row_not_found_err(e, ServiceError::AccountNotFound))<span class="question-mark">?</span>,
Login::Email(e) =&gt; <span class="macro">sqlx::query_as!</span>(
Password,
<span class="string">r#&quot;SELECT name, password FROM librepages_users WHERE email = ($1)&quot;#</span>,
e,
)
.fetch_one(<span class="kw-2">&amp;</span><span class="self">self</span>.pool)
.<span class="kw">await
</span>.map_err(|e| map_row_not_found_err(e, ServiceError::AccountNotFound))<span class="question-mark">?</span>,
};
<span class="kw">let </span>res = NameHash {
hash: rec.password,
username: rec.name,
};
<span class="prelude-val">Ok</span>(res)
}
<span class="doccomment">/// update user&#39;s password
</span><span class="kw">pub async fn </span>update_password(<span class="kw-2">&amp;</span><span class="self">self</span>, p: <span class="kw-2">&amp;</span>NameHash) -&gt; ServiceResult&lt;()&gt; {
<span class="macro">sqlx::query!</span>(
<span class="string">&quot;UPDATE librepages_users set password = $1
WHERE name = $2&quot;</span>,
<span class="kw-2">&amp;</span>p.hash,
<span class="kw-2">&amp;</span>p.username,
)
.execute(<span class="kw-2">&amp;</span><span class="self">self</span>.pool)
.<span class="kw">await
</span>.map_err(|e| map_row_not_found_err(e, ServiceError::AccountNotFound))<span class="question-mark">?</span>;
<span class="prelude-val">Ok</span>(())
}
<span class="doccomment">/// update username
</span><span class="kw">pub async fn </span>update_username(<span class="kw-2">&amp;</span><span class="self">self</span>, current: <span class="kw-2">&amp;</span>str, new: <span class="kw-2">&amp;</span>str) -&gt; ServiceResult&lt;()&gt; {
<span class="macro">sqlx::query!</span>(
<span class="string">&quot;UPDATE librepages_users set name = $1
WHERE name = $2&quot;</span>,
new,
current,
)
.execute(<span class="kw-2">&amp;</span><span class="self">self</span>.pool)
.<span class="kw">await
</span>.map_err(|e| map_row_not_found_err(e, ServiceError::AccountNotFound))<span class="question-mark">?</span>;
<span class="prelude-val">Ok</span>(())
}
<span class="kw">pub async fn </span>add_site(<span class="kw-2">&amp;</span><span class="self">self</span>, msg: <span class="kw-2">&amp;</span>Site) -&gt; ServiceResult&lt;()&gt; {
<span class="macro">sqlx::query!</span>(
<span class="string">&quot;
INSERT INTO librepages_sites
(site_secret, repo_url, branch, hostname, owned_by)
VALUES ($1, $2, $3, $4, ( SELECT ID FROM librepages_users WHERE name = $5 ));
&quot;</span>,
msg.site_secret,
msg.repo_url,
msg.branch,
msg.hostname,
msg.owner,
)
.execute(<span class="kw-2">&amp;</span><span class="self">self</span>.pool)
.<span class="kw">await
</span>.map_err(|e| map_row_not_found_err(e, ServiceError::AccountNotFound))<span class="question-mark">?</span>;
<span class="prelude-val">Ok</span>(())
}
<span class="kw">pub async fn </span>get_site_from_secret(<span class="kw-2">&amp;</span><span class="self">self</span>, site_secret: <span class="kw-2">&amp;</span>str) -&gt; ServiceResult&lt;Site&gt; {
<span class="kw">struct </span>S {
repo_url: String,
branch: String,
hostname: String,
owned_by: i32,
}
<span class="kw">let </span>site = <span class="macro">sqlx::query_as!</span>(
S,
<span class="string">&quot;SELECT repo_url, branch, hostname, owned_by
FROM librepages_sites
WHERE site_secret = $1
&quot;</span>,
site_secret,
)
.fetch_one(<span class="kw-2">&amp;</span><span class="self">self</span>.pool)
.<span class="kw">await
</span>.map_err(|e| map_row_not_found_err(e, ServiceError::WebsiteNotFound))<span class="question-mark">?</span>;
<span class="kw">struct </span>Owner {
name: String,
}
<span class="kw">let </span>owner = <span class="macro">sqlx::query_as!</span>(
Owner,
<span class="string">&quot;SELECT name FROM librepages_users WHERE ID = $1&quot;</span>,
site.owned_by
)
.fetch_one(<span class="kw-2">&amp;</span><span class="self">self</span>.pool)
.<span class="kw">await
</span>.map_err(|e| map_row_not_found_err(e, ServiceError::WebsiteNotFound))<span class="question-mark">?</span>;
<span class="kw">let </span>site = Site {
site_secret: site_secret.to_owned(),
branch: site.branch,
hostname: site.hostname,
owner: owner.name,
repo_url: site.repo_url,
};
<span class="prelude-val">Ok</span>(site)
}
<span class="kw">pub async fn </span>get_site(<span class="kw-2">&amp;</span><span class="self">self</span>, owner: <span class="kw-2">&amp;</span>str, hostname: <span class="kw-2">&amp;</span>str) -&gt; ServiceResult&lt;Site&gt; {
<span class="kw">let </span>site = <span class="macro">sqlx::query_as!</span>(
InnerSite,
<span class="string">&quot;SELECT site_secret, repo_url, branch, hostname
FROM librepages_sites
WHERE owned_by = (SELECT ID FROM librepages_users WHERE name = $1 )
AND hostname = $2;
&quot;</span>,
owner,
hostname
)
.fetch_one(<span class="kw-2">&amp;</span><span class="self">self</span>.pool)
.<span class="kw">await
</span>.map_err(|e| map_row_not_found_err(e, ServiceError::WebsiteNotFound))<span class="question-mark">?</span>;
<span class="kw">let </span>res = site.to_site(owner.into());
<span class="prelude-val">Ok</span>(res)
}
<span class="kw">pub async fn </span>list_all_sites(<span class="kw-2">&amp;</span><span class="self">self</span>, owner: <span class="kw-2">&amp;</span>str) -&gt; ServiceResult&lt;Vec&lt;Site&gt;&gt; {
<span class="kw">let </span><span class="kw-2">mut </span>sites = <span class="macro">sqlx::query_as!</span>(
InnerSite,
<span class="string">&quot;SELECT site_secret, repo_url, branch, hostname
FROM librepages_sites
WHERE owned_by = (SELECT ID FROM librepages_users WHERE name = $1 );
&quot;</span>,
owner,
)
.fetch_all(<span class="kw-2">&amp;</span><span class="self">self</span>.pool)
.<span class="kw">await
</span>.map_err(|e| map_row_not_found_err(e, ServiceError::AccountNotFound))<span class="question-mark">?</span>;
<span class="kw">let </span>res = sites.drain(<span class="number">0</span>..).map(|s| s.to_site(owner.into())).collect();
<span class="prelude-val">Ok</span>(res)
}
<span class="kw">pub async fn </span>delete_site(<span class="kw-2">&amp;</span><span class="self">self</span>, owner: <span class="kw-2">&amp;</span>str, hostname: <span class="kw-2">&amp;</span>str) -&gt; ServiceResult&lt;()&gt; {
<span class="macro">sqlx::query!</span>(
<span class="string">&quot;DELETE FROM librepages_sites
WHERE hostname = ($1)
AND owned_by = ( SELECT ID FROM librepages_users WHERE name = $2);
&quot;</span>,
hostname,
owner
)
.execute(<span class="kw-2">&amp;</span><span class="self">self</span>.pool)
.<span class="kw">await
</span>.map_err(|e| map_row_not_found_err(e, ServiceError::WebsiteNotFound))<span class="question-mark">?</span>;
<span class="prelude-val">Ok</span>(())
}
<span class="doccomment">/// check if hostname exists
</span><span class="kw">pub async fn </span>hostname_exists(<span class="kw-2">&amp;</span><span class="self">self</span>, hostname: <span class="kw-2">&amp;</span>str) -&gt; ServiceResult&lt;bool&gt; {
<span class="kw">let </span>res = <span class="macro">sqlx::query!</span>(
<span class="string">&quot;SELECT EXISTS (SELECT 1 from librepages_sites WHERE hostname = $1)&quot;</span>,
hostname,
)
.fetch_one(<span class="kw-2">&amp;</span><span class="self">self</span>.pool)
.<span class="kw">await
</span>.map_err(map_register_err)<span class="question-mark">?</span>;
<span class="kw">let </span><span class="kw-2">mut </span>resp = <span class="bool-val">false</span>;
<span class="kw">if let </span><span class="prelude-val">Some</span>(x) = res.exists {
resp = x;
}
<span class="prelude-val">Ok</span>(resp)
}
}
<span class="kw">struct </span>InnerSite {
site_secret: String,
repo_url: String,
branch: String,
hostname: String,
}
<span class="kw">impl </span>InnerSite {
<span class="kw">fn </span>to_site(<span class="self">self</span>, owner: String) -&gt; Site {
Site {
site_secret: <span class="self">self</span>.site_secret,
repo_url: <span class="self">self</span>.repo_url,
branch: <span class="self">self</span>.branch,
hostname: <span class="self">self</span>.hostname,
owner,
}
}
}
<span class="attribute">#[derive(Clone, Debug, Deserialize, Serialize, Eq, PartialEq)]
</span><span class="doccomment">/// Data required to add a new site
</span><span class="kw">pub struct </span>Site {
<span class="kw">pub </span>site_secret: String,
<span class="kw">pub </span>repo_url: String,
<span class="kw">pub </span>branch: String,
<span class="kw">pub </span>hostname: String,
<span class="kw">pub </span>owner: String,
}
<span class="attribute">#[derive(Clone, Debug, Deserialize, Serialize, Eq, PartialEq)]
</span><span class="doccomment">/// Data required to register a new user
</span><span class="kw">pub struct </span>Register&lt;<span class="lifetime">&#39;a</span>&gt; {
<span class="doccomment">/// username of new user
</span><span class="kw">pub </span>username: <span class="kw-2">&amp;</span><span class="lifetime">&#39;a </span>str,
<span class="doccomment">/// hashed password of new use
</span><span class="kw">pub </span>hash: <span class="kw-2">&amp;</span><span class="lifetime">&#39;a </span>str,
<span class="doccomment">/// Optionally, email of new use
</span><span class="kw">pub </span>email: <span class="kw-2">&amp;</span><span class="lifetime">&#39;a </span>str,
}
<span class="attribute">#[derive(Clone, Debug, Deserialize, Serialize, Eq, PartialEq)]
</span><span class="doccomment">/// data required to update them email of a user
</span><span class="kw">pub struct </span>UpdateEmail&lt;<span class="lifetime">&#39;a</span>&gt; {
<span class="doccomment">/// username of the user
</span><span class="kw">pub </span>username: <span class="kw-2">&amp;</span><span class="lifetime">&#39;a </span>str,
<span class="doccomment">/// new email address of the user
</span><span class="kw">pub </span>new_email: <span class="kw-2">&amp;</span><span class="lifetime">&#39;a </span>str,
}
<span class="attribute">#[derive(Clone, Debug, Deserialize, Serialize, Eq, PartialEq)]
</span><span class="doccomment">/// types of credentials used as identifiers during login
</span><span class="kw">pub enum </span>Login&lt;<span class="lifetime">&#39;a</span>&gt; {
<span class="doccomment">/// username as login
</span>Username(<span class="kw-2">&amp;</span><span class="lifetime">&#39;a </span>str),
<span class="doccomment">/// email as login
</span>Email(<span class="kw-2">&amp;</span><span class="lifetime">&#39;a </span>str),
}
<span class="attribute">#[derive(Clone, Debug, Deserialize, Serialize, Eq, PartialEq)]
</span><span class="doccomment">/// type encapsulating username and hashed password of a user
</span><span class="kw">pub struct </span>NameHash {
<span class="doccomment">/// username
</span><span class="kw">pub </span>username: String,
<span class="doccomment">/// hashed password
</span><span class="kw">pub </span>hash: String,
}
<span class="kw">fn </span>now_unix_time_stamp() -&gt; OffsetDateTime {
OffsetDateTime::now_utc()
}
<span class="kw">pub async fn </span>get_db(settings: <span class="kw-2">&amp;</span><span class="kw">crate</span>::settings::Settings) -&gt; Database {
<span class="kw">let </span>pool_options = PgPoolOptions::new().max_connections(settings.database.pool);
ConnectionOptions::Fresh(Fresh {
pool_options,
url: settings.database.url.clone(),
disable_logging: !settings.debug,
})
.connect()
.<span class="kw">await
</span>.unwrap()
}
<span class="doccomment">/// map custom row not found error to DB error
</span><span class="kw">pub fn </span>map_row_not_found_err(e: sqlx::Error, row_not_found: ServiceError) -&gt; ServiceError {
<span class="kw">if let </span>sqlx::Error::RowNotFound = e {
row_not_found
} <span class="kw">else </span>{
map_register_err(e)
}
}
<span class="doccomment">/// map postgres errors to [ServiceError](ServiceError) types
</span><span class="kw">fn </span>map_register_err(e: sqlx::Error) -&gt; ServiceError {
<span class="kw">use </span>sqlx::Error;
<span class="kw">use </span>std::borrow::Cow;
<span class="kw">if let </span>Error::Database(err) = e {
<span class="kw">if </span>err.code() == <span class="prelude-val">Some</span>(Cow::from(<span class="string">&quot;23505&quot;</span>)) {
<span class="kw">let </span>msg = err.message();
<span class="macro">println!</span>(<span class="string">&quot;{}&quot;</span>, msg);
<span class="kw">if </span>msg.contains(<span class="string">&quot;librepages_users_name_key&quot;</span>) {
ServiceError::UsernameTaken
} <span class="kw">else if </span>msg.contains(<span class="string">&quot;librepages_users_email_key&quot;</span>) {
ServiceError::EmailTaken
} <span class="kw">else </span>{
<span class="macro">log::error!</span>(<span class="string">&quot;{}&quot;</span>, msg);
ServiceError::InternalServerError
}
} <span class="kw">else </span>{
ServiceError::InternalServerError
}
} <span class="kw">else </span>{
ServiceError::InternalServerError
}
}
<span class="attribute">#[cfg(test)]
</span><span class="kw">mod </span>tests {
<span class="kw">use super</span>::<span class="kw-2">*</span>;
<span class="kw">use </span><span class="kw">crate</span>::settings::Settings;
<span class="attribute">#[actix_rt::test]
</span><span class="kw">async fn </span>db_works() {
<span class="kw">let </span>settings = Settings::new().unwrap();
<span class="kw">let </span>pool_options = PgPoolOptions::new().max_connections(<span class="number">1</span>);
<span class="kw">let </span>db = ConnectionOptions::Fresh(Fresh {
pool_options,
url: settings.database.url.clone(),
disable_logging: !settings.debug,
})
.connect()
.<span class="kw">await
</span>.unwrap();
<span class="macro">assert!</span>(db.ping().<span class="kw">await</span>);
<span class="kw">const </span>EMAIL: <span class="kw-2">&amp;</span>str = <span class="string">&quot;postgresuser@foo.com&quot;</span>;
<span class="kw">const </span>EMAIL2: <span class="kw-2">&amp;</span>str = <span class="string">&quot;postgresuser2@foo.com&quot;</span>;
<span class="kw">const </span>NAME: <span class="kw-2">&amp;</span>str = <span class="string">&quot;postgresuser&quot;</span>;
<span class="kw">const </span>PASSWORD: <span class="kw-2">&amp;</span>str = <span class="string">&quot;pasdfasdfasdfadf&quot;</span>;
db.migrate().<span class="kw">await</span>.unwrap();
<span class="kw">let </span>p = <span class="kw">super</span>::Register {
username: NAME,
email: EMAIL,
hash: PASSWORD,
};
<span class="kw">if </span>db.username_exists(p.username).<span class="kw">await</span>.unwrap() {
db.delete_user(p.username).<span class="kw">await</span>.unwrap();
<span class="macro">assert!</span>(
!db.username_exists(p.username).<span class="kw">await</span>.unwrap(),
<span class="string">&quot;user is deleted so username shouldn&#39;t exist&quot;
</span>);
}
db.register(<span class="kw-2">&amp;</span>p).<span class="kw">await</span>.unwrap();
<span class="macro">assert!</span>(<span class="macro">matches!</span>(
db.register(<span class="kw-2">&amp;</span>p).<span class="kw">await</span>,
<span class="prelude-val">Err</span>(ServiceError::UsernameTaken)
));
<span class="comment">// testing get_password
// with username
</span><span class="kw">let </span>name_hash = db.get_password(<span class="kw-2">&amp;</span>Login::Username(p.username)).<span class="kw">await</span>.unwrap();
<span class="macro">assert_eq!</span>(name_hash.hash, p.hash, <span class="string">&quot;user password matches&quot;</span>);
<span class="macro">assert_eq!</span>(name_hash.username, p.username, <span class="string">&quot;username matches&quot;</span>);
<span class="comment">// with email
</span><span class="kw">let </span><span class="kw-2">mut </span>name_hash = db.get_password(<span class="kw-2">&amp;</span>Login::Email(p.email)).<span class="kw">await</span>.unwrap();
<span class="macro">assert_eq!</span>(name_hash.hash, p.hash, <span class="string">&quot;user password matches&quot;</span>);
<span class="macro">assert_eq!</span>(name_hash.username, p.username, <span class="string">&quot;username matches&quot;</span>);
<span class="comment">// testing get_email
</span><span class="macro">assert_eq!</span>(db.get_email(p.username).<span class="kw">await</span>.unwrap(), p.email);
<span class="comment">// testing email exists
</span><span class="macro">assert!</span>(
db.email_exists(p.email).<span class="kw">await</span>.unwrap(),
<span class="string">&quot;user is registered so email should exist&quot;
</span>);
<span class="macro">assert!</span>(
db.username_exists(p.username).<span class="kw">await</span>.unwrap(),
<span class="string">&quot;user is registered so username should exist&quot;
</span>);
<span class="comment">// update password test. setting password = username
</span>name_hash.hash = name_hash.username.clone();
db.update_password(<span class="kw-2">&amp;</span>name_hash).<span class="kw">await</span>.unwrap();
<span class="kw">let </span>name_hash = db.get_password(<span class="kw-2">&amp;</span>Login::Username(p.username)).<span class="kw">await</span>.unwrap();
<span class="macro">assert_eq!</span>(
name_hash.hash, p.username,
<span class="string">&quot;user password matches with changed value&quot;
</span>);
<span class="macro">assert_eq!</span>(name_hash.username, p.username, <span class="string">&quot;username matches&quot;</span>);
<span class="comment">// update username to p.email
</span><span class="macro">assert!</span>(
!db.username_exists(p.email).<span class="kw">await</span>.unwrap(),
<span class="string">&quot;user with p.email doesn&#39;t exist. pre-check to update username to p.email&quot;
</span>);
db.update_username(p.username, p.email).<span class="kw">await</span>.unwrap();
<span class="macro">assert!</span>(
db.username_exists(p.email).<span class="kw">await</span>.unwrap(),
<span class="string">&quot;user with p.email exist post-update&quot;
</span>);
<span class="comment">// testing update email
</span><span class="kw">let </span>update_email = UpdateEmail {
username: p.username,
new_email: EMAIL2,
};
db.update_email(<span class="kw-2">&amp;</span>update_email).<span class="kw">await</span>.unwrap();
<span class="macro">println!</span>(
<span class="string">&quot;null user email: {}&quot;</span>,
db.email_exists(p.email).<span class="kw">await</span>.unwrap()
);
<span class="macro">assert!</span>(
db.email_exists(p.email).<span class="kw">await</span>.unwrap(),
<span class="string">&quot;user was with empty email but email is set; so email should exist&quot;
</span>);
<span class="comment">// deleting user
</span>db.delete_user(p.email).<span class="kw">await</span>.unwrap();
<span class="macro">assert!</span>(
!db.username_exists(p.email).<span class="kw">await</span>.unwrap(),
<span class="string">&quot;user is deleted so username shouldn&#39;t exist&quot;
</span>);
}
<span class="attribute">#[actix_rt::test]
</span><span class="kw">pub async fn </span>test_db_sites() {
<span class="kw">let </span>settings = Settings::new().unwrap();
<span class="kw">let </span>pool_options = PgPoolOptions::new().max_connections(<span class="number">1</span>);
<span class="kw">let </span>db = ConnectionOptions::Fresh(Fresh {
pool_options,
url: settings.database.url.clone(),
disable_logging: !settings.debug,
})
.connect()
.<span class="kw">await
</span>.unwrap();
<span class="macro">assert!</span>(db.ping().<span class="kw">await</span>);
<span class="kw">const </span>EMAIL: <span class="kw-2">&amp;</span>str = <span class="string">&quot;postgresdbsiteuser@foo.com&quot;</span>;
<span class="kw">const </span>NAME: <span class="kw-2">&amp;</span>str = <span class="string">&quot;postgresdbsiteuser&quot;</span>;
<span class="kw">const </span>PASSWORD: <span class="kw-2">&amp;</span>str = <span class="string">&quot;pasdfasdfasdfadf&quot;</span>;
db.migrate().<span class="kw">await</span>.unwrap();
<span class="kw">let </span>p = <span class="kw">super</span>::Register {
username: NAME,
email: EMAIL,
hash: PASSWORD,
};
<span class="kw">if </span>db.username_exists(p.username).<span class="kw">await</span>.unwrap() {
db.delete_user(p.username).<span class="kw">await</span>.unwrap();
<span class="macro">assert!</span>(
!db.username_exists(p.username).<span class="kw">await</span>.unwrap(),
<span class="string">&quot;user is deleted so username shouldn&#39;t exist&quot;
</span>);
}
db.register(<span class="kw-2">&amp;</span>p).<span class="kw">await</span>.unwrap();
<span class="kw">let </span>site = Site {
site_secret: <span class="string">&quot;foobar&quot;</span>.into(),
repo_url: <span class="string">&quot;https://git.batsense.net/LibrePages/librepages.git&quot;</span>.into(),
branch: <span class="string">&quot;librepages&quot;</span>.into(),
hostname: <span class="string">&quot;db_works.tests.librepages.librepages.org&quot;</span>.into(),
owner: p.username.into(),
};
<span class="comment">// test if hostname exists. Should be false
</span><span class="macro">assert!</span>(!db.hostname_exists(<span class="kw-2">&amp;</span>site.hostname).<span class="kw">await</span>.unwrap());
<span class="comment">// testing adding site
</span>db.add_site(<span class="kw-2">&amp;</span>site).<span class="kw">await</span>.unwrap();
<span class="comment">// test if hostname exists. Should be true
</span><span class="macro">assert!</span>(db.hostname_exists(<span class="kw-2">&amp;</span>site.hostname).<span class="kw">await</span>.unwrap());
<span class="comment">// get site
</span><span class="kw">let </span>db_site = db.get_site(p.username, <span class="kw-2">&amp;</span>site.hostname).<span class="kw">await</span>.unwrap();
<span class="macro">assert_eq!</span>(db_site, site);
<span class="comment">// get site by secret
</span><span class="macro">assert_eq!</span>(
db_site,
db.get_site_from_secret(<span class="kw-2">&amp;</span>site.site_secret).<span class="kw">await</span>.unwrap()
);
<span class="comment">// list all sites owned by user
</span><span class="kw">let </span>db_sites = db.list_all_sites(p.username).<span class="kw">await</span>.unwrap();
<span class="macro">assert_eq!</span>(db_sites.len(), <span class="number">1</span>);
<span class="macro">assert_eq!</span>(db_sites, <span class="macro">vec!</span>[site.clone()]);
<span class="comment">// delete site
</span>db.delete_site(p.username, <span class="kw-2">&amp;</span>site.hostname).<span class="kw">await</span>.unwrap();
<span class="comment">// test if hostname exists. Should be false
</span><span class="macro">assert!</span>(!db.hostname_exists(<span class="kw-2">&amp;</span>site.hostname).<span class="kw">await</span>.unwrap());
}
}
</code></pre></div>
</section></div></main><div id="rustdoc-vars" data-root-path="../../" data-current-crate="pages" data-themes="ayu,dark,light" data-resource-suffix="" data-rustdoc-version="1.65.0 (897e37553 2022-11-02)" ></div></body></html>