Initial commit

src/util/JobState.ts

@@ -0,0 +1,104 @@
+import fs from 'fs'
+import path from 'path'
+import type { Config } from '../interface/Config'
+
+type AccountCompletionMeta = {
+  runId?: string
+  completedAt: string
+  totalCollected?: number
+  banned?: boolean
+  errors?: number
+}
+
+type DayState = {
+  doneOfferIds: string[]
+  accountCompleted?: boolean
+  accountMeta?: AccountCompletionMeta
+}
+
+type FileState = {
+  days: Record<string, DayState>
+}
+
+export class JobState {
+  private baseDir: string
+
+  constructor(cfg: Config) {
+    const dir = cfg.jobState?.dir || path.join(process.cwd(), cfg.sessionPath, 'job-state')
+    this.baseDir = dir
+    if (!fs.existsSync(this.baseDir)) fs.mkdirSync(this.baseDir, { recursive: true })
+  }
+
+  private fileFor(email: string): string {
+    const safe = email.replace(/[^a-z0-9._-]/gi, '_')
+    return path.join(this.baseDir, `${safe}.json`)
+  }
+
+  private load(email: string): FileState {
+    const file = this.fileFor(email)
+    if (!fs.existsSync(file)) return { days: {} }
+    try {
+      const raw = fs.readFileSync(file, 'utf-8')
+      const parsed = JSON.parse(raw)
+      return parsed && typeof parsed === 'object' && parsed.days ? parsed as FileState : { days: {} }
+    } catch { return { days: {} } }
+  }
+
+  private save(email: string, state: FileState): void {
+    const file = this.fileFor(email)
+    fs.writeFileSync(file, JSON.stringify(state, null, 2), 'utf-8')
+  }
+
+  isDone(email: string, day: string, offerId: string): boolean {
+    const st = this.load(email)
+    const d = st.days[day]
+    if (!d) return false
+    return d.doneOfferIds.includes(offerId)
+  }
+
+  markDone(email: string, day: string, offerId: string): void {
+    const st = this.load(email)
+    if (!st.days[day]) st.days[day] = { doneOfferIds: [] }
+    const d = st.days[day]
+    if (!d.doneOfferIds.includes(offerId)) d.doneOfferIds.push(offerId)
+    this.save(email, st)
+  }
+
+  isAccountComplete(email: string, day: string): boolean {
+    const st = this.load(email)
+    const d = st.days[day]
+    return d?.accountCompleted === true
+  }
+
+  markAccountComplete(
+    email: string,
+    day: string,
+    meta?: { runId?: string; totalCollected?: number; banned?: boolean; errors?: number }
+  ): void {
+    const st = this.load(email)
+    if (!st.days[day]) st.days[day] = { doneOfferIds: [] }
+    const d = st.days[day]
+    d.accountCompleted = true
+    d.accountMeta = {
+      completedAt: new Date().toISOString(),
+      runId: meta?.runId,
+      totalCollected: meta?.totalCollected,
+      banned: meta?.banned ?? false,
+      errors: meta?.errors ?? 0
+    }
+    this.save(email, st)
+  }
+
+  clearAccountComplete(email: string, day: string): void {
+    const st = this.load(email)
+    const d = st.days[day]
+    if (!d) return
+    if (d.accountCompleted || d.accountMeta) {
+      delete d.accountCompleted
+      delete d.accountMeta
+      this.save(email, st)
+    }
+  }
+}
+
+export default JobState