Merge pull request #47 from Nutlope/add-locations

Add countries to DB

.gitignore

@@ -36,3 +36,6 @@ yarn-error.log*
 *.tsbuildinfo
 next-env.d.ts
 .env
+
+# Scripts
+getUsersEmails.ts

pages/api/remaining.ts

@@ -2,6 +2,7 @@ import type { NextApiRequest, NextApiResponse } from "next";
 import { getServerSession } from "next-auth/next";
 import { authOptions } from "./auth/[...nextauth]";
 import prisma from "../../lib/prismadb";
+import requestIp from "request-ip";
 
 export default async function handler(
   req: NextApiRequest,
@@ -21,8 +22,25 @@ export default async function handler(
     },
     select: {
       credits: true,
+      location: true,
     },
   });
 
+  if (!user?.location) {
+    const ip = requestIp.getClientIp(req);
+    const location = await fetch(
+      `http://api.ipstack.com/${ip}?access_key=${process.env.IPSTACK_API_KEY}`
+    ).then((res) => res.json());
+
+    await prisma.user.update({
+      where: {
+        email: session.user.email!,
+      },
+      data: {
+        location: location.country_code,
+      },
+    });
+  }
+
   return res.status(200).json({ remainingGenerations: user?.credits });
 }

prisma/migrations/20230325035657_location/migration.sql

@@ -0,0 +1,9 @@
+/*
+  Warnings:
+
+  - You are about to drop the column `boughtCredits` on the `User` table. All the data in the column will be lost.
+
+*/
+-- AlterTable
+ALTER TABLE "User" DROP COLUMN "boughtCredits",
+ADD COLUMN     "location" TEXT;

prisma/schema.prisma

@@ -42,7 +42,7 @@ model User {
   emailVerified DateTime?
   image         String?
   credits       Int        @default(3)
-  boughtCredits Int        @default(0) // remove boughtCredits
+  location      String?
   accounts      Account[]
   sessions      Session[]
   rooms         Room[]