major overhaul of schedule parsing code
This commit is contained in:
		@@ -1,10 +1,99 @@
 | 
			
		||||
import { add } from "date-fns"
 | 
			
		||||
import { DateResolvable } from "discord.js"
 | 
			
		||||
import { CustomError, errorCodes } from "../interfaces"
 | 
			
		||||
 | 
			
		||||
export interface RepetitonInfo {
 | 
			
		||||
	startDate?: Date, // If defined will take precedence over repetitonAmount
 | 
			
		||||
	endDate?: Date,// If defined will take precedence over repetitonAmount
 | 
			
		||||
	totalAmount: number,
 | 
			
		||||
	alreadyOccured: number,
 | 
			
		||||
	schedule: supportedSchedule
 | 
			
		||||
	schedule: Schedule
 | 
			
		||||
}
 | 
			
		||||
export const scheduleNames = ['daily', 'weekly', 'monthly', 'everyTwoWeeks', 'everyNDays']
 | 
			
		||||
export const scheduleNames = ['daily', 'weekly', 'monthly', 'everyNWeeks', 'everyNDays', 'everyNMonths']
 | 
			
		||||
export type supportedSchedule = typeof scheduleNames[number]
 | 
			
		||||
export interface IScheduleType {
 | 
			
		||||
	name: supportedSchedule,
 | 
			
		||||
	multiplier: number,
 | 
			
		||||
	duration: Duration
 | 
			
		||||
}
 | 
			
		||||
export const scheduleTypes: IScheduleType[] = [
 | 
			
		||||
	{
 | 
			
		||||
		name: 'daily',
 | 
			
		||||
		multiplier: 1,
 | 
			
		||||
		duration: {
 | 
			
		||||
			days: 1
 | 
			
		||||
		}
 | 
			
		||||
	},
 | 
			
		||||
	{
 | 
			
		||||
		name: 'weekly',
 | 
			
		||||
		multiplier: 1,
 | 
			
		||||
		duration: {
 | 
			
		||||
			weeks: 1
 | 
			
		||||
		}
 | 
			
		||||
	},
 | 
			
		||||
]
 | 
			
		||||
export class Schedule {
 | 
			
		||||
	private scheduleName: string
 | 
			
		||||
	private multiplier = 1
 | 
			
		||||
	private duration: Duration
 | 
			
		||||
	private baseScheduleTypes = ['daily', 'weekly', 'monthly', 'yearly']
 | 
			
		||||
	private _scheduleString: string
 | 
			
		||||
	constructor(scheduleString: string) {
 | 
			
		||||
		this._scheduleString = scheduleString.toLowerCase()
 | 
			
		||||
		this.scheduleName = this._scheduleString
 | 
			
		||||
		if (this.baseScheduleTypes.includes(this._scheduleString)) {
 | 
			
		||||
			this.multiplier = 1
 | 
			
		||||
		}
 | 
			
		||||
		if (this._scheduleString.includes('every')) {
 | 
			
		||||
			this.scheduleName = this.getBaseScheduleNameFromVariableString()
 | 
			
		||||
			this.multiplier = this.getMultiplierFromVariableString()
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		switch (this.scheduleName) {
 | 
			
		||||
			case 'daily':
 | 
			
		||||
				this.duration = { days: 1 }
 | 
			
		||||
				break
 | 
			
		||||
			case 'weekly':
 | 
			
		||||
				this.duration = { weeks: 1 }
 | 
			
		||||
				break
 | 
			
		||||
			case 'monthly':
 | 
			
		||||
				this.duration = { months: 1 }
 | 
			
		||||
				break
 | 
			
		||||
			case 'yearly':
 | 
			
		||||
				this.duration = { years: 1 }
 | 
			
		||||
				break
 | 
			
		||||
			default:
 | 
			
		||||
				throw new CustomError('Schedule type not supported', errorCodes.schedule_not_supported)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
	private getBaseScheduleNameFromVariableString(): string {
 | 
			
		||||
		if (this._scheduleString.includes('week')) return 'weekly'
 | 
			
		||||
		if (this._scheduleString.includes('day')) return 'daily'
 | 
			
		||||
		if (this._scheduleString.includes('month')) return 'monthly'
 | 
			
		||||
		if (this._scheduleString.includes('year')) return 'yearly'
 | 
			
		||||
		return ''
 | 
			
		||||
	}
 | 
			
		||||
	public getMultiplierFromVariableString(): number {
 | 
			
		||||
		const matches = this._scheduleString.match(/\d+/)
 | 
			
		||||
		if (matches) {
 | 
			
		||||
			const multi = matches[0]
 | 
			
		||||
			if (multi)
 | 
			
		||||
				return parseInt(multi)
 | 
			
		||||
		}
 | 
			
		||||
		return 1
 | 
			
		||||
	}
 | 
			
		||||
	public calculateDuration(): Duration {
 | 
			
		||||
		const dur: Duration = {
 | 
			
		||||
			days: this.duration.days ? this.duration.days * this.multiplier : undefined,
 | 
			
		||||
			weeks: this.duration.weeks ? this.duration.weeks * this.multiplier : undefined,
 | 
			
		||||
			months: this.duration.months ? this.duration.months * this.multiplier : undefined,
 | 
			
		||||
			years: this.duration.years ? this.duration.years * this.multiplier : undefined,
 | 
			
		||||
		}
 | 
			
		||||
		return dur
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	public getNewDate(oldDate: Date): Date {
 | 
			
		||||
		const newDate = add(oldDate, this.calculateDuration())
 | 
			
		||||
		return newDate
 | 
			
		||||
	}
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user