dex-sdk 0.2.2

// Portions of this file are derived from indeedeng/iwf-java-sdk.
// Those portions are licensed under the Apache License, Version 2.0.
// See LICENSES/Apache-2.0.txt and LEGACY_NOTICES.md.
//
// Modifications Copyright (c) 2026 Super Durable, Inc.
//
// Modifications are licensed under the Super Durable Source License 1.0.
// Third-Party Materials remain under the Apache License, Version 2.0.
// See LICENSE and LEGACY_NOTICES.md.

use dex_sdk::{
    Context, Flow, HandlerResult, PersistenceSchema, Step, StepDecision, StepList, Stream,
};

#[derive(Clone)]
pub(crate) struct StreamTestWorkflow {
    pub(crate) progress: Stream<String>,
    start: StreamTestStep,
}

impl StreamTestWorkflow {
    pub(crate) fn new() -> Self {
        let progress = Stream::new("stream-test-progress", 1 << 20);
        Self {
            progress: progress.clone(),
            start: StreamTestStep { progress },
        }
    }
}

impl Flow for StreamTestWorkflow {
    type StartInput = ();

    fn steps(&self) -> StepList<'_, Self::StartInput> {
        StepList::start(&self.start)
    }

    fn persistence(&self) -> PersistenceSchema {
        PersistenceSchema::new().stream(&self.progress)
    }
}

#[derive(Clone)]
struct StreamTestStep {
    progress: Stream<String>,
}

impl Step for StreamTestStep {
    type Input = ();

    fn execute(&self, context: &mut Context, _input: ()) -> HandlerResult<StepDecision> {
        self.progress.write(context, "step-progress".to_string())?;
        Ok(StepDecision::graceful_complete(()))
    }
}