From 6b10288ea133002693404860a385932e01634d21 Mon Sep 17 00:00:00 2001
From: Tau <tau@nowhere>
Date: Sun, 21 Apr 2019 23:55:35 -0400
Subject: [PATCH] Save time attack grade

---
 src/idz/decoder/saveTimeAttack.ts | 2 +-
 src/idz/encoder/loadProfile2.ts   | 2 ++
 src/idz/model/timeAttack.ts       | 1 +
 src/idz/request/saveTimeAttack.ts | 1 -
 4 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/idz/decoder/saveTimeAttack.ts b/src/idz/decoder/saveTimeAttack.ts
index 27574cd..c206d46 100644
--- a/src/idz/decoder/saveTimeAttack.ts
+++ b/src/idz/decoder/saveTimeAttack.ts
@@ -21,6 +21,7 @@ export function saveTimeAttack(buf: Buffer): SaveTimeAttackRequest {
         buf.readUInt32LE(0x0028),
         buf.readUInt32LE(0x002c),
       ],
+      grade: buf.readUInt8(0x0062),
     },
     field_0002: buf.readUInt16LE(0x0002),
     field_0008: buf.readUInt32LE(0x0008),
@@ -30,6 +31,5 @@ export function saveTimeAttack(buf: Buffer): SaveTimeAttackRequest {
     field_005D: buf.readUInt8(0x005d),
     field_005E: buf.readUInt16LE(0x005e),
     field_0060: buf.readUInt16LE(0x0060),
-    field_0062: buf.readUInt8(0x0062),
   };
 }
diff --git a/src/idz/encoder/loadProfile2.ts b/src/idz/encoder/loadProfile2.ts
index 3f6f235..02cadb3 100644
--- a/src/idz/encoder/loadProfile2.ts
+++ b/src/idz/encoder/loadProfile2.ts
@@ -21,9 +21,11 @@ export function loadProfile2(res: LoadProfileResponse2) {
       0x00e4 + courseId * 4
     );
 
+    buf.writeUInt16LE(0, 0x0560 + 2 * courseId); // ???
     buf.writeUInt16LE(0xffff, 0x0164 + 2 * courseId); // National rank
     buf.writeUInt32LE(score.totalMsec, 0x04e0 + 4 * courseId);
     buf.writeUInt8(score.flags, 0x05a0 + courseId);
+    buf.writeUInt8(score.grade, 0x0680 + courseId);
 
     for (let i = 0; i < 3; i++) {
       buf.writeUInt16LE(
diff --git a/src/idz/model/timeAttack.ts b/src/idz/model/timeAttack.ts
index c820c86..63ff2d0 100644
--- a/src/idz/model/timeAttack.ts
+++ b/src/idz/model/timeAttack.ts
@@ -4,4 +4,5 @@ export interface TimeAttackScore {
   flags: number;
   totalMsec: number;
   stageMsec: number[];
+  grade: number;
 }
diff --git a/src/idz/request/saveTimeAttack.ts b/src/idz/request/saveTimeAttack.ts
index 9ecc9cc..febc9bb 100644
--- a/src/idz/request/saveTimeAttack.ts
+++ b/src/idz/request/saveTimeAttack.ts
@@ -15,5 +15,4 @@ export interface SaveTimeAttackRequest {
   field_005D: number; // u8
   field_005E: number; // u16
   field_0060: number; // u16
-  field_0062: number; // u8
 }