set-parser.js 5.1 KB

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