import { ScheduledJobOwnerType } from '@n8n/constants';
import type { ScheduledJobOwner, ScheduledJobOwnerRef } from '@n8n/db';
import { WorkflowPublishedVersionRepository } from '@n8n/db';
import { Service } from '@n8n/di';

import type { ScheduledJobOwnerResolver } from '@n8n/scheduler';

/**
 * Marks a workflow as the owner of the scheduled jobs its trigger nodes create.
 * For reconciliation, a workflow still exists while it has a published version.
 */
@Service()
export class WorkflowScheduledJobOwner implements ScheduledJobOwnerResolver {
	readonly ownerType = ScheduledJobOwnerType.Workflow;

	constructor(private readonly publishedVersions: WorkflowPublishedVersionRepository) {}

	/** The owner of the jobs one trigger node provisions. */
	member(workflowId: string, nodeId: string): ScheduledJobOwner {
		return { ownerType: this.ownerType, ownerId: workflowId, ownerMemberId: nodeId };
	}

	/** The owner of every job the workflow holds, whichever node provisioned it. */
	ref(workflowId: string): ScheduledJobOwnerRef {
		return { ownerType: this.ownerType, ownerId: workflowId };
	}

	async findExisting(ownerIds: string[]): Promise<Set<string>> {
		return await this.publishedVersions.findPublishedWorkflowIds(ownerIds);
	}
}