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

Revamp JoinRecords #93

Open
wants to merge 39 commits into
base: main
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from 18 commits
Commits
Show all changes
39 commits
Select commit Hold shift + click to select a range
7e5ea5d
Stub out updating S3 submissions with httpCode
WillNilges Oct 13, 2024
799a96a
checkpoint
WillNilges Oct 29, 2024
a5635a8
Add replay stats on JoinRecords
WillNilges Oct 29, 2024
68a40d4
what the fuck amazon
WillNilges Oct 29, 2024
c44f157
Shit now I need S3 in the github tests
WillNilges Oct 29, 2024
397984c
First wack updating github tests
WillNilges Oct 29, 2024
211c405
change a million things
WillNilges Oct 31, 2024
7562d4d
Checkpoint. Need to update JoinRecord here to match MeshDB
WillNilges Oct 31, 2024
0912cdc
checkpoint
WillNilges Oct 31, 2024
280271e
Nullable install Number. Make tests pass
WillNilges Oct 31, 2024
3439e9b
Merge branch 'main' into wdn/updated-s3-submissions
WillNilges Nov 1, 2024
4feb9d9
pretty
WillNilges Nov 1, 2024
e61fe71
checkpoint
WillNilges Nov 1, 2024
31b55ad
This is kinda jank
WillNilges Nov 1, 2024
cb0e38a
Add submission time
WillNilges Nov 1, 2024
01591a0
Submission time part 2
WillNilges Nov 1, 2024
03aad06
Add Join Record to more tests
WillNilges Nov 1, 2024
6e619dc
pretty
WillNilges Nov 1, 2024
4d9c466
chom
WillNilges Nov 3, 2024
7798ec2
pretty
WillNilges Nov 3, 2024
f5c2c0a
Gracefully handle S3 misconfiguration.
WillNilges Nov 3, 2024
0f9bdde
pretty
WillNilges Nov 3, 2024
65b1a2a
i just crashed kde
WillNilges Nov 3, 2024
1c50579
it's the whole fucking point
WillNilges Nov 3, 2024
14c81f6
type type type
WillNilges Nov 3, 2024
1ca0ab9
if you crash one more time I am going to atomize your entire bloodline
WillNilges Nov 3, 2024
beae3d2
i completely broke everything
WillNilges Nov 3, 2024
ceb320d
ugh
WillNilges Nov 3, 2024
deda4b6
ugh
WillNilges Nov 3, 2024
1a1e195
fuck dirty
WillNilges Nov 3, 2024
9d355d2
Delete shotgun debugs
WillNilges Nov 3, 2024
f06e7f7
pretty
WillNilges Nov 3, 2024
74a4de1
fix github action
WillNilges Nov 3, 2024
a888508
Fix(?) github action
WillNilges Nov 4, 2024
328af23
Cleanup
WillNilges Nov 4, 2024
923029a
Delete demo test
WillNilges Nov 5, 2024
d1a7b45
Add test for meshdb being hard down
WillNilges Nov 5, 2024
50586c1
Tests are failing when run in parallel. Probably some nonsense with b…
WillNilges Nov 5, 2024
0037c61
Maybe this will work too
WillNilges Nov 5, 2024
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
22 changes: 22 additions & 0 deletions .github/workflows/playwright.yml
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,22 @@ jobs:
name: Run Playwright Tests
timeout-minutes: 60
runs-on: ubuntu-latest
services:
minio:
image: quay.io/minio/minio
ports:
- '9000:9000'
- '9001:9001'
volumes:
- 'minio_data:/data'
command: |
server /data
environment:
- MINIO_ROOT_USER=admin
- MINIO_ROOT_PASSWORD=admin1234
- MINIO_DEFAULT_BUCKETS=meshdb-join-form-log
- MINIO_ACCESS_KEY=sampleaccesskey
- MINIO_SECRET_KEY=samplesecretkey
WillNilges marked this conversation as resolved.
Show resolved Hide resolved
steps:
- uses: actions/checkout@692973e3d937129bcbf40652eb9f2f61becf3332
- uses: actions/setup-node@0a44ba7841725637a19e28fa30b79a866c81b0a6
Expand All @@ -24,6 +40,12 @@ jobs:
env:
NEXT_PUBLIC_MESHDB_URL: https://127.0.0.1:8000 # Throwaway to make the mock work
MESHDB_URL: https://127.0.0.1:8000 # Throwaway to make the mock work
# We now check the JoinRecord stuff, so submit that too.
S3_ENDPOINT: http://127.0.0.1:9000
S3_BUCKET_NAME: meshdb-join-form-log
S3_BASE_NAME: dev-join-form-submissions
S3_ACCESS_KEY: testaccesskey
S3_SECRET_KEY: testsecretkey
- uses: actions/upload-artifact@v4
if: always()
with:
Expand Down
76 changes: 0 additions & 76 deletions app/data.ts

This file was deleted.

127 changes: 127 additions & 0 deletions app/join_record.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,127 @@
"use server";
// import { access, constants, appendFileSync, readFile } from "node:fs";
import {
S3Client,
PutObjectCommand,
GetObjectCommand,
} from "@aws-sdk/client-s3";
import { JoinFormValues } from "@/components/JoinForm/JoinForm";
import { Readable } from "stream";
import { JoinRecord } from "./types";

// const JOIN_FORM_LOG = process.env.JOIN_FORM_LOG as string;

class JoinRecordS3 {
private s3Client: S3Client;

private S3_REGION: string;
private S3_ENDPOINT: string;
private S3_BUCKET_NAME: string;
private S3_BASE_NAME: string;
private S3_ACCESS_KEY: string;
private S3_SECRET_KEY: string;

constructor() {
this.S3_REGION = process.env.S3_REGION as string;
this.S3_ENDPOINT = process.env.S3_ENDPOINT as string;
this.S3_BUCKET_NAME = process.env.S3_BUCKET_NAME as string;
this.S3_BASE_NAME = process.env.S3_BASE_NAME as string;
this.S3_ACCESS_KEY = process.env.S3_ACCESS_KEY as string;
this.S3_SECRET_KEY = process.env.S3_SECRET_KEY as string;

// Setup the S3 client
this.s3Client = new S3Client({
region: this.S3_REGION != undefined ? this.S3_REGION : "us-east-1",
endpoint:
this.S3_ENDPOINT != undefined
? this.S3_ENDPOINT
: "https://s3.us-east-1.amazonaws.com",
credentials: {
accessKeyId: this.S3_ACCESS_KEY,
secretAccessKey: this.S3_SECRET_KEY,
},
WillNilges marked this conversation as resolved.
Show resolved Hide resolved
});
}

// Records the submission we just got as a JSON object in an S3 bucket.
// submission: A Join Form Submission. We append a few things to this.
// key: The S3 path we store the submission at
// responseCode: If we have a response code for this submission, add it here.
async save(joinRecord: JoinRecord, key: string = "") {
// Bail if there's no S3 key
if (this.S3_ACCESS_KEY === undefined || this.S3_SECRET_KEY === undefined) {
WillNilges marked this conversation as resolved.
Show resolved Hide resolved
console.error(
"S3 credentials not configured. I WILL NOT SAVE THIS SUBMISSION.",
);
return;
}

// Get the date to store this submission under (this is part of the path)
const submissionKey = joinRecord.submission_time
.replace(/[-:T]/g, "/")
.slice(0, 19);

// Create the path, or use the one provided.
key = key != "" ? key : `${this.S3_BASE_NAME}/${submissionKey}.json`;

let body = JSON.stringify(joinRecord);

const command = new PutObjectCommand({
Bucket: this.S3_BUCKET_NAME,
Key: key,
Body: body,
});

try {
const response = await this.s3Client.send(command);
console.log(response);
} catch (err) {
// Oof, guess we'll drop this on the floor.
console.error(err);
Comment on lines +78 to +79
Copy link
Member

Choose a reason for hiding this comment

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

Can we set up some kind of datadog alert for this at least?

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

We can definitely emit a metric. TBH I was trying to avoid setting up metrics for as long as possible 😂

throw err;
}

// Return the key later so we can update it.
return key;
}

// Gets the contents of a JoinRecord for testing
async get(key: string) {
const getObjectCommand = new GetObjectCommand({
Bucket: this.S3_BUCKET_NAME,
Key: key,
});
const getObjectResponse = await this.s3Client.send(getObjectCommand);

if (getObjectResponse.Body) {
const content = await this.streamToString(
getObjectResponse.Body as Readable,
);
return JSON.parse(content);
}
throw new Error("Could not get Record from S3");
}

// Decode S3 response
async streamToString(stream: Readable): Promise<string> {
return await new Promise((resolve, reject) => {
const chunks: Buffer[] = [];
stream.on("data", (chunk) => chunks.push(chunk));
stream.on("end", () => resolve(Buffer.concat(chunks).toString("utf-8")));
stream.on("error", reject);
});
}
}

const joinRecordS3 = new JoinRecordS3();

export async function saveJoinRecordToS3(
submission: JoinRecord,
key: string = "",
) {
return joinRecordS3.save(submission, key);
}

export async function getJoinRecordFromS3(key: string) {
return joinRecordS3.get(key);
}
10 changes: 10 additions & 0 deletions app/types.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
import { JoinFormValues } from "@/components/JoinForm/JoinForm";

type JoinRecord = JoinFormValues & {
submission_time: string;
code: string;
replayed: number;
install_number: number | null;
};

export type { JoinRecord };
50 changes: 42 additions & 8 deletions components/JoinForm/JoinForm.tsx
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
"use client";

import React, { useState } from "react";
import React, { useEffect, useState } from "react";
import { useForm, SubmitHandler } from "react-hook-form";
import styles from "./JoinForm.module.scss";
import { parsePhoneNumberFromString } from "libphonenumber-js";
Expand All @@ -13,9 +13,10 @@ import {
} from "@mui/material";
import { ToastContainer, toast } from "react-toastify";
import "react-toastify/dist/ReactToastify.css";
import { recordJoinFormSubmissionToS3 } from "@/app/data";
import { saveJoinRecordToS3 } from "@/app/join_record";
import { getMeshDBAPIEndpoint } from "@/app/endpoint";
import InfoConfirmationDialog from "../InfoConfirmation/InfoConfirmation";
import { JoinRecord } from "@/app/types";

type JoinFormValues = {
first_name: string;
Expand Down Expand Up @@ -86,6 +87,8 @@ export default function App() {
useState(false);
const [isSubmitted, setIsSubmitted] = useState(false);
const [isBadPhoneNumber, setIsBadPhoneNumber] = useState(false);
const [joinRecordKey, setJoinRecordKey] = useState("");

const isBeta = true;

// Store the values submitted by the user or returned by the server
Expand Down Expand Up @@ -150,20 +153,54 @@ export default function App() {
};

async function submitJoinFormToMeshDB(joinFormSubmission: JoinFormValues) {
console.debug(JSON.stringify(joinFormSubmission));

// Before we try anything else, submit to S3 for safety.
let record: JoinRecord = Object.assign(
structuredClone(joinFormSubmission),
{
submission_time: new Date().toISOString(),
code: "",
replayed: 0,
install_number: null,
},
) as JoinRecord;

saveJoinRecordToS3(record, joinRecordKey).then((key) => {
setJoinRecordKey(key as string);
});
WillNilges marked this conversation as resolved.
Show resolved Hide resolved

// Attempt to submit the join form
return fetch(`${await getMeshDBAPIEndpoint()}/api/v1/join/`, {
method: "POST",
body: JSON.stringify(joinFormSubmission),
})
.then(async (response) => {
WillNilges marked this conversation as resolved.
Show resolved Hide resolved
// Update the submission in S3 with the status code.
record.code = response.status.toString();

if (response.ok) {
console.debug("Join Form submitted successfully");
setIsLoading(false);
setIsSubmitted(true);

// Add the Install Number
const install_number = (await response.json()).install_number;
record.install_number = Number(install_number);

// Update Join Record
saveJoinRecordToS3(record, joinRecordKey).then((key) => {
setJoinRecordKey(key as string);
});

return;
}

// If we didn't succeed, then update the join record before handling
// the error
saveJoinRecordToS3(record, joinRecordKey).then((key) => {
setJoinRecordKey(key as string);
});

// If the response was not good, then get angry
throw response;
})
.catch(async (error) => {
Expand All @@ -174,11 +211,8 @@ export default function App() {
if (error.status == 409) {
let needsConfirmation: Array<ConfirmationField> = [];
const changedInfo = errorJson.changed_info;
console.log(joinFormSubmission);
console.log(errorJson.changed_info);

for (const key in joinFormSubmission) {
console.log(key);
if (
joinFormSubmission.hasOwnProperty(key) &&
changedInfo.hasOwnProperty(key)
Expand Down Expand Up @@ -209,7 +243,6 @@ export default function App() {

const onSubmit: SubmitHandler<JoinFormValues> = (data) => {
setIsLoading(true);
recordJoinFormSubmissionToS3(data);
data.trust_me_bro = false;
submitJoinFormToMeshDB(data);
};
Expand Down Expand Up @@ -401,6 +434,7 @@ export default function App() {
handleClickReject={handleClickReject}
handleClickCancel={handleClickCancel}
/>
<div data-testid="test-join-record-key" data-state={joinRecordKey}></div>
</>
);
}
16 changes: 15 additions & 1 deletion package-lock.json

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Loading
Loading