activity diagram serializer

Author: invincibleXiaobao

plantuml/parser/puml_cli/src/main.rs

@@ -23,13 +23,16 @@ use std::rc::Rc;
 
 use puml_lobster::{write_lobster_to_file, LobsterModel};
 use puml_parser::{
-    DiagramParser, ErrorLocation, Preprocessor, PumlClassParser, PumlComponentParser,
-    PumlSequenceParser,
+    DiagramParser, ErrorLocation, Preprocessor, PumlActivityParser, PumlClassParser,
+    PumlComponentParser, PumlSequenceParser,
 };
 use puml_resolver::{
-    ClassResolver, DiagramResolver, ElementResolver, LogicElement, SequenceResolver, SequenceTree,
+    ActivityDiagram, ActivityResolver, ClassResolver, DiagramResolver, ElementResolver,
+    LogicElement, SequenceResolver, SequenceTree,
+};
+use puml_serializer::{
+    ActivitySerializer, ClassSerializer, ComponentSerializer, SequenceSerializer,
 };
-use puml_serializer::{ClassSerializer, ComponentSerializer, SequenceSerializer};
 use puml_utils::{write_fbs_to_file, write_json_to_file, LogLevel};
 
 /// CLI wrapper for LogLevel that implements ValueEnum
@@ -99,6 +102,7 @@ struct Args {
 #[derive(Copy, Clone, ValueEnum, Debug)]
 enum DiagramType {
     None,
+    Activity,
     Component,
     Deployment,
     Class,
@@ -108,6 +112,7 @@ enum DiagramType {
 #[allow(dead_code)] // Class and Sequence variants are WIP
 #[derive(Debug, Serialize)]
 enum ParsedDiagram {
+    Activity(puml_parser::RawActivityDiagram),
     Component(puml_parser::CompPumlDocument),
     Class(puml_parser::ClassUmlFile),
     Sequence(puml_parser::SeqPumlDocument),
@@ -191,6 +196,7 @@ fn run() -> Result<(), Box<dyn std::error::Error>> {
                     let lobster_model = match &logic_result {
                         ResolvedDiagram::Component(model) => LobsterModel::Component(model),
                         ResolvedDiagram::Class(model) => LobsterModel::Class(model),
+                        ResolvedDiagram::Activity(_) => LobsterModel::Empty,
                         ResolvedDiagram::Sequence(_) => LobsterModel::Empty,
                     };
                     write_lobster_to_file(lobster_model, path, ldir)?;
@@ -208,6 +214,9 @@ fn run() -> Result<(), Box<dyn std::error::Error>> {
 
 fn serialize_resolved_diagram(resolved_content: &ResolvedDiagram, source_file: &str) -> Vec<u8> {
     match resolved_content {
+        ResolvedDiagram::Activity(resolved_content) => {
+            ActivitySerializer::serialize(resolved_content, source_file)
+        }
         ResolvedDiagram::Component(resolved_content) => {
             ComponentSerializer::serialize(resolved_content, source_file)
         }
@@ -222,6 +231,7 @@ fn serialize_resolved_diagram(resolved_content: &ResolvedDiagram, source_file: &
 
 #[derive(Debug, Serialize)]
 pub enum ResolvedDiagram {
+    Activity(ActivityDiagram),
     Component(HashMap<String, LogicElement>),
     Class(class_diagram::ClassDiagram),
     Sequence(SequenceTree),
@@ -231,6 +241,10 @@ fn resolve_parsed_diagram(
     parsed_content: ParsedDiagram,
 ) -> Result<ResolvedDiagram, Box<dyn std::error::Error>> {
     match parsed_content {
+        ParsedDiagram::Activity(parsed_content) => {
+            let mut resolver = ActivityResolver::new();
+            puml_resolver(&mut resolver, &parsed_content).map(ResolvedDiagram::Activity)
+        }
         ParsedDiagram::Component(parsed_content) => {
             let mut resolver = ElementResolver::new();
             puml_resolver(&mut resolver, &parsed_content).map(ResolvedDiagram::Component)
@@ -289,6 +303,10 @@ fn parse_puml_file(
     diagram_type: DiagramType,
 ) -> Result<ParsedDiagram, Box<dyn std::error::Error>> {
     match diagram_type {
+        DiagramType::Activity => {
+            parse_with_parser(&mut PumlActivityParser, path, content, log_level)
+                .map(ParsedDiagram::Activity)
+        }
         DiagramType::Component | DiagramType::Deployment => {
             parse_with_parser(&mut PumlComponentParser, path, content, log_level)
                 .map(ParsedDiagram::Component)
@@ -326,6 +344,18 @@ fn parse_in_order(
         }
     }
 
+    match PumlActivityParser.parse_file(path, content, log_level) {
+        Ok(doc) => {
+            debug!("Successfully detected as Activity diagram");
+            return Ok(ParsedDiagram::Activity(doc));
+        }
+        Err(e) => {
+            let loc = e.error_location();
+            debug!("Activity parser failed at {:?}: {}", loc, e);
+            attempts.push(("Activity", Box::new(e), loc));
+        }
+    }
+
     match PumlClassParser.parse_file(path, content, log_level) {
         Ok(doc) => {
             debug!("Successfully detected as Class diagram");

plantuml/parser/puml_serializer/BUILD

@@ -17,12 +17,19 @@ rust_library(
     srcs = ["src/lib.rs"],
     visibility = ["//plantuml/parser:__subpackages__"],
     deps = [
+        ":activity_serializer",
         ":class_serializer",
         ":component_serializer",
         ":sequence_serializer",
     ],
 )
 
+alias(
+    name = "activity_serializer",
+    actual = "//plantuml/parser/puml_serializer/src/serialize:puml_serialize_activity",
+    visibility = ["//plantuml/parser:__subpackages__"],
+)
+
 alias(
     name = "component_serializer",
     actual = "//plantuml/parser/puml_serializer/src/serialize:puml_serialize_component",

plantuml/parser/puml_serializer/src/fbs/BUILD

@@ -41,6 +41,34 @@ rust_library(
     ],
 )
 
+flatbuffer_library_public(
+    name = "activity_fbs_codegen",
+    srcs = [
+        "activity.fbs",
+    ],
+    outs = [
+        "activity_generated.rs",
+    ],
+    flatc_args = [],
+    language_flag = "--rust",
+    visibility = ["//plantuml/parser:__subpackages__"],
+)
+
+rust_library(
+    name = "activity_fbs",
+    srcs = [
+        ":activity_fbs_codegen",
+    ],
+    # Generated code from flatc - suppress all lints we can't fix
+    rustc_flags = ["--cap-lints=allow"],
+    visibility = [
+        "//visibility:public",
+    ],
+    deps = [
+        "@crates//:flatbuffers",
+    ],
+)
+
 flatbuffer_library_public(
     name = "class_fbs_codegen",
     srcs = [

plantuml/parser/puml_serializer/src/fbs/activity.fbs

@@ -0,0 +1,90 @@
+// *******************************************************************************
+// Copyright (c) 2026 Contributors to the Eclipse Foundation
+//
+// See the NOTICE file(s) distributed with this work for additional
+// information regarding copyright ownership.
+//
+// This program and the accompanying materials are made available under the
+// terms of the Apache License Version 2.0 which is available at
+// https://www.apache.org/licenses/LICENSE-2.0
+//
+// SPDX-License-Identifier: Apache-2.0
+// *******************************************************************************
+
+namespace activity;
+
+enum ControlKind:byte {
+  Stop,
+  Kill,
+  Detach,
+  Break,
+  Continue
+}
+
+table IfDisplay {
+  then_label:string;
+  else_label:string;
+}
+
+table LoopDisplay {
+  continue_label:string;
+  exit_label:string;
+}
+
+table TitleNode {
+  text:string;
+}
+
+table ActionNode {
+  label:string;
+}
+
+table BackwardNode {
+  label:string;
+}
+
+table ControlNode {
+  kind:ControlKind = Stop;
+}
+
+table IfNode {
+  condition:string;
+  body:[ActivityStmt];
+  else_branch:[ActivityStmt];
+  display:IfDisplay;
+}
+
+table WhileNode {
+  condition:string;
+  body:[ActivityStmt];
+  backward:BackwardNode;
+  display:LoopDisplay;
+}
+
+table RepeatWhileNode {
+  body:[ActivityStmt];
+  condition:string;
+  backward:BackwardNode;
+  display:LoopDisplay;
+}
+
+union ActivityStmtValue {
+  TitleNode,
+  ActionNode,
+  IfNode,
+  WhileNode,
+  RepeatWhileNode,
+  ControlNode
+}
+
+table ActivityStmt {
+  value:ActivityStmtValue;
+}
+
+table ActivityDiagram {
+  name:string;
+  statements:[ActivityStmt];
+  source_file:string;
+}
+
+root_type ActivityDiagram;

plantuml/parser/puml_serializer/src/lib.rs

@@ -11,6 +11,7 @@
 // SPDX-License-Identifier: Apache-2.0
 // *******************************************************************************
 
+pub use activity_serializer::ActivitySerializer;
 pub use class_serializer::ClassSerializer;
 pub use component_serializer::ComponentSerializer;
 pub use sequence_serializer::SequenceSerializer;

plantuml/parser/puml_serializer/src/serialize/BUILD

@@ -12,6 +12,21 @@
 # *******************************************************************************
 load("@rules_rust//rust:defs.bzl", "rust_library")
 
+rust_library(
+    name = "puml_serialize_activity",
+    srcs = [
+        "activity_serializer.rs",
+    ],
+    crate_name = "activity_serializer",
+    crate_root = "activity_serializer.rs",
+    visibility = ["//plantuml/parser:__subpackages__"],
+    deps = [
+        "//plantuml/parser/puml_resolver:activity_resolver",
+        "//plantuml/parser/puml_serializer/src/fbs:activity_fbs",
+        "@crates//:flatbuffers",
+    ],
+)
+
 rust_library(
     name = "puml_serialize_component",
     srcs = [