et ajoute le truc pour que les owner d'entreprise puisse a jour leur ent

softpython2884 · softpython2884 · commit 21ddb48c28ca · 2025-07-06T16:02:55.000Z
diff --git a/src/app/companies/[companyId]/page.tsx b/src/app/companies/[companyId]/page.tsx
@@ -3,7 +3,7 @@ import { notFound } from 'next/navigation';
 import { Card, CardHeader, CardTitle, CardDescription, CardContent } from "@/components/ui/card";
 import { Badge } from '@/components/ui/badge';
 import { Button } from '@/components/ui/button';
-import { ArrowLeft, Landmark, Users, DollarSign, LineChart, Briefcase, Percent, Package, Cpu, Settings, Server, Bitcoin } from 'lucide-react';
+import { ArrowLeft, Landmark, Users, DollarSign, LineChart, Briefcase, Percent, Package, Cpu, Settings, Server, Bitcoin, TrendingUp } from 'lucide-react';
 import Link from 'next/link';
 import { Table, TableBody, TableCell, TableHead, TableHeader, TableRow } from '@/components/ui/table';
 import { Avatar, AvatarFallback } from '@/components/ui/avatar';
@@ -14,6 +14,7 @@ import { AddCompanyCashDialog } from '@/components/add-company-cash-dialog';
 import { getRigById } from '@/lib/mining';
 import { ManageMembersDialog } from '@/components/manage-members-dialog';
 import { WithdrawCompanyCashDialog } from '@/components/withdraw-company-cash-dialog';
+import { ListCompanyButton } from '@/components/list-company-button';
 
 
 function getInitials(name: string) {
@@ -56,7 +57,14 @@ export default async function CompanyDetailPage({ params }: { params: { companyI
           </Link>
         </Button>
         <div className="flex-1">
-          <h1 className="text-2xl font-bold tracking-tight">{company.name}</h1>
+            <div className="flex items-center gap-3">
+                 <h1 className="text-2xl font-bold tracking-tight">{company.name} ({company.ticker})</h1>
+                 {company.isListed ? (
+                    <Badge variant="secondary">En Bourse</Badge>
+                ) : (
+                    <Badge variant="outline">Non Cotée</Badge>
+                )}
+            </div>
           <p className="text-muted-foreground">{company.description}</p>
         </div>
         <div className="flex items-center gap-2">
@@ -68,6 +76,7 @@ export default async function CompanyDetailPage({ params }: { params: { companyI
                     <WithdrawCompanyCashDialog companyId={company.id} companyCash={company.cash}>
                         <Button variant="outline">Retirer des fonds</Button>
                     </WithdrawCompanyCashDialog>
+                     {!company.isListed && <ListCompanyButton companyId={company.id} />}
                 </>
             )}
             <InvestDialog company={company}>
@@ -93,7 +102,7 @@ export default async function CompanyDetailPage({ params }: { params: { companyI
                   <LineChart className="h-4 w-4 text-muted-foreground" />
               </CardHeader>
               <CardContent>
-                  <div className="text-2xl font-bold">${company.sharePrice.toFixed(2)}</div>
+                  <div className="text-2xl font-bold">${company.sharePrice.toFixed(4)}</div>
                    <p className="text-xs text-muted-foreground">Prix par part de l'entreprise</p>
               </CardContent>
           </Card>
diff --git a/src/components/list-company-button.tsx b/src/components/list-company-button.tsx
@@ -0,0 +1,62 @@
+'use client';
+import { Button } from './ui/button';
+import { listCompanyOnMarket } from '@/lib/actions/companies';
+import { useToast } from '@/hooks/use-toast';
+import { useRouter } from 'next/navigation';
+import { useState } from 'react';
+import { Loader2, TrendingUp } from 'lucide-react';
+import {
+    AlertDialog,
+    AlertDialogAction,
+    AlertDialogCancel,
+    AlertDialogContent,
+    AlertDialogDescription,
+    AlertDialogFooter,
+    AlertDialogHeader,
+    AlertDialogTitle,
+    AlertDialogTrigger,
+  } from "@/components/ui/alert-dialog"
+
+export function ListCompanyButton({ companyId }: { companyId: number }) {
+    const { toast } = useToast();
+    const router = useRouter();
+    const [isLoading, setIsLoading] = useState(false);
+
+    const handleList = async () => {
+        setIsLoading(true);
+        const result = await listCompanyOnMarket(companyId);
+        if (result.error) {
+            toast({ variant: 'destructive', title: 'Erreur', description: result.error });
+        } else {
+            toast({ title: 'Succès !', description: result.success });
+            router.refresh();
+        }
+        setIsLoading(false);
+    }
+
+    return (
+        <AlertDialog>
+            <AlertDialogTrigger asChild>
+                <Button variant="outline">
+                    <TrendingUp className="mr-2 h-4 w-4" />
+                    Mettre en Bourse
+                </Button>
+            </AlertDialogTrigger>
+            <AlertDialogContent>
+                <AlertDialogHeader>
+                    <AlertDialogTitle>Mettre l'entreprise en bourse ?</AlertDialogTitle>
+                    <AlertDialogDescription>
+                        Cette action est irréversible. Une fois cotée, les actions de votre entreprise apparaîtront dans la Salle des Marchés et pourront être échangées par tous les joueurs. Le prix de l'action sera déterminé par la valeur totale de l'entreprise (trésorerie + actifs).
+                    </AlertDialogDescription>
+                </AlertDialogHeader>
+                <AlertDialogFooter>
+                    <AlertDialogCancel>Annuler</AlertDialogCancel>
+                    <AlertDialogAction onClick={handleList} disabled={isLoading}>
+                         {isLoading && <Loader2 className="mr-2 h-4 w-4 animate-spin" />}
+                        Confirmer la Cotation
+                    </AlertDialogAction>
+                </AlertDialogFooter>
+            </AlertDialogContent>
+        </AlertDialog>
+    )
+}
diff --git a/src/context/market-data-context.tsx b/src/context/market-data-context.tsx
@@ -80,6 +80,12 @@ export const MarketDataProvider = ({ children }: { children: ReactNode }) => {
                 
                 for (const ticker in newAssets) {
                     const asset = { ...newAssets[ticker] };
+
+                    // Prevent simulation for company shares as their price is calculated
+                    if (asset.type === 'Company Share') {
+                        continue;
+                    }
+
                     const initialPrice = initialAssets[ticker]?.price;
                     if (!initialPrice) continue;
 
diff --git a/src/lib/actions/assets.ts b/src/lib/actions/assets.ts
@@ -1,7 +1,7 @@
 'use server';
 
 import { db } from '@/lib/db';
-import { assets as assetsSchema } from '@/lib/db/schema';
+import { assets as assetsSchema, companies as companiesSchema } from '@/lib/db/schema';
 import { assets as initialAssets } from '@/lib/assets';
 import { eq } from 'drizzle-orm';
 import { revalidatePath } from 'next/cache';
@@ -21,7 +21,40 @@ export async function getAssets() {
             return seededAssets.map(a => ({...a, price: parseFloat(a.price)}));
         }
 
-        return assetsInDb.map(a => ({...a, price: parseFloat(a.price)}));
+        const baseAssets = assetsInDb.map(a => ({...a, price: parseFloat(a.price)}));
+
+        const listedCompanies = await db.query.companies.findMany({
+            where: eq(companiesSchema.isListed, true),
+        });
+
+        const companyAssets = listedCompanies.map(c => {
+            const sharePrice = parseFloat(c.sharePrice);
+            const totalShares = parseFloat(c.totalShares);
+            const marketCap = sharePrice * totalShares;
+
+            let marketCapString = `$${marketCap.toLocaleString(undefined, {maximumFractionDigits: 0})}`;
+            if (marketCap >= 1e12) {
+                marketCapString = `$${(marketCap / 1e12).toFixed(2)}T`;
+            } else if (marketCap >= 1e9) {
+                marketCapString = `$${(marketCap / 1e9).toFixed(2)}B`;
+            } else if (marketCap >= 1e6) {
+                marketCapString = `$${(marketCap / 1e6).toFixed(2)}M`;
+            }
+
+
+            return {
+                ticker: c.ticker,
+                name: c.name,
+                description: c.description,
+                type: 'Company Share',
+                price: sharePrice,
+                change24h: '+0.00%', // Placeholder, as company value change is not tracked over 24h yet
+                marketCap: marketCapString,
+            }
+        });
+
+        return [...baseAssets, ...companyAssets];
+
     } catch (error) {
         console.error("Error getting assets:", error);
         return [];
@@ -46,7 +79,7 @@ export async function updatePriceFromTrade(ticker: string, tradeValue: number) {
             where: eq(assetsSchema.ticker, ticker),
         });
 
-        if (!asset) return;
+        if (!asset || asset.type === 'Company Share') return;
 
         const marketCap = parseMarketCap(asset.marketCap);
         if (marketCap === 0) return;
diff --git a/src/lib/actions/companies.ts b/src/lib/actions/companies.ts
@@ -10,9 +10,9 @@ import { updatePriceFromTrade } from './assets';
 import { getRigById } from '@/lib/mining';
 
 const createCompanySchema = z.object({
-  name: z.string().min(3, "Le nom doit faire au moins 3 caractères.").max(50),
-  industry: z.string().min(3, "L'industrie doit faire au moins 3 caractères.").max(50),
-  description: z.string().min(10, "La description doit faire au moins 10 caractères.").max(200),
+  name: z.string().min(3, "Le nom doit faire au moins 3 caractères.").max(50, "Le nom ne doit pas dépasser 50 caractères."),
+  industry: z.string().min(3, "L'industrie doit faire au moins 3 caractères.").max(50, "L'industrie ne doit pas dépasser 50 caractères."),
+  description: z.string().min(10, "La description doit faire au moins 10 caractères.").max(200, "La description ne doit pas dépasser 200 caractères."),
 });
 
 export async function createCompany(values: z.infer<typeof createCompanySchema>): Promise<{ success?: string; error?: string }> {
@@ -28,6 +28,7 @@ export async function createCompany(values: z.infer<typeof createCompanySchema>)
     return { error: "Données invalides." };
   }
   const { name, industry, description } = validatedFields.data;
+  const ticker = name.replace(/[^a-zA-Z0-9]/g, '').substring(0, 4).toUpperCase();
 
   try {
     const result = await db.transaction(async (tx) => {
@@ -37,6 +38,11 @@ export async function createCompany(values: z.infer<typeof createCompanySchema>)
       const userCash = parseFloat(user.cash);
       if (userCash < creationCost) throw new Error(`Fonds insuffisants. La création d'une entreprise coûte ${creationCost.toLocaleString()}$.`);
 
+      const existingTicker = await tx.query.companies.findFirst({ where: eq(companies.ticker, ticker) });
+      if (existingTicker) {
+        throw new Error("Une entreprise avec un ticker similaire existe déjà. Veuillez choisir un nom légèrement différent.");
+      }
+
       // Deduct cost from user
       const newUserCash = userCash - creationCost;
       await tx.update(users).set({ cash: newUserCash.toFixed(2) }).where(eq(users.id, session.id));
@@ -49,7 +55,8 @@ export async function createCompany(values: z.infer<typeof createCompanySchema>)
         creatorId: session.id,
         cash: creationCost.toFixed(2),
         totalShares: '1000.00000000',
-        sharePrice: '1.00'
+        sharePrice: '1.00',
+        ticker: ticker,
       }).returning();
 
       // Add the creator as the CEO
@@ -68,8 +75,8 @@ export async function createCompany(values: z.infer<typeof createCompanySchema>)
     revalidatePath('/');
     return result;
   } catch (error: any) {
-    // Check for unique constraint violation
-    if (error?.code === '23505') {
+    // Check for unique constraint violation on name
+    if (error?.code === '23505' && error.constraint === 'companies_name_key') {
         return { error: "Une entreprise avec ce nom existe déjà." };
     }
     console.error("Error creating company:", error);
@@ -237,6 +244,14 @@ export async function getCompanyById(companyId: number) {
     const totalShares = parseFloat(company.totalShares);
     const sharePrice = totalShares > 0 ? companyValue / totalShares : parseFloat(company.sharePrice);
 
+    // Update the stored share price to reflect the latest calculation
+    if (Math.abs(sharePrice - parseFloat(company.sharePrice)) > 0.00001) { // Only update if there's a meaningful change
+        await db.update(companies)
+            .set({ sharePrice: sharePrice.toString() })
+            .where(eq(companies.id, company.id));
+    }
+
+
     return {
       ...company,
       cash: companyCash,
@@ -701,3 +716,30 @@ export async function removeMemberFromCompany(companyId: number, memberIdToRemov
         return { error: error.message || "Une erreur est survenue." };
     }
 }
+
+
+export async function listCompanyOnMarket(companyId: number): Promise<{ success?: string; error?: string }> {
+    const session = await getSession();
+    if (!session?.id) {
+      return { error: 'Non autorisé.' };
+    }
+  
+    try {
+      const member = await db.query.companyMembers.findFirst({
+        where: and(eq(companyMembers.companyId, companyId), eq(companyMembers.userId, session.id)),
+      });
+  
+      if (!member || member.role !== 'ceo') {
+        return { error: 'Seul le PDG peut mettre une entreprise en bourse.' };
+      }
+  
+      await db.update(companies).set({ isListed: true }).where(eq(companies.id, companyId));
+  
+      revalidatePath('/trading');
+      revalidatePath(`/companies/${companyId}`);
+  
+      return { success: 'Entreprise mise en bourse avec succès !' };
+    } catch (error: any) {
+      return { error: error.message || "Une erreur est survenue." };
+    }
+  }
diff --git a/src/lib/db/schema.ts b/src/lib/db/schema.ts
@@ -8,6 +8,7 @@ import {
   integer,
   uniqueIndex,
   index,
+  boolean,
 } from 'drizzle-orm/pg-core';
 import { relations, desc } from 'drizzle-orm';
 
@@ -183,12 +184,14 @@ export const userMiningRigsRelations = relations(userMiningRigs, ({ one }) => ({
 export const companies = pgTable('companies', {
   id: serial('id').primaryKey(),
   name: varchar('name', { length: 256 }).notNull().unique(),
+  ticker: varchar('ticker', { length: 10 }).notNull().unique(),
   industry: varchar('industry', { length: 100 }).notNull(),
   description: text('description').notNull(),
   cash: numeric('cash', { precision: 15, scale: 2 }).default('0.00').notNull(),
   creatorId: integer('creator_id').notNull().references(() => users.id, { onDelete: 'cascade' }),
-  sharePrice: numeric('share_price', { precision: 10, scale: 2 }).default('1.00').notNull(),
+  sharePrice: numeric('share_price', { precision: 20, scale: 8 }).default('1.00').notNull(),
   totalShares: numeric('total_shares', { precision: 20, scale: 8 }).default('1000.00').notNull(),
+  isListed: boolean('is_listed').default(false).notNull(),
   unclaimedBtc: numeric('unclaimed_btc', { precision: 18, scale: 8 }).default('0').notNull(),
   lastMiningUpdateAt: timestamp('last_mining_update_at', { withTimezone: true }).defaultNow().notNull(),
   createdAt: timestamp('created_at', { withTimezone: true }).defaultNow().notNull(),
