draftify-cli 1.0.52 → 1.0.56

package/dist/index.js

@@ -11,6 +11,7 @@ const refactor_1 = require("./commands/refactor");
 const repl_1 = require("./repl");
 const ui_1 = require("./utils/ui");
 const config_1 = require("./utils/config");
+const api_1 = require("./utils/api");
 const program = new commander_1.Command();
 program
     .name("draftify")
@@ -87,14 +88,38 @@ async function checkVersion() {
     await checkVersion();
     // Custom logic for when `draftify` is run without any arguments
     if (process.argv.slice(2).length === 0) {
-        const token = (0, config_1.getToken)();
-        if (!token) {
-            // No token, run interactive login flow
-            (0, login_1.loginCommand)().then((username) => (0, repl_1.startRepl)(username)).catch(() => process.exit(1));
+        let isValid = false;
+        let username = "Developer";
+        if ((0, config_1.getToken)()) {
+            try {
+                ui_1.ui.info("Hitelesítés ellenőrzése...");
+                const profile = await (0, api_1.getUserProfile)();
+                if (profile) {
+                    isValid = true;
+                    username = profile.username;
+                }
+                else {
+                    // Network error but not 401, we allow them in
+                    isValid = true;
+                }
+            }
+            catch (e) {
+                if (e.message === "UNAUTHORIZED") {
+                    (0, config_1.saveConfig)({ token: null });
+                    isValid = false;
+                }
+                else {
+                    isValid = true;
+                }
+            }
+        }
+        if (!isValid) {
+            // No valid token, run interactive login flow
+            (0, login_1.loginCommand)().then((uname) => (0, repl_1.startRepl)(uname)).catch(() => process.exit(1));
         }
         else {
-            // Token exists, start REPL
-            (0, repl_1.startRepl)().catch((err) => {
+            // Token exists and is valid, start REPL
+            (0, repl_1.startRepl)(username).catch((err) => {
                 ui_1.ui.error(`REPL crashed: ${err.message}`);
                 process.exit(1);
             });

package/dist/utils/api.js

@@ -6,7 +6,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.refactorCodeApi = refactorCodeApi;
 exports.getUserProfile = getUserProfile;
 const config_1 = require("./config");
-const genai_1 = require("@google/genai");
+const vertexai_1 = require("@google-cloud/vertexai");
 const dotenv_1 = __importDefault(require("dotenv"));
 dotenv_1.default.config();
 async function refactorCodeApi(fileName, code, instruction, history, modelName, skillsData, thinkingLevel, onChunk, abortSignal) {
@@ -14,7 +14,7 @@ async function refactorCodeApi(fileName, code, instruction, history, modelName,
     // 1. HELYI VÉGREHAJTÁS (Ha van GEMINI_API_KEY a .env-ben, közvetlenül a Geminit hívjuk)
     // Ez tökéletes a helyi teszteléshez, mielőtt a backendre (Vercel) kerül a kód.
     if (apiKey) {
-        const ai = new genai_1.GoogleGenAI({ apiKey: apiKey });
+        const ai = new vertexai_1.VertexAI({ project: 'draftify-prod', location: 'us-central1' });
         // Modell kiválasztása a kívánt logika alapján
         let geminiModel = 'gemini-3.5-flash'; // Alapértelmezett (Medium, Low, Adaptive)
         if (thinkingLevel === 'High') {
@@ -160,23 +160,24 @@ Throughout, balance deep technical proficiency with an accessible, friendly, wel
                 abortSignal.addEventListener("abort", onAbort);
             }
             try {
-                const responseStream = await ai.models.generateContentStream({
+                const generativeModel = ai.getGenerativeModel({
                     model: geminiModel,
-                    contents: contents,
-                    config: {
-                        systemInstruction: systemInstruction
-                    }
+                    systemInstruction: { role: 'system', parts: [{ text: systemInstruction }] }
+                });
+                const responseStream = await generativeModel.generateContentStream({
+                    contents: contents
                 });
                 let tempResult = "";
-                for await (const chunk of responseStream) {
+                for await (const chunk of responseStream.stream) {
                     if (abortSignal?.aborted) {
                         reject(new Error("Aborted"));
                         return;
                     }
-                    if (chunk.text) {
-                        tempResult += chunk.text;
+                    const text = chunk.candidates?.[0]?.content?.parts?.[0]?.text || "";
+                    if (text) {
+                        tempResult += text;
                         if (onChunk)
-                            onChunk(chunk.text);
+                            onChunk(text);
                     }
                 }
                 resolve(tempResult);
@@ -276,20 +277,25 @@ Throughout, balance deep technical proficiency with an accessible, friendly, wel
 async function getUserProfile() {
     const token = (0, config_1.getToken)();
     if (!token)
-        return null;
+        throw new Error("UNAUTHORIZED");
     const apiUrl = (0, config_1.getApiUrl)();
     try {
         const response = await fetch(apiUrl, {
             method: "GET",
             headers: { Authorization: `Bearer ${token}` }
         });
+        if (response.status === 401 || response.status === 403) {
+            throw new Error("UNAUTHORIZED");
+        }
         if (response.ok) {
             const data = await response.json();
             return { username: data.username, plan: data.plan };
         }
     }
     catch (e) {
-        // Ignore errors
+        if (e.message === "UNAUTHORIZED")
+            throw e;
+        // Ignore network errors
     }
     return null;
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "draftify-cli",
-  "version": "1.0.52",
+  "version": "1.0.56",
   "description": "Draftify AI CLI tool",
   "main": "dist/index.js",
   "bin": {
@@ -24,6 +24,7 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
+    "@google-cloud/vertexai": "^1.12.0",
     "@google/genai": "^2.8.0",
     "ansi-escapes": "^7.3.0",
     "axios": "^1.7.2",