Skip to content

Bloblang playground refactor #290

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
mihaitodor wants to merge 1 commit into
base: main
Choose a base branch
from
Draft

Bloblang playground refactor #290

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
71 changes: 36 additions & 35 deletions blobl-editor/wasm/main.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -22,22 +22,37 @@ func main() {
select {}
}

func blobl(_ js.Value, args []js.Value) (output any) {
defer func() {
// Make sure we return a string instead of an error
if o, ok := output.(error); ok {
output = "Error: " + o.Error()
}
}()
func blobl(_ js.Value, args []js.Value) any {
msg, meta, err := bloblImpl(args)
var bloblErr string
if err != nil {
bloblErr = err.Error()
}
payload, err := json.Marshal(struct {
Msg any `json:"msg"`
Meta map[string]any `json:"meta,omitempty"`
Err string `json:"err"`
}{
Msg: msg,
Meta: meta,
Err: bloblErr,
})
if err != nil {
return fmt.Sprintf(`{"err": "Error: failed to marshal output: %s"`, err)
}

return string(payload)
}

func bloblImpl(args []js.Value) (message any, metadata map[string]any, err error) {
if len(args) < 2 || len(args) > 3 {
return fmt.Errorf("expected 2 or 3 arguments, received %d instead", len(args))
return nil, nil, fmt.Errorf("expected 2 or 3 arguments, received %d instead", len(args))
}

// Parse the mapping
mapping, err := globalEnv.Parse(args[0].String())
if err != nil {
return fmt.Errorf("failed to parse mapping: %s", err)
return nil, nil, fmt.Errorf("failed to parse mapping: %s", err)
}

// Take the raw data without unmarshaling
Expand All @@ -47,61 +62,47 @@ func blobl(_ js.Value, args []js.Value) (output any) {
msg := service.NewMessage(payloadBytes)

// Parse the optional metadata (this can still be JSON)
metadata := map[string]any{}
inputMetadata := map[string]any{}
if len(args) == 3 {
if err := json.Unmarshal([]byte(args[2].String()), &metadata); err != nil {
return fmt.Errorf("failed to parse metadata: %s", err)
if err := json.Unmarshal([]byte(args[2].String()), &inputMetadata); err != nil {
return nil, nil, fmt.Errorf("failed to parse metadata: %s", err)
}
}

// Apply metadata to the message
for key, value := range metadata {
for key, value := range inputMetadata {
strValue, ok := value.(string)
if !ok {
return fmt.Errorf("metadata value for key '%s' must be a string, got %T", key, value)
return nil, nil, fmt.Errorf("metadata value for key '%s' must be a string, got %T", key, value)
}
msg.MetaSetMut(key, strValue)
}

// Execute the mapping
result, err := msg.BloblangQuery(mapping)
if err != nil {
return fmt.Errorf("failed to execute mapping: %s", err)
return nil, nil, fmt.Errorf("failed to execute mapping: %s", err)
}

message, err := result.AsStructured()
message, err = result.AsStructured()
if err != nil {
res, err := result.AsBytes()
if err != nil {
return fmt.Errorf("failed to extract message: %s", err)
return nil, nil, fmt.Errorf("failed to extract message: %s", err)
}
message = string(res)
}

// Extract metadata
var extractedMetadata map[string]any
metadata = map[string]any{}
if err = result.MetaWalkMut(func(key string, value any) error {
if extractedMetadata == nil {
extractedMetadata = make(map[string]any)
}
extractedMetadata[key] = value
metadata[key] = value
return nil
}); err != nil {
return fmt.Errorf("failed to extract metadata: %s", err)
return nil, nil, fmt.Errorf("failed to extract metadata: %s", err)
}

payload, err := json.MarshalIndent(struct {
Msg any `json:"msg"`
Meta map[string]any `json:"meta,omitempty"`
}{
Msg: message,
Meta: extractedMetadata,
}, "", " ")
if err != nil {
return fmt.Errorf("failed to marshal output: %s", err)
}

return string(payload)
return message, metadata, nil
}

// createMockEnvironment creates a shared Bloblang environment with mocked I/O functions.
Expand Down
41 changes: 21 additions & 20 deletions src/partials/bloblang-playground.hbs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -309,34 +309,35 @@ function execute() {
}
const result = blobl(getMapping(), getInput(), getInputMetadata());

if (isValidJSON(result)) {
customAIButton.style.display = "none";
const parsedResult = JSON.parse(result);
customAIButton.style.display = "none";

// Separate message and metadata
const message = parsedResult.msg;
const metadata = parsedResult.meta || {};
// TODO: Fix `root = "3" -> "e.match is not a function"
// TODO: Prevent the output of `root = "[3,4,5]"` from being interpreted as a JSON
const parsedResult = JSON.parse(result);

// Display message output
if (parsedResult.err !== "") {
throw new Error(parsedResult.err);
}

// Display message output
const message = parsedResult.msg
if (typeof message === "object") {
aceOutputEditor.session.setMode("ace/mode/json");
aceOutputEditor.setValue(JSON.stringify(message, null, 2), 1);

// Display metadata
if (Object.keys(metadata).length > 0) {
aceOutputMetadataEditor.session.setMode("ace/mode/json");
aceOutputMetadataEditor.setValue(JSON.stringify(metadata, null, 2), 1);
} else {
aceOutputMetadataEditor.session.setMode("ace/mode/text");
aceOutputMetadataEditor.setValue("{}", 1);
}
} else {
// If the result is not JSON, handle it as raw text
aceOutputEditor.session.setMode("ace/mode/text");
aceOutputEditor.setValue(result, 1);
aceOutputEditor.setValue(message, 1);
customAIButton.style.display = "inline-block";
}

aceOutputMetadataEditor.session.setMode("ace/mode/text");
aceOutputMetadataEditor.setValue("No metadata available", 1);
// Display metadata
const metadata = parsedResult.meta || {};
aceOutputMetadataEditor.session.setMode("ace/mode/json");
if (Object.keys(metadata).length > 0) {
aceOutputMetadataEditor.setValue(JSON.stringify(metadata, null, 2), 1);
} else {
aceOutputMetadataEditor.setValue("{}", 1);
}
} catch (error) {
// Handle general errors
Expand All @@ -345,7 +346,7 @@ function execute() {
customAIButton.style.display = "inline-block";

aceOutputMetadataEditor.session.setMode("ace/mode/text");
aceOutputMetadataEditor.setValue("Error: " + error.message, 1);
aceOutputMetadataEditor.setValue("No metadata available", 1);
}
}

Expand Down
Loading