Skip to content

Improve onyx state export masking#73428

Merged
MarioExpensify merged 9 commits into
mainfrom
michal-improve-masking
Oct 24, 2025
Merged

Improve onyx state export masking#73428
MarioExpensify merged 9 commits into
mainfrom
michal-improve-masking

Conversation

@mjasikowski

@mjasikowski mjasikowski commented Oct 24, 2025

Copy link
Copy Markdown
Contributor

Explanation of Change

  • Added onyx export masking to additional fields, including report/transaction amount and titles
  • Added the ability to mask while maintaining the field type - i.e. number will stay a number, date will stay a date
  • Added the ability to mask every field within a node while maintaining types and structures
  • Added full masking to reservationList

Additionally, improves on #72932, which made masking a little bit trigger happy in some circumstances

Fixed Issues

$
PROPOSAL:

Tests

Manual tests:

  1. Go to "Troubleshoot"
  2. Turn on "Mask fragile member data while exporting Onyx state"
  3. Export onyx state
  4. Check if the transaction and report amounts and titles are randomized
  5. Check if every field within the reservationList node is randomized, travel itineraries and confirmation codes are masked or randomized (all the dates should be set to today)
  6. Turn off "Mask fragile member data while exporting Onyx state"
  7. Check if the transaction and report fields are exported without masking
  8. Check if fields in the reservationList node are not randomized
  9. In another browser window, import the onyx state
  10. Check if the state has been imported without errors
  • Verify that no errors appear in the JS console

Offline tests

QA Steps

Same as tests

  • Verify that no errors appear in the JS console

PR Author Checklist

  • I linked the correct issue in the ### Fixed Issues section above
  • I wrote clear testing steps that cover the changes made in this PR
    • I added steps for local testing in the Tests section
    • I added steps for the expected offline behavior in the Offline steps section
    • I added steps for Staging and/or Production testing in the QA steps section
    • I added steps to cover failure scenarios (i.e. verify an input displays the correct error message if the entered data is not correct)
    • I turned off my network connection and tested it while offline to ensure it matches the expected behavior (i.e. verify the default avatar icon is displayed if app is offline)
    • I tested this PR with a High Traffic account against the staging or production API to ensure there are no regressions (e.g. long loading states that impact usability).
  • I included screenshots or videos for tests on all platforms
  • I ran the tests on all platforms & verified they passed on:
    • Android: Native
    • Android: mWeb Chrome
    • iOS: Native
    • iOS: mWeb Safari
    • MacOS: Chrome / Safari
    • MacOS: Desktop
  • I verified there are no console errors (if there's a console error not related to the PR, report it or open an issue for it to be fixed)
  • I verified there are no new alerts related to the canBeMissing param for useOnyx
  • I followed proper code patterns (see Reviewing the code)
    • I verified that any callback methods that were added or modified are named for what the method does and never what callback they handle (i.e. toggleReport and not onIconClick)
    • I verified that comments were added to code that is not self explanatory
    • I verified that any new or modified comments were clear, correct English, and explained "why" the code was doing something instead of only explaining "what" the code was doing.
    • I verified any copy / text shown in the product is localized by adding it to src/languages/* files and using the translation method
      • If any non-english text was added/modified, I used JaimeGPT to get English > Spanish translation. I then posted it in #expensify-open-source and it was approved by an internal Expensify engineer. Link to Slack message:
    • I verified all numbers, amounts, dates and phone numbers shown in the product are using the localization methods
    • I verified any copy / text that was added to the app is grammatically correct in English. It adheres to proper capitalization guidelines (note: only the first word of header/labels should be capitalized), and is either coming verbatim from figma or has been approved by marketing (in order to get marketing approval, ask the Bug Zero team member to add the Waiting for copy label to the issue)
    • I verified proper file naming conventions were followed for any new files or renamed files. All non-platform specific files are named after what they export and are not named "index.js". All platform-specific files are named for the platform the code supports as outlined in the README.
    • I verified the JSDocs style guidelines (in STYLE.md) were followed
  • If a new code pattern is added I verified it was agreed to be used by multiple Expensify engineers
  • I followed the guidelines as stated in the Review Guidelines
  • I tested other components that can be impacted by my changes (i.e. if the PR modifies a shared library or component like Avatar, I verified the components using Avatar are working as expected)
  • I verified all code is DRY (the PR doesn't include any logic written more than once, with the exception of tests)
  • I verified any variables that can be defined as constants (ie. in CONST.ts or at the top of the file that uses the constant) are defined as such
  • I verified that if a function's arguments changed that all usages have also been updated correctly
  • If any new file was added I verified that:
    • The file has a description of what it does and/or why is needed at the top of the file if the code is not self explanatory
  • If a new CSS style is added I verified that:
    • A similar style doesn't already exist
    • The style can't be created with an existing StyleUtils function (i.e. StyleUtils.getBackgroundAndBorderStyle(theme.componentBG))
  • If new assets were added or existing ones were modified, I verified that:
    • The assets are optimized and compressed (for SVG files, run npm run compress-svg)
    • The assets load correctly across all supported platforms.
  • If the PR modifies code that runs when editing or sending messages, I tested and verified there is no unexpected behavior for all supported markdown - URLs, single line code, code blocks, quotes, headings, bold, strikethrough, and italic.
  • If the PR modifies a generic component, I tested and verified that those changes do not break usages of that component in the rest of the App (i.e. if a shared library or component like Avatar is modified, I verified that Avatar is working as expected in all cases)
  • If the PR modifies a component related to any of the existing Storybook stories, I tested and verified all stories for that component are still working as expected.
  • If the PR modifies a component or page that can be accessed by a direct deeplink, I verified that the code functions as expected when the deeplink is used - from a logged in and logged out account.
  • If the PR modifies the UI (e.g. new buttons, new UI components, changing the padding/spacing/sizing, moving components, etc) or modifies the form input styles:
    • I verified that all the inputs inside a form are aligned with each other.
    • I added Design label and/or tagged @Expensify/design so the design team can review the changes.
  • If a new page is added, I verified it's using the ScrollView component to make it scrollable when more elements are added to the page.
  • I added unit tests for any new feature or bug fix in this PR to help automatically prevent regressions in this user flow.
  • If the main branch was merged into this PR after a review, I tested again and verified the outcome was still expected according to the Test steps.

@mjasikowski mjasikowski requested a review from a team as a code owner October 24, 2025 14:22
@melvin-bot melvin-bot Bot requested review from MarioExpensify and removed request for a team October 24, 2025 14:23
@melvin-bot

melvin-bot Bot commented Oct 24, 2025

Copy link
Copy Markdown

@MarioExpensify Please copy/paste the Reviewer Checklist from here into a new comment on this PR and complete it. If you have the K2 extension, you can simply click: [this button]

Comment thread src/libs/ExportOnyxState/common.ts Outdated
@@ -249,9 +288,35 @@ const maskFragileData = (data: OnyxState | unknown[] | null, parentKey?: string)

const value = data[propertyName];

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

❌ Potential Bug (docs)

There's a logical error here. You're accessing data[propertyName] but propertyName could be different from key when the key is an email (lines 283-287). This means you might be accessing undefined values.

Fix: Use the original key to access the data:

const value = (data as Record<string, unknown>)[key];

Comment thread src/libs/ExportOnyxState/common.ts Outdated
} else if (key.startsWith(ONYXKEYS.COLLECTION.TRANSACTION) && typeof value === 'object') {
maskedData[propertyName] = maskFragileData(value as OnyxState, key);
} else if (amountKeysToRandomize.includes(key) && typeof value === 'number') {
maskedData[propertyName] = randomizeAmount(value);

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

❌ Code Style (docs)

Trailing whitespace detected at the end of this line.

Fix: Remove the trailing space after randomizeAmount(value);

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@mjasikowski this will fail the style check

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@MarioExpensify the funny thing is the trailing space was never there. I triple checked that.

Comment thread src/libs/ExportOnyxState/common.ts
Comment thread src/libs/ExportOnyxState/common.ts
@github-actions

Copy link
Copy Markdown
Contributor

LGTM :feelsgood:. Thank you for your hard work!

@MarioExpensify

MarioExpensify commented Oct 24, 2025

Copy link
Copy Markdown
Contributor

Reviewer Checklist

  • I have verified the author checklist is complete (all boxes are checked off).
  • I verified the correct issue is linked in the ### Fixed Issues section above
  • I verified testing steps are clear and they cover the changes made in this PR
    • I verified the steps for local testing are in the Tests section
    • I verified the steps for Staging and/or Production testing are in the QA steps section
    • I verified the steps cover any possible failure scenarios (i.e. verify an input displays the correct error message if the entered data is not correct)
    • I turned off my network connection and tested it while offline to ensure it matches the expected behavior (i.e. verify the default avatar icon is displayed if app is offline)
  • I checked that screenshots or videos are included for tests on all platforms
  • I included screenshots or videos for tests on all platforms
  • I verified that the composer does not automatically focus or open the keyboard on mobile unless explicitly intended. This includes checking that returning the app from the background does not unexpectedly open the keyboard.
  • I verified tests pass on all platforms & I tested again on:
    • Android: HybridApp
    • Android: mWeb Chrome
    • iOS: HybridApp
    • iOS: mWeb Safari
    • MacOS: Chrome / Safari
    • MacOS: Desktop
  • If there are any errors in the console that are unrelated to this PR, I either fixed them (preferred) or linked to where I reported them in Slack
  • I verified there are no new alerts related to the canBeMissing param for useOnyx
  • I verified proper code patterns were followed (see Reviewing the code)
    • I verified that any callback methods that were added or modified are named for what the method does and never what callback they handle (i.e. toggleReport and not onIconClick).
    • I verified that comments were added to code that is not self explanatory
    • I verified that any new or modified comments were clear, correct English, and explained "why" the code was doing something instead of only explaining "what" the code was doing.
    • I verified any copy / text shown in the product is localized by adding it to src/languages/* files and using the translation method
    • I verified all numbers, amounts, dates and phone numbers shown in the product are using the localization methods
    • I verified any copy / text that was added to the app is grammatically correct in English. It adheres to proper capitalization guidelines (note: only the first word of header/labels should be capitalized), and is either coming verbatim from figma or has been approved by marketing (in order to get marketing approval, ask the Bug Zero team member to add the Waiting for copy label to the issue)
    • I verified proper file naming conventions were followed for any new files or renamed files. All non-platform specific files are named after what they export and are not named "index.js". All platform-specific files are named for the platform the code supports as outlined in the README.
    • I verified the JSDocs style guidelines (in STYLE.md) were followed
  • If a new code pattern is added I verified it was agreed to be used by multiple Expensify engineers
  • I verified that this PR follows the guidelines as stated in the Review Guidelines
  • I verified other components that can be impacted by these changes have been tested, and I retested again (i.e. if the PR modifies a shared library or component like Avatar, I verified the components using Avatar have been tested & I retested again)
  • I verified all code is DRY (the PR doesn't include any logic written more than once, with the exception of tests)
  • I verified any variables that can be defined as constants (ie. in CONST.ts or at the top of the file that uses the constant) are defined as such
  • If a new component is created I verified that:
    • A similar component doesn't exist in the codebase
    • All props are defined accurately and each prop has a /** comment above it */
    • The file is named correctly
    • The component has a clear name that is non-ambiguous and the purpose of the component can be inferred from the name alone
    • The only data being stored in the state is data necessary for rendering and nothing else
    • For Class Components, any internal methods passed to components event handlers are bound to this properly so there are no scoping issues (i.e. for onClick={this.submit} the method this.submit should be bound to this in the constructor)
    • Any internal methods bound to this are necessary to be bound (i.e. avoid this.submit = this.submit.bind(this); if this.submit is never passed to a component event handler like onClick)
    • All JSX used for rendering exists in the render method
    • The component has the minimum amount of code necessary for its purpose, and it is broken down into smaller components in order to separate concerns and functions
  • If any new file was added I verified that:
    • The file has a description of what it does and/or why is needed at the top of the file if the code is not self explanatory
  • If a new CSS style is added I verified that:
    • A similar style doesn't already exist
    • The style can't be created with an existing StyleUtils function (i.e. StyleUtils.getBackgroundAndBorderStyle(theme.componentBG)
  • If the PR modifies code that runs when editing or sending messages, I tested and verified there is no unexpected behavior for all supported markdown - URLs, single line code, code blocks, quotes, headings, bold, strikethrough, and italic.
  • If the PR modifies a generic component, I tested and verified that those changes do not break usages of that component in the rest of the App (i.e. if a shared library or component like Avatar is modified, I verified that Avatar is working as expected in all cases)
  • If the PR modifies a component related to any of the existing Storybook stories, I tested and verified all stories for that component are still working as expected.
  • If the PR modifies a component or page that can be accessed by a direct deeplink, I verified that the code functions as expected when the deeplink is used - from a logged in and logged out account.
  • If the PR modifies the UI (e.g. new buttons, new UI components, changing the padding/spacing/sizing, moving components, etc) or modifies the form input styles:
    • I verified that all the inputs inside a form are aligned with each other.
    • I added Design label and/or tagged @Expensify/design so the design team can review the changes.
  • If a new page is added, I verified it's using the ScrollView component to make it scrollable when more elements are added to the page.
  • For any bug fix or new feature in this PR, I verified that sufficient unit tests are included to prevent regressions in this flow.
  • If the main branch was merged into this PR after a review, I tested again and verified the outcome was still expected according to the Test steps.
  • I have checked off every checkbox in the PR reviewer checklist, including those that don't apply to this PR.

Screenshots/Videos

Android: HybridApp
Android: mWeb Chrome
iOS: HybridApp
iOS: mWeb Safari
MacOS: Chrome / Safari
MacOS: Desktop

@github-actions

Copy link
Copy Markdown
Contributor

LGTM :feelsgood:. Thank you for your hard work!

@mjasikowski

Copy link
Copy Markdown
Contributor Author

@MarioExpensify I'll need to fix jest tests and ESLint

Comment thread src/libs/ExportOnyxState/common.ts Outdated
return datePatterns.some((pattern) => pattern.test(value));
};

const getCurrentDate = (): string => {

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

What kind of date are we masking? Returning always the same or very close dates may make it difficult / impossible to triage problems in date arithmetics. Not sure how much value we lose on that though.

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@MarioExpensify travel dates - most notably flight dates. Just for safety, I don't want anybody to know when I'm not home.

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Awesome!

@github-actions

Copy link
Copy Markdown
Contributor

LGTM :feelsgood:. Thank you for your hard work!

Comment thread src/libs/ExportOnyxState/common.ts
Comment thread src/libs/ExportOnyxState/common.ts Outdated
}

const value = data[propertyName];
const value = (data as Record<string, unknown>)[key];

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

❌ Type Safety Issue

There's a potential type safety issue here. The code accesses (data as Record<string, unknown>)[key] but then assigns to maskedData[propertyName] where propertyName might be different from key (when the key is an email that gets masked). This could lead to inconsistent key handling.

Consider ensuring the key transformation is consistent:

const value = (data as Record<string, unknown>)[key];
// ... processing logic ...
maskedData[propertyName] = processedValue; // propertyName might be different from key

The logic should use the same key reference consistently or clearly document why different keys are used.

Comment thread src/libs/ExportOnyxState/common.ts
Comment thread src/libs/ExportOnyxState/common.ts Outdated
Comment thread src/libs/ExportOnyxState/common.ts
Comment thread src/libs/ExportOnyxState/common.ts
Comment thread src/libs/ExportOnyxState/common.ts Outdated

const emailRegex = /[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}/;

const emailMap = new Map<string, string>();

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

❌ Missing Memory Management

The emailMap is a module-level Map that persists across function calls. While it's cleared at the end of maskOnyxState, if the function throws an error before reaching the cleanup, the map will retain potentially sensitive data in memory.

const emailMap = new Map<string, string>();

Consider using a try-finally block or moving the map inside the function to ensure proper cleanup:

const maskOnyxState: MaskOnyxState = (data, isMaskingFragileDataEnabled) => {
    const emailMap = new Map<string, string>();
    try {
        // ... masking logic ...
    } finally {
        emailMap.clear();
    }
    return onyxState;
};

Comment thread tests/unit/ExportOnyxStateTest.ts
Comment thread src/libs/ExportOnyxState/common.ts Outdated
maskedData[propertyName] = maskFragileData(value as OnyxState, key);
} else if (key.startsWith(ONYXKEYS.COLLECTION.TRANSACTION) && typeof value === 'object') {
maskedData[propertyName] = maskFragileData(value as OnyxState, key);
} else if (amountKeysToRandomize.includes(key) && typeof value === 'number') {

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

❌ Logic Issue - Inconsistent Value Assignment

There's an inconsistency in how values are assigned. In some places, the code uses maskedData[key] = ... and in others maskedData[propertyName] = .... This could lead to incorrect key mappings, especially when the propertyName is different from the original key (e.g., when emails are masked).

} else if (amountKeysToRandomize.includes(key) && typeof value === 'number') {
    maskedData[propertyName] = randomizeAmount(value);
    // Handle expensify_text_title masking
} else if (parentKey === 'expensify_text_title' && key === 'value' && typeof value === 'string') {
    maskedData[propertyName] = maskValuePreservingLength(value);

vs.

} else if (keysToMask.includes(key)) {
    if (Array.isArray(value)) {
        maskedData[key] = value.map(() => MASKING_PATTERN);
    } else {
        maskedData[key] = maskValuePreservingLength(value);
    }

The assignment should be consistent - either always use propertyName or always use key, but not mix them.

Comment thread src/libs/ExportOnyxState/common.ts
Comment thread src/libs/ExportOnyxState/common.ts
@github-actions

Copy link
Copy Markdown
Contributor

Code Review Summary

I've completed a comprehensive review of PR #73428. The changes improve onyx state export masking functionality, but I've identified several issues that should be addressed:

Key Issues Found:

  1. Logic Issue: The shouldMask parameter is not being used in processOnyxKeyWithRule
  2. Performance Issue: Inefficient object creation in the masking loop
  3. Memory Management: Potential memory leak if function throws before cleanup
  4. Type Safety: Inconsistent key/propertyName usage throughout the code
  5. Security: Hardcoded field lists that may miss future fields

Positive Aspects:

  • Good separation of concerns with dedicated functions
  • Comprehensive test coverage updates
  • Proper handling of different data types (dates, amounts, emails)
  • Smart email masking with consistency

The functionality appears sound, but the implementation could be more robust and performant. Please address the inline comments for a production-ready solution.

@codecov

codecov Bot commented Oct 24, 2025

Copy link
Copy Markdown

Codecov Report

❌ Patch coverage is 71.11111% with 26 lines in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
src/libs/ExportOnyxState/common.ts 71.11% 26 Missing ⚠️
Files with missing lines Coverage Δ
src/libs/ExportOnyxState/common.ts 79.87% <71.11%> (-13.82%) ⬇️

... and 35 files with indirect coverage changes

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.
  • 📦 JS Bundle Analysis: Save yourself from yourself by tracking and limiting bundle sizes in JS merges.

@github-actions

Copy link
Copy Markdown
Contributor

LGTM :feelsgood:. Thank you for your hard work!

1 similar comment
@github-actions

Copy link
Copy Markdown
Contributor

LGTM :feelsgood:. Thank you for your hard work!

@MarioExpensify MarioExpensify left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM!

@MarioExpensify MarioExpensify merged commit 2f78586 into main Oct 24, 2025
28 checks passed
@MarioExpensify MarioExpensify deleted the michal-improve-masking branch October 24, 2025 18:47
@OSBotify

Copy link
Copy Markdown
Contributor

✋ This PR was not deployed to staging yet because QA is ongoing. It will be automatically deployed to staging after the next production release.

@OSBotify

Copy link
Copy Markdown
Contributor

🚀 Deployed to staging by https://github.com/MarioExpensify in version: 9.2.39-0 🚀

platform result
🖥 desktop 🖥 success ✅
🕸 web 🕸 success ✅
🤖 android 🤖 success ✅
🍎 iOS 🍎 success ✅

@OSBotify

Copy link
Copy Markdown
Contributor

🚀 Deployed to production by https://github.com/puneetlath in version: 9.2.39-3 🚀

platform result
🖥 desktop 🖥 success ✅
🕸 web 🕸 success ✅
🤖 android 🤖 success ✅
🍎 iOS 🍎 success ✅

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants