set-parser.js 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173
  1. /**
  2. * 语句:set 设置变量 + 变量与值解析(原 value-resolver:extractVarName、replaceVariablesInString、resolveValue、parseValue、时间/延迟)
  3. */
  4. const types = ['set']
  5. function extractVarName(varName) {
  6. if (typeof varName === 'string' && varName.startsWith('{') && varName.endsWith('}')) {
  7. return varName.slice(1, -1)
  8. }
  9. return varName
  10. }
  11. function replaceVariablesInString(str, context) {
  12. if (typeof str !== 'string') return str
  13. const replacer = (match, varName) => {
  14. const varValue = context[varName]
  15. if (varValue === undefined || varValue === null || varValue === '') return ''
  16. if (varValue === 'undefined' || varValue === 'null') return ''
  17. if (typeof varValue === 'string') {
  18. try {
  19. const parsed = JSON.parse(varValue)
  20. if (Array.isArray(parsed)) return parsed.length === 0 ? '' : varValue
  21. } catch (e) {}
  22. }
  23. if (Array.isArray(varValue) || (typeof varValue === 'object' && varValue !== null)) {
  24. try {
  25. return JSON.stringify(varValue)
  26. } catch (e) {
  27. return String(varValue)
  28. }
  29. }
  30. return String(varValue)
  31. }
  32. const doubleBracePattern = /\{\{([\w-]+)\}\}/g
  33. const singleBracePattern = /\{([\w-]+)\}/g
  34. return str.replace(doubleBracePattern, replacer).replace(singleBracePattern, replacer)
  35. }
  36. function resolveValue(value, context) {
  37. if (typeof value === 'string' && value.startsWith('{') && value.endsWith('}')) {
  38. const varName = value.slice(1, -1)
  39. return context[varName] !== undefined ? context[varName] : value
  40. }
  41. if (Array.isArray(value)) {
  42. return value.map(item => resolveValue(item, context))
  43. }
  44. if (typeof value === 'object' && value !== null) {
  45. const resolved = {}
  46. for (const key in value) {
  47. resolved[key] = resolveValue(value[key], context)
  48. }
  49. return resolved
  50. }
  51. return value
  52. }
  53. function parseTimeString(timeStr) {
  54. if (!timeStr || typeof timeStr !== 'string' || timeStr.trim() === '') return null
  55. try {
  56. const parts = timeStr.trim().split(' ')
  57. if (parts.length !== 2) return null
  58. const datePart = parts[0].split('/')
  59. const timePart = parts[1].split(':')
  60. if (datePart.length !== 3 || timePart.length !== 2) return null
  61. const date = new Date(
  62. parseInt(datePart[0], 10),
  63. parseInt(datePart[1], 10) - 1,
  64. parseInt(datePart[2], 10),
  65. parseInt(timePart[0], 10),
  66. parseInt(timePart[1], 10),
  67. 0, 0
  68. )
  69. return isNaN(date.getTime()) ? null : date
  70. } catch (e) {
  71. return null
  72. }
  73. }
  74. const DELAY_UNIT_MS = {
  75. ms: 1,
  76. s: 1000,
  77. m: 60 * 1000,
  78. h: 60 * 60 * 1000,
  79. d: 24 * 60 * 60 * 1000,
  80. w: 7 * 24 * 60 * 60 * 1000,
  81. '月': 30 * 24 * 60 * 60 * 1000,
  82. y: 365 * 24 * 60 * 60 * 1000,
  83. }
  84. /** 解析时长字符串,支持单位 ms/s/m/h/d/w/月,支持组合如 1h3m48s、2d30m */
  85. function parseDelayString(delayStr) {
  86. if (!delayStr || typeof delayStr !== 'string' || delayStr.trim() === '') return 0
  87. try {
  88. const trimmed = delayStr.trim()
  89. const re = /(\d+)\s*(ms|s|m|h|d|w|月|y)/g
  90. let total = 0
  91. let hadMatch = false
  92. let match
  93. while ((match = re.exec(trimmed)) !== null) {
  94. const value = parseInt(match[1], 10)
  95. const unit = match[2]
  96. if (value < 0 || !DELAY_UNIT_MS[unit]) continue
  97. total += value * DELAY_UNIT_MS[unit]
  98. hadMatch = true
  99. }
  100. return hadMatch ? total : null
  101. } catch (e) {
  102. return null
  103. }
  104. }
  105. function calculateWaitTime(data, delay) {
  106. return 0
  107. }
  108. function parseValue(str) {
  109. if (typeof str !== 'string') return str
  110. str = str.trim()
  111. if (str === 'true') return true
  112. if (str === 'false') return false
  113. if ((str.startsWith('"') && str.endsWith('"')) || (str.startsWith("'") && str.endsWith("'"))) {
  114. return str.slice(1, -1).replace(/\\"/g, '"').replace(/\\'/g, "'")
  115. }
  116. if (/^-?\d+(\.\d+)?$/.test(str)) return parseFloat(str)
  117. if (str === 'undefined') return undefined
  118. return str
  119. }
  120. function parse(action, parseContext) {
  121. const variableContext = parseContext.variableContext || {}
  122. const parsed = {
  123. type: 'set',
  124. variable: action.variable,
  125. value: resolveValue(action.value, variableContext),
  126. }
  127. return Object.assign({}, action, parsed)
  128. }
  129. async function execute(action, ctx) {
  130. const { variableContext, evaluateExpression } = ctx
  131. if (!action.variable) return { success: true }
  132. const varName = extractVarName(action.variable)
  133. let finalValue = action.value
  134. if (typeof action.value === 'string' && /[+\-*/]/.test(action.value)) {
  135. finalValue = evaluateExpression(action.value, variableContext)
  136. } else if (typeof action.value === 'string') {
  137. const varPattern = /\{([\w-]+)\}/g
  138. let hasVariables = false
  139. finalValue = action.value.replace(varPattern, (match, vn) => {
  140. hasVariables = true
  141. const val = variableContext[vn]
  142. return val === undefined || val === null ? match : String(val)
  143. })
  144. if (!hasVariables) finalValue = resolveValue(action.value, variableContext)
  145. } else {
  146. finalValue = resolveValue(action.value, variableContext)
  147. }
  148. variableContext[varName] = finalValue
  149. return { success: true }
  150. }
  151. module.exports = {
  152. types,
  153. parse,
  154. execute,
  155. extractVarName,
  156. replaceVariablesInString,
  157. resolveValue,
  158. parseTimeString,
  159. parseDelayString,
  160. calculateWaitTime,
  161. parseValue,
  162. }