Skip to content
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.

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

feat: allow-developer-to-set-a-custom-refill-time-when-using-the #2114

Open
wants to merge 28 commits into
base: main
Choose a base branch
from

Conversation

MichaelUnkey
Copy link
Collaborator

@MichaelUnkey MichaelUnkey commented Sep 19, 2024

What does this PR do?

Eng-1324
Fixes # (issue)

If there is not an issue for this, please create one first. This is used to tracking purposes and also helps use understand why this PR exists

Type of change

  • Bug fix (non-breaking change which fixes an issue)
  • Chore (refactoring code, technical debt, workflow improvements)
  • Enhancement (small improvements)
  • New feature (non-breaking change which adds functionality)
  • Breaking change (fix or feature that would cause existing functionality to not work as expected)
  • This change requires a documentation update

How should this be tested?

  • Test A
  • Test B

Checklist

Required

  • Filled out the "How to test" section in this PR
  • Read Contributing Guide
  • Self-reviewed my own code
  • Commented on my code in hard-to-understand areas
  • Ran pnpm build
  • Ran pnpm fmt
  • Checked for warnings, there are none
  • Removed all console.logs
  • Merged the latest changes from main onto my branch with git pull origin main
  • My changes don't cause any responsiveness issues

Appreciated

  • If a UI change was made: Added a screen recording or screenshots to this PR
  • Updated the Unkey Docs if changes were necessary

Summary by CodeRabbit

Release Notes

  • New Features

    • Introduced a new optional property refillDay for specifying the day of the month for refills, applicable when the refill interval is set to "monthly."
    • Enhanced logic for handling refill configurations, including validation checks to prevent invalid settings for refillDay.
  • Documentation

    • Updated API documentation to reflect the addition of refillDay, including its default behavior and usage examples.
  • Bug Fixes

    • Improved validation logic for refill-related properties to ensure accurate processing and handling of refill schedules.
  • Tests

    • Added new test cases to validate the behavior of the refillDay parameter and ensure proper error handling for invalid configurations.

Copy link

changeset-bot bot commented Sep 19, 2024

⚠️ No Changeset found

Latest commit: 13529b1

Merging this PR will not cause a version bump for any packages. If these changes should not result in a new version, you're good to go. If these changes should result in a version bump, you need to add a changeset.

This PR includes no changesets

When changesets are added to this PR, you'll see the packages that this PR includes changesets for and the associated semver types

Click here to learn what changesets are, and how to add one.

Click here if you're a maintainer who wants to add a changeset to this PR

Copy link

vercel bot commented Sep 19, 2024

The latest updates on your projects. Learn more about Vercel for Git ↗︎

Name Status Preview Comments Updated (UTC)
play ✅ Ready (Inspect) Visit Preview 💬 Add feedback Sep 29, 2024 8:49pm
www ✅ Ready (Inspect) Visit Preview 💬 Add feedback Sep 29, 2024 8:49pm
3 Skipped Deployments
Name Status Preview Comments Updated (UTC)
dashboard ⬜️ Ignored (Inspect) Visit Preview Sep 29, 2024 8:49pm
planetfall ⬜️ Ignored (Inspect) Visit Preview Sep 29, 2024 8:49pm
workflows ⬜️ Ignored (Inspect) Visit Preview Sep 29, 2024 8:49pm

Copy link
Contributor

coderabbitai bot commented Sep 19, 2024

📝 Walkthrough
📝 Walkthrough

Walkthrough

The changes introduce a new optional property, refillDay, to the refill logic in the key management system. This property allows users to specify the day of the month for refills when the interval is set to "monthly." The updates include validation checks to ensure compatibility with daily intervals, adjustments to request handling, and modifications to related tests and documentation. Overall, these changes enhance the functionality and clarity of the refill scheduling process.

Changes

Files Change Summary
apps/api/src/routes/v1_keys_createKey.ts, apps/api/src/routes/v1_keys_updateKey.ts, apps/api/src/routes/v1_migrations_createKey.ts Added refillDay property to the refill object schema, including validation and request handling logic updates.
apps/api/src/routes/v1_keys_updateKey.error.test.ts, apps/api/src/routes/v1_keys_updateKey.happy.test.ts, apps/api/src/routes/v1_migrations_createKey.happy.test.ts Added tests to verify the behavior of the refillDay parameter, ensuring correct error handling and default values.

📜 Recent review details

Configuration used: CodeRabbit UI
Review profile: CHILL

📥 Commits

Files that changed from the base of the PR and between 666c6a3 and 13529b1.

📒 Files selected for processing (1)
  • apps/api/src/routes/v1_migrations_createKey.happy.test.ts (1 hunks)
🚧 Files skipped from review as they are similar to previous changes (1)
  • apps/api/src/routes/v1_migrations_createKey.happy.test.ts

Thank you for using CodeRabbit. We offer it for free to the OSS community and would appreciate your support in helping us grow. If you find it useful, would you consider giving us a shout-out on your favorite social media?

❤️ Share
🪧 Tips

Chat

There are 3 ways to chat with CodeRabbit:

  • Review comments: Directly reply to a review comment made by CodeRabbit. Example:
    • I pushed a fix in commit <commit_id>, please review it.
    • Generate unit testing code for this file.
    • Open a follow-up GitHub issue for this discussion.
  • Files and specific lines of code (under the "Files changed" tab): Tag @coderabbitai in a new review comment at the desired location with your query. Examples:
    • @coderabbitai generate unit testing code for this file.
    • @coderabbitai modularize this function.
  • PR comments: Tag @coderabbitai in a new PR comment to ask questions about the PR branch. For the best results, please provide a very specific query, as very limited context is provided in this mode. Examples:
    • @coderabbitai gather interesting stats about this repository and render them as a table. Additionally, render a pie chart showing the language distribution in the codebase.
    • @coderabbitai read src/utils.ts and generate unit testing code.
    • @coderabbitai read the files in the src/scheduler package and generate a class diagram using mermaid and a README in the markdown format.
    • @coderabbitai help me debug CodeRabbit configuration file.

Note: Be mindful of the bot's finite context window. It's strongly recommended to break down tasks such as reading entire modules into smaller chunks. For a focused discussion, use review comments to chat about specific files and their changes, instead of using the PR comments.

CodeRabbit Commands (Invoked using PR comments)

  • @coderabbitai pause to pause the reviews on a PR.
  • @coderabbitai resume to resume the paused reviews.
  • @coderabbitai review to trigger an incremental review. This is useful when automatic reviews are disabled for the repository.
  • @coderabbitai full review to do a full review from scratch and review all the files again.
  • @coderabbitai summary to regenerate the summary of the PR.
  • @coderabbitai resolve resolve all the CodeRabbit review comments.
  • @coderabbitai configuration to show the current CodeRabbit configuration for the repository.
  • @coderabbitai help to get help.

Other keywords and placeholders

  • Add @coderabbitai ignore anywhere in the PR description to prevent this PR from being reviewed.
  • Add @coderabbitai summary to generate the high-level summary at a specific location in the PR description.
  • Add @coderabbitai anywhere in the PR title to generate the title automatically.

CodeRabbit Configuration File (.coderabbit.yaml)

  • You can programmatically configure CodeRabbit by adding a .coderabbit.yaml file to the root of your repository.
  • Please see the configuration documentation for more information.
  • If your editor has YAML language server enabled, you can add the path at the top of this file to enable auto-completion and validation: # yaml-language-server: $schema=https://coderabbit.ai/integrations/schema.v2.json

Documentation and Community

  • Visit our Documentation for detailed information on how to use CodeRabbit.
  • Join our Discord Community to get help, request features, and share feedback.
  • Follow us on X/Twitter for updates and announcements.

Copy link
Contributor

github-actions bot commented Sep 19, 2024

Thank you for following the naming conventions for pull request titles! 🙏

…-allow-developer-to-set-a-custom-refill-time-when-using-the
…en-using-the' of https://github.com/unkeyed/unkey into eng-1324-allow-developer-to-set-a-custom-refill-time-when-using-the
…-allow-developer-to-set-a-custom-refill-time-when-using-the
apps/api/src/routes/schema.ts Show resolved Hide resolved
apps/api/src/routes/v1_apis_listKeys.ts Outdated Show resolved Hide resolved
apps/api/src/routes/v1_keys_createKey.happy.test.ts Outdated Show resolved Hide resolved
apps/api/src/routes/v1_keys_createKey.happy.test.ts Outdated Show resolved Hide resolved
apps/api/src/routes/v1_keys_createKey.ts Outdated Show resolved Hide resolved
apps/workflows/jobs/refill-daily.ts Outdated Show resolved Hide resolved
apps/workflows/jobs/refill-daily.ts Show resolved Hide resolved
apps/workflows/jobs/refill-daily.ts Show resolved Hide resolved
apps/workflows/jobs/refill-daily.ts Outdated Show resolved Hide resolved
apps/workflows/jobs/refill-daily.ts Outdated Show resolved Hide resolved
Copy link
Contributor

@coderabbitai coderabbitai bot left a comment

Choose a reason for hiding this comment

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

Actionable comments posted: 2

🧹 Outside diff range and nitpick comments (9)
tools/migrate/tinybird-export.ts (3)

6-7: Consider removing or conditionally executing the console.log statement.

While this log statement is useful for debugging and tracking the export process, it's generally not recommended to leave console.log statements in production code. Consider one of the following options:

  1. Remove the console.log statement before merging to production.
  2. Wrap it in a conditional statement that only executes in non-production environments.
  3. Use a proper logging library that can be configured based on the environment.

If you decide to keep it, you could refactor it to use a logging library or a conditional statement:

if (process.env.NODE_ENV !== 'production') {
  console.log({ cursor });
}

Or using a logging library:

import logger from './your-logging-library';

logger.debug({ cursor });

17-18: Consider removing or conditionally executing the console.log statement.

Similar to the previous log statement, while this is useful for debugging and tracking the number of records fetched, it's generally not recommended to leave console.log statements in production code. Consider applying the same solution as suggested for the previous log statement.

If you decide to keep it, you could refactor it to use a logging library or a conditional statement:

if (process.env.NODE_ENV !== 'production') {
  console.log("found", body.data.length);
}

Or using a logging library:

import logger from './your-logging-library';

logger.debug("found", body.data.length);

Line range hint 1-28: Consider adding error handling to the main function.

The overall structure and logic of the main function look good. However, it lacks error handling for the fetch operation and file writing. This could lead to silent failures or incomplete data exports if an error occurs during the process.

Consider adding try-catch blocks to handle potential errors. Here's a suggested refactor:

async function main() {
  const exportFile = Bun.file("./export_audit_log.json");
  const writer = exportFile.writer();
  let cursor = "";
  try {
    do {
      if (process.env.NODE_ENV !== 'production') {
        console.log({ cursor });
      }
      const res = await fetch(
        `https://api.tinybird.co/v0/pipes/export_audit_log.json?cursor=${cursor}`,
        {
          headers: {
            Authorization: `Bearer ${process.env.TINYBIRD_TOKEN}`,
          },
        },
      );
      if (!res.ok) {
        throw new Error(`HTTP error! status: ${res.status}`);
      }
      const body = (await res.json()) as { data: { auditLogId: string }[] };
      if (process.env.NODE_ENV !== 'production') {
        console.log("found", body.data.length);
      }
      for (const row of body.data) {
        await writer.write(JSON.stringify(row) + "\n");
      }
      cursor = body.data.at(-1)?.auditLogId ?? "";
    } while (cursor);
  } catch (error) {
    console.error("An error occurred during the export process:", error);
  } finally {
    await writer.end();
  }
}

main().catch(error => console.error("Unhandled error in main:", error));

This refactored version:

  1. Adds a try-catch block to handle potential errors during the fetch and write operations.
  2. Checks if the fetch response is ok before proceeding.
  3. Uses async/await for the file writing to properly handle potential errors.
  4. Ensures the writer is closed in a finally block.
  5. Adds error handling for the main function call itself.

These changes will make the script more robust and easier to debug if issues occur during the export process.

tools/migrate/main.ts (1)

Line range hint 31-43: Overall, the changes enhance tracking without affecting functionality.

The added console.log statements improve the ability to track and debug the migration process without impacting the core functionality. This aligns well with the PR objectives.

As a minor suggestion, consider using a logging library like Winston or Pino instead of console.log for more structured and configurable logging in the future. This would allow for easier log management and potential integration with monitoring tools.

apps/api/src/routes/v1_migrations_createKey.error.test.ts (1)

115-150: Approve with suggestions for improvement

The new test case effectively checks for invalid refill configuration when a daily interval has a non-null refillDay. The structure and assertions are correct. However, consider the following suggestions for improvement:

  1. Make the test name more specific, e.g., "reject refill config with daily interval and non-null refillDay".
  2. Use a named constant for the refillDay value (4) to improve readability.
  3. Consider asserting all relevant properties of the error object in the response.

Example improvement:

test("reject refill config with daily interval and non-null refillDay", async (t) => {
  // ... (existing setup code)

  const INVALID_REFILL_DAY = 4;
  const res = await h.post<V1MigrationsCreateKeysRequest, ErrorResponse>({
    // ... (existing request code)
    body: [
      {
        // ... (other properties)
        refill: {
          amount: 100,
          refillDay: INVALID_REFILL_DAY,
          interval: "daily",
        },
      },
    ],
  });

  expect(res.status).toEqual(400);
  expect(res.body).toMatchObject({
    error: {
      code: "BAD_REQUEST",
      docs: "https://unkey.dev/docs/api-reference/errors/code/BAD_REQUEST",
      message: "when interval is set to 'daily', 'refillDay' must be null.",
      // Consider adding other relevant properties to assert
    },
  });
});

These changes would enhance the test's clarity and maintainability.

apps/api/src/routes/v1_migrations_createKey.happy.test.ts (2)

501-501: Test case name could be more specific

The current test name "should provide default value" is somewhat vague. It would be helpful to specify what default value is being tested.

Consider renaming the test case to be more specific:

-  test("should provide default value", async (t) => {
+  test("should default refillDay to 1 when not provided", async (t) => {

525-525: Explicit undefined value

While setting refillDay to undefined is correct, it might be beneficial to add a comment explaining why this is done explicitly, as undefined is the default value for an omitted property.

Consider adding a comment to explain the explicit undefined:

-            refillDay: undefined,
+            refillDay: undefined, // Explicitly set to undefined to test default behavior
apps/api/src/routes/v1_keys_updateKey.ts (2)

356-360: Consider renaming lastDayOfMonth for clarity

The logic for determining the refill day is correct. However, when refillDay is provided and valid, the variable lastDayOfMonth no longer represents the last day of the month. Consider renaming it to something like refillDayOfMonth for better clarity.

Here's a suggested change:

 const date = new Date();
-let lastDayOfMonth = new Date(date.getFullYear(), date.getMonth() + 1, 0).getDate();
-if (req.refill?.refillDay && req?.refill?.refillDay <= lastDayOfMonth) {
-  lastDayOfMonth = req.refill.refillDay;
+let refillDayOfMonth = new Date(date.getFullYear(), date.getMonth() + 1, 0).getDate();
+if (req.refill?.refillDay && req?.refill?.refillDay <= refillDayOfMonth) {
+  refillDayOfMonth = req.refill.refillDay;
 }

Line range hint 1-1: Reminder: Add test cases for the new refillDay feature

The implementation of the refillDay feature looks good, but I don't see any new tests added. Please ensure that you add appropriate test cases to cover this new functionality, including edge cases and error scenarios.

Would you like assistance in creating these test cases or opening a GitHub issue to track this task?

📜 Review details

Configuration used: CodeRabbit UI
Review profile: CHILL

📥 Commits

Files that changed from the base of the PR and between cb283dc and 6476ba5.

⛔ Files ignored due to path filters (1)
  • pnpm-lock.yaml is excluded by !**/pnpm-lock.yaml
📒 Files selected for processing (19)
  • apps/api/src/routes/schema.ts (2 hunks)
  • apps/api/src/routes/v1_apis_listKeys.ts (1 hunks)
  • apps/api/src/routes/v1_keys_createKey.happy.test.ts (1 hunks)
  • apps/api/src/routes/v1_keys_createKey.ts (4 hunks)
  • apps/api/src/routes/v1_keys_updateKey.error.test.ts (2 hunks)
  • apps/api/src/routes/v1_keys_updateKey.happy.test.ts (1 hunks)
  • apps/api/src/routes/v1_keys_updateKey.ts (4 hunks)
  • apps/api/src/routes/v1_migrations_createKey.error.test.ts (1 hunks)
  • apps/api/src/routes/v1_migrations_createKey.happy.test.ts (1 hunks)
  • apps/api/src/routes/v1_migrations_createKey.ts (3 hunks)
  • apps/dashboard/app/(app)/apis/[apiId]/keys/[keyAuthId]/new/client.tsx (9 hunks)
  • apps/dashboard/lib/trpc/routers/key/create.ts (2 hunks)
  • apps/docs/libraries/ts/sdk/keys/get.mdx (1 hunks)
  • apps/docs/libraries/ts/sdk/keys/update.mdx (2 hunks)
  • apps/workflows/jobs/refill-daily.ts (3 hunks)
  • internal/db/src/schema/key_migrations.ts (1 hunks)
  • tools/migrate/auditlog-import.ts (2 hunks)
  • tools/migrate/main.ts (2 hunks)
  • tools/migrate/tinybird-export.ts (2 hunks)
✅ Files skipped from review due to trivial changes (1)
  • tools/migrate/auditlog-import.ts
🚧 Files skipped from review as they are similar to previous changes (13)
  • apps/api/src/routes/schema.ts
  • apps/api/src/routes/v1_apis_listKeys.ts
  • apps/api/src/routes/v1_keys_createKey.happy.test.ts
  • apps/api/src/routes/v1_keys_createKey.ts
  • apps/api/src/routes/v1_keys_updateKey.error.test.ts
  • apps/api/src/routes/v1_keys_updateKey.happy.test.ts
  • apps/api/src/routes/v1_migrations_createKey.ts
  • apps/dashboard/app/(app)/apis/[apiId]/keys/[keyAuthId]/new/client.tsx
  • apps/dashboard/lib/trpc/routers/key/create.ts
  • apps/docs/libraries/ts/sdk/keys/get.mdx
  • apps/docs/libraries/ts/sdk/keys/update.mdx
  • apps/workflows/jobs/refill-daily.ts
  • internal/db/src/schema/key_migrations.ts
🧰 Additional context used
📓 Learnings (1)
apps/api/src/routes/v1_migrations_createKey.error.test.ts (1)
Learnt from: MichaelUnkey
PR: unkeyed/unkey#2114
File: apps/api/src/routes/v1_keys_updateKey.error.test.ts:0-0
Timestamp: 2024-09-27T15:20:05.475Z
Learning: In the `v1/keys.updateKey` endpoint, the server validates the refill configuration before checking if the key exists. Therefore, tests can assert validation errors without needing to create the key first.
🔇 Additional comments (5)
tools/migrate/main.ts (2)

31-32: LGTM: Useful logging for tracking migration progress.

The added console.log statement provides valuable information for tracking the creation of new identities during the migration process. The comment to ignore the linter warning is appropriate in this context.


42-43: LGTM: Helpful logging for tracking key-identity connections.

The added console.log statement effectively tracks the connection between identities and keys during the migration process. The comment to ignore the linter warning is appropriate in this context.

apps/api/src/routes/v1_keys_updateKey.ts (3)

154-157: LGTM: New refillDay field added correctly

The new refillDay field is correctly implemented as an optional number between 1 and 31, with a clear OpenAPI description explaining its purpose for monthly refills.


340-345: LGTM: Proper validation for refillDay

The added validation logic correctly prevents setting refillDay when the interval is "daily", throwing an informative error message.


391-391: LGTM: Correct refillDay assignment

The refillDay is correctly set based on the refill interval. If you accept the suggestion to rename lastDayOfMonth to refillDayOfMonth, remember to update it here as well.

apps/api/src/routes/v1_migrations_createKey.happy.test.ts Outdated Show resolved Hide resolved
apps/api/src/routes/v1_migrations_createKey.happy.test.ts Outdated Show resolved Hide resolved
…en-using-the' of https://github.com/unkeyed/unkey into eng-1324-allow-developer-to-set-a-custom-refill-time-when-using-the
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.

2 participants