| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280 |
- <!doctype html>
- <html lang="en">
- <head>
- <title>Code coverage report for middleware/usageLimit.ts</title>
- <meta charset="utf-8" />
- <link rel="stylesheet" href="../prettify.css" />
- <link rel="stylesheet" href="../base.css" />
- <link rel="shortcut icon" type="image/x-icon" href="../favicon.png" />
- <meta name="viewport" content="width=device-width, initial-scale=1" />
- <style type='text/css'>
- .coverage-summary .sorter {
- background-image: url(../sort-arrow-sprite.png);
- }
- </style>
- </head>
-
- <body>
- <div class='wrapper'>
- <div class='pad1'>
- <h1><a href="../index.html">All files</a> / <a href="index.html">middleware</a> usageLimit.ts</h1>
- <div class='clearfix'>
-
- <div class='fl pad1y space-right2'>
- <span class="strong">0% </span>
- <span class="quiet">Statements</span>
- <span class='fraction'>0/45</span>
- </div>
-
-
- <div class='fl pad1y space-right2'>
- <span class="strong">0% </span>
- <span class="quiet">Branches</span>
- <span class='fraction'>0/1</span>
- </div>
-
-
- <div class='fl pad1y space-right2'>
- <span class="strong">0% </span>
- <span class="quiet">Functions</span>
- <span class='fraction'>0/1</span>
- </div>
-
-
- <div class='fl pad1y space-right2'>
- <span class="strong">0% </span>
- <span class="quiet">Lines</span>
- <span class='fraction'>0/45</span>
- </div>
-
-
- </div>
- <p class="quiet">
- Press <em>n</em> or <em>j</em> to go to the next uncovered block, <em>b</em>, <em>p</em> or <em>k</em> for the previous block.
- </p>
- <template id="filterTemplate">
- <div class="quiet">
- Filter:
- <input type="search" id="fileSearch">
- </div>
- </template>
- </div>
- <div class='status-line low'></div>
- <pre><table class="coverage">
- <tr><td class="line-count quiet"><a name='L1'></a><a href='#L1'>1</a>
- <a name='L2'></a><a href='#L2'>2</a>
- <a name='L3'></a><a href='#L3'>3</a>
- <a name='L4'></a><a href='#L4'>4</a>
- <a name='L5'></a><a href='#L5'>5</a>
- <a name='L6'></a><a href='#L6'>6</a>
- <a name='L7'></a><a href='#L7'>7</a>
- <a name='L8'></a><a href='#L8'>8</a>
- <a name='L9'></a><a href='#L9'>9</a>
- <a name='L10'></a><a href='#L10'>10</a>
- <a name='L11'></a><a href='#L11'>11</a>
- <a name='L12'></a><a href='#L12'>12</a>
- <a name='L13'></a><a href='#L13'>13</a>
- <a name='L14'></a><a href='#L14'>14</a>
- <a name='L15'></a><a href='#L15'>15</a>
- <a name='L16'></a><a href='#L16'>16</a>
- <a name='L17'></a><a href='#L17'>17</a>
- <a name='L18'></a><a href='#L18'>18</a>
- <a name='L19'></a><a href='#L19'>19</a>
- <a name='L20'></a><a href='#L20'>20</a>
- <a name='L21'></a><a href='#L21'>21</a>
- <a name='L22'></a><a href='#L22'>22</a>
- <a name='L23'></a><a href='#L23'>23</a>
- <a name='L24'></a><a href='#L24'>24</a>
- <a name='L25'></a><a href='#L25'>25</a>
- <a name='L26'></a><a href='#L26'>26</a>
- <a name='L27'></a><a href='#L27'>27</a>
- <a name='L28'></a><a href='#L28'>28</a>
- <a name='L29'></a><a href='#L29'>29</a>
- <a name='L30'></a><a href='#L30'>30</a>
- <a name='L31'></a><a href='#L31'>31</a>
- <a name='L32'></a><a href='#L32'>32</a>
- <a name='L33'></a><a href='#L33'>33</a>
- <a name='L34'></a><a href='#L34'>34</a>
- <a name='L35'></a><a href='#L35'>35</a>
- <a name='L36'></a><a href='#L36'>36</a>
- <a name='L37'></a><a href='#L37'>37</a>
- <a name='L38'></a><a href='#L38'>38</a>
- <a name='L39'></a><a href='#L39'>39</a>
- <a name='L40'></a><a href='#L40'>40</a>
- <a name='L41'></a><a href='#L41'>41</a>
- <a name='L42'></a><a href='#L42'>42</a>
- <a name='L43'></a><a href='#L43'>43</a>
- <a name='L44'></a><a href='#L44'>44</a>
- <a name='L45'></a><a href='#L45'>45</a>
- <a name='L46'></a><a href='#L46'>46</a>
- <a name='L47'></a><a href='#L47'>47</a>
- <a name='L48'></a><a href='#L48'>48</a>
- <a name='L49'></a><a href='#L49'>49</a>
- <a name='L50'></a><a href='#L50'>50</a>
- <a name='L51'></a><a href='#L51'>51</a>
- <a name='L52'></a><a href='#L52'>52</a>
- <a name='L53'></a><a href='#L53'>53</a>
- <a name='L54'></a><a href='#L54'>54</a>
- <a name='L55'></a><a href='#L55'>55</a>
- <a name='L56'></a><a href='#L56'>56</a>
- <a name='L57'></a><a href='#L57'>57</a>
- <a name='L58'></a><a href='#L58'>58</a>
- <a name='L59'></a><a href='#L59'>59</a>
- <a name='L60'></a><a href='#L60'>60</a>
- <a name='L61'></a><a href='#L61'>61</a>
- <a name='L62'></a><a href='#L62'>62</a>
- <a name='L63'></a><a href='#L63'>63</a>
- <a name='L64'></a><a href='#L64'>64</a>
- <a name='L65'></a><a href='#L65'>65</a>
- <a name='L66'></a><a href='#L66'>66</a></td><td class="line-coverage quiet"><span class="cline-any cline-no"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-neutral"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span>
- <span class="cline-any cline-no"> </span></td><td class="text"><pre class="prettyprint lang-js"><span class="cstat-no" title="statement not covered" ><span class="fstat-no" title="function not covered" ><span class="branch-0 cbranch-no" title="branch not covered" >import { Context, Next } from 'koa';</span></span></span>
- import { prisma } from '../models';
- import { QuotaExceededError, ForbiddenError } from './errorHandler';
- import { MEMBER_QUOTA, MemberLevel } from '../types';
-
- // 检查使用次数限制
- <span class="cstat-no" title="statement not covered" >export async function usageLimitMiddleware(ctx: Context, next: Next): Promise<void> {</span>
- <span class="cstat-no" title="statement not covered" > const userId = ctx.state.user?.userId;</span>
-
- // 测试模式:无需登录即可使用
- <span class="cstat-no" title="statement not covered" > if (!userId) {</span>
- <span class="cstat-no" title="statement not covered" > ctx.state.userQuota = { dailyLimit: -1, wordLimit: -1 };</span>
- <span class="cstat-no" title="statement not covered" > await next();</span>
- <span class="cstat-no" title="statement not covered" > return;</span>
- <span class="cstat-no" title="statement not covered" > }</span>
-
- <span class="cstat-no" title="statement not covered" > const user = await prisma.user.findUnique({ where: { id: parseInt(userId) } });</span>
- <span class="cstat-no" title="statement not covered" > if (!user) {</span>
- <span class="cstat-no" title="statement not covered" > throw new ForbiddenError('用户不存在');</span>
- <span class="cstat-no" title="statement not covered" > }</span>
-
- <span class="cstat-no" title="statement not covered" > const today = new Date().toISOString().slice(0, 10);</span>
- <span class="cstat-no" title="statement not covered" > const memberLevel = user.memberLevel;</span>
- <span class="cstat-no" title="statement not covered" > const quota = MEMBER_QUOTA[memberLevel as MemberLevel];</span>
-
- // 重置每日使用次数
- <span class="cstat-no" title="statement not covered" > if (user.lastUsageDate !== today) {</span>
- <span class="cstat-no" title="statement not covered" > await prisma.user.update({</span>
- <span class="cstat-no" title="statement not covered" > where: { id: parseInt(userId) },</span>
- <span class="cstat-no" title="statement not covered" > data: {</span>
- <span class="cstat-no" title="statement not covered" > dailyUsage: 0,</span>
- <span class="cstat-no" title="statement not covered" > lastUsageDate: today,</span>
- <span class="cstat-no" title="statement not covered" > },</span>
- <span class="cstat-no" title="statement not covered" > });</span>
- <span class="cstat-no" title="statement not covered" > user.dailyUsage = 0;</span>
- <span class="cstat-no" title="statement not covered" > user.lastUsageDate = today;</span>
- <span class="cstat-no" title="statement not covered" > }</span>
-
- // 检查次数限制 (-1 表示无限制)
- <span class="cstat-no" title="statement not covered" > if (quota.dailyLimit !== -1 && user.dailyUsage >= quota.dailyLimit) {</span>
- <span class="cstat-no" title="statement not covered" > throw new QuotaExceededError(`今日使用次数已达上限(${quota.dailyLimit}次),请升级会员`);</span>
- <span class="cstat-no" title="statement not covered" > }</span>
-
- // 将用户信息和配额存入 state
- <span class="cstat-no" title="statement not covered" > ctx.state.userQuota = quota;</span>
- <span class="cstat-no" title="statement not covered" > ctx.state.userInfo = user;</span>
-
- <span class="cstat-no" title="statement not covered" > await next();</span>
- <span class="cstat-no" title="statement not covered" >}</span>
-
- // 检查字数限制
- <span class="cstat-no" title="statement not covered" >export function checkWordLimit(wordCount: number): (ctx: Context, next: Next) => Promise<void> {</span>
- <span class="cstat-no" title="statement not covered" > return async (ctx: Context, next: Next) => {</span>
- <span class="cstat-no" title="statement not covered" > const quota = ctx.state.userQuota;</span>
- <span class="cstat-no" title="statement not covered" > if (!quota) {</span>
- <span class="cstat-no" title="statement not covered" > throw new ForbiddenError('请先检查使用限制');</span>
- <span class="cstat-no" title="statement not covered" > }</span>
-
- // -1 表示无限制
- <span class="cstat-no" title="statement not covered" > if (quota.wordLimit !== -1 && wordCount > quota.wordLimit) {</span>
- <span class="cstat-no" title="statement not covered" > throw new QuotaExceededError(`文本字数超出限制(${quota.wordLimit}字),请升级会员或缩短文本`);</span>
- <span class="cstat-no" title="statement not covered" > }</span>
-
- <span class="cstat-no" title="statement not covered" > await next();</span>
- <span class="cstat-no" title="statement not covered" > };</span>
- <span class="cstat-no" title="statement not covered" >}</span></pre></td></tr></table></pre>
- <div class='push'></div><!-- for sticky footer -->
- </div><!-- /wrapper -->
- <div class='footer quiet pad2 space-top1 center small'>
- Code coverage generated by
- <a href="https://istanbul.js.org/" target="_blank" rel="noopener noreferrer">istanbul</a>
- at 2026-05-19T14:27:48.313Z
- </div>
- <script src="../prettify.js"></script>
- <script>
- window.onload = function () {
- prettyPrint();
- };
- </script>
- <script src="../sorter.js"></script>
- <script src="../block-navigation.js"></script>
- </body>
- </html>
-
|