ai-code-assistant.js

/**
 * Rahyana AI API - دستیار کدنویسی هوش مصنوعی برای توسعه‌دهندگان
 * 
 * این پروژه یک دستیار کامل و آماده تولید برای برنامه‌نویسان با قابلیت‌های پیشرفته:
 * - تولید کد در زبان‌های مختلف (JavaScript, Python, Java, C#, Go)
 * - بررسی و اصلاح کد با تحلیل کیفیت
 * - توضیح کد پیچیده و مستندسازی
 * - بهینه‌سازی عملکرد و کارایی
 * - تبدیل کد بین زبان‌ها و پلتفرم‌ها
 * - تولید تست‌های واحد (Jest, Mocha, Pytest)
 * - تولید مستندات API و README
 * - تولید API های RESTful و GraphQL
 * - تولید کامپوننت‌های React
 * - تولید schema های دیتابیس
 * - بررسی امنیت کد و آسیب‌پذیری‌ها
 * 
 * 🚀 ویژگی‌های کلیدی:
 * - پشتیبانی از چندین زبان برنامه‌نویسی
 * - تحلیل کد پیشرفته و هوشمند
 * - تولید کد بهینه و قابل نگهداری
 * - مستندسازی خودکار
 * - بررسی امنیت و بهترین شیوه‌ها
 * 
 * 💼 مناسب برای:
 * - توسعه‌دهندگان حرفه‌ای
 * - دانشجویان برنامه‌نویسی
 * - تیم‌های توسعه نرم‌افزار
 * - پروژه‌های آموزشی و یادگیری
 * - شرکت‌های فناوری
 */

// Configuration - supports environment variable or placeholder
const API_KEY = process.env.RAHYANA_API_KEY || 'YOUR_API_KEY_HERE'; // کلید API خود را اینجا قرار دهید یا از متغیر محیطی RAHYANA_API_KEY استفاده کنید
const BASE_URL = process.env.RAHYANA_BASE_URL || 'https://rahyana.ir/api/v1';

class AICodeAssistant {
  constructor(apiKey, baseUrl) {
    this.apiKey = apiKey;
    this.baseUrl = baseUrl;
  }

  async generateCode(prompt, language = 'javascript', options = {}) {
    try {
      const systemPrompt = `شما یک برنامه‌نویس متخصص هستید. 
      کد ${language} تمیز، بهینه و قابل فهم بنویسید.
      از بهترین شیوه‌های برنامه‌نویسی استفاده کنید.
      کامنت‌های مفید اضافه کنید.`;

      const response = await fetch(`${this.baseUrl}/chat/completions`, {
        method: 'POST',
        headers: {
          'Authorization': `Bearer ${this.apiKey}`,
          'Content-Type': 'application/json',
          'HTTP-Referer': 'https://code-assistant-app.com',
          'X-Title': 'AI Code Assistant'
        },
        body: JSON.stringify({
          model: 'openai/gpt-4o', // مدل GPT-4o برای کدنویسی
          messages: [
            { role: 'system', content: systemPrompt },
            { role: 'user', content: prompt }
          ],
          stream: false,
          temperature: options.temperature || 0.1, // دقت بالا برای کد
          max_tokens: options.maxTokens || 2000
        })
      });

      if (!response.ok) {
        throw new Error(`خطای HTTP! وضعیت: ${response.status}`);
      }

      const data = await response.json();
      return data.choices[0].message.content;

    } catch (error) {
      console.error('خطا در تولید کد:', error);
      throw error;
    }
  }

  // بررسی و اصلاح کد
  async reviewCode(code, language = 'javascript') {
    const prompt = `
    کد ${language} زیر را بررسی کنید و اصلاحات لازم را اعمال کنید:
    
    \`\`\`${language}
    ${code}
    \`\`\`
    
    موارد بررسی:
    1. خطاهای نحوی و منطقی
    2. بهینه‌سازی عملکرد
    3. امنیت کد
    4. خوانایی و ساختار
    5. بهترین شیوه‌های برنامه‌نویسی
    
    کد اصلاح شده را با توضیحات ارائه دهید.
    `;

    return await this.generateCode(prompt, language);
  }

  // توضیح کد
  async explainCode(code, language = 'javascript') {
    const prompt = `
    کد ${language} زیر را به زبان ساده توضیح دهید:
    
    \`\`\`${language}
    ${code}
    \`\`\`
    
    توضیحات شامل:
    1. هدف کلی کد
    2. توضیح خط به خط
    3. متغیرها و توابع
    4. منطق برنامه
    5. مثال‌های عملی
    `;

    return await this.generateCode(prompt, language);
  }

  // تبدیل کد بین زبان‌ها
  async convertCode(code, fromLanguage, toLanguage) {
    const prompt = `
    کد ${fromLanguage} زیر را به ${toLanguage} تبدیل کنید:
    
    \`\`\`${fromLanguage}
    ${code}
    \`\`\`
    
    الزامات:
    1. حفظ منطق اصلی
    2. استفاده از ویژگی‌های خاص ${toLanguage}
    3. کد تمیز و بهینه
    4. کامنت‌های مناسب
    `;

    return await this.generateCode(prompt, toLanguage);
  }

  // تولید تست‌های واحد
  async generateUnitTests(code, language = 'javascript', testFramework = 'jest') {
    const prompt = `
    برای کد ${language} زیر، تست‌های واحد ${testFramework} بنویسید:
    
    \`\`\`${language}
    ${code}
    \`\`\`
    
    الزامات:
    1. پوشش کامل توابع
    2. تست موارد عادی و استثنایی
    3. استفاده از mock ها در صورت نیاز
    4. نام‌گذاری واضح تست‌ها
    5. توضیحات مناسب
    `;

    return await this.generateCode(prompt, language);
  }

  // تولید مستندات
  async generateDocumentation(code, language = 'javascript') {
    const prompt = `
    برای کد ${language} زیر، مستندات کامل تولید کنید:
    
    \`\`\`${language}
    ${code}
    \`\`\`
    
    شامل:
    1. توضیح کلی پروژه
    2. مستندات توابع و کلاس‌ها
    3. پارامترها و مقادیر بازگشتی
    4. مثال‌های استفاده
    5. نکات مهم
    `;

    return await this.generateCode(prompt, 'markdown');
  }

  // بهینه‌سازی عملکرد
  async optimizeCode(code, language = 'javascript') {
    const prompt = `
    کد ${language} زیر را از نظر عملکرد بهینه‌سازی کنید:
    
    \`\`\`${language}
    ${code}
    \`\`\`
    
    موارد بهینه‌سازی:
    1. پیچیدگی زمانی و مکانی
    2. استفاده بهینه از حافظه
    3. کاهش تعداد حلقه‌ها
    4. استفاده از الگوریتم‌های بهتر
    5. حذف کدهای غیرضروری
    `;

    return await this.generateCode(prompt, language);
  }

  // تولید API
  async generateAPI(specification, language = 'javascript', framework = 'express') {
    const prompt = `
    بر اساس مشخصات زیر، API ${framework} در ${language} تولید کنید:
    
    ${specification}
    
    الزامات:
    1. ساختار RESTful
    2. مدیریت خطاها
    3. اعتبارسنجی ورودی
    4. مستندات API
    5. کد تمیز و قابل نگهداری
    `;

    return await this.generateCode(prompt, language);
  }

  // تولید کامپوننت React
  async generateReactComponent(description, props = []) {
    const prompt = `
    کامپوننت React زیر را تولید کنید:
    
    توضیحات: ${description}
    Props: ${props.join(', ') || 'هیچ'}
    
    الزامات:
    1. استفاده از Hooks
    2. TypeScript (اختیاری)
    3. استایل‌دهی با CSS Modules یا styled-components
    4. مدیریت state
    5. کامپوننت قابل استفاده مجدد
    `;

    return await this.generateCode(prompt, 'jsx');
  }

  // تولید دیتابیس schema
  async generateDatabaseSchema(description, dbType = 'postgresql') {
    const prompt = `
    بر اساس توضیحات زیر، schema دیتابیس ${dbType} تولید کنید:
    
    ${description}
    
    الزامات:
    1. جداول و روابط مناسب
    2. کلیدهای اصلی و خارجی
    3. ایندکس‌های بهینه
    4. محدودیت‌های داده
    5. migration ها
    `;

    return await this.generateCode(prompt, 'sql');
  }

  // بررسی امنیت کد
  async securityAudit(code, language = 'javascript') {
    const prompt = `
    کد ${language} زیر را از نظر امنیت بررسی کنید:
    
    \`\`\`${language}
    ${code}
    \`\`\`
    
    موارد بررسی:
    1. آسیب‌پذیری‌های امنیتی
    2. تزریق SQL
    3. XSS
    4. CSRF
    5. مدیریت احراز هویت
    6. رمزگذاری داده‌ها
    7. اعتبارسنجی ورودی
    `;

    return await this.generateCode(prompt, 'markdown');
  }
}

// مثال استفاده
async function demonstrateCodeAssistant() {
  console.log('💻 شروع دستیار کدنویسی...\n');

  const assistant = new AICodeAssistant(API_KEY, BASE_URL);

  try {
    // تولید کد ساده
    console.log('🔨 تولید تابع محاسبه فاکتوریل:');
    const factorialCode = await assistant.generateCode(
      'تابعی برای محاسبه فاکتوریل یک عدد بنویسید',
      'javascript'
    );
    console.log(factorialCode + '\n');

    // بررسی کد
    console.log('🔍 بررسی کد:');
    const buggyCode = `
    function add(a, b) {
      return a + b;
    }
    
    function multiply(a, b) {
      return a * b;
    }
    
    function calculate(a, b, operation) {
      if (operation === 'add') {
        return add(a, b);
      } else if (operation === 'multiply') {
        return multiply(a, b);
      }
    }
    `;
    
    const reviewedCode = await assistant.reviewCode(buggyCode, 'javascript');
    console.log(reviewedCode + '\n');

    // توضیح کد
    console.log('📖 توضیح کد:');
    const complexCode = `
    const fibonacci = (n) => {
      if (n <= 1) return n;
      return fibonacci(n - 1) + fibonacci(n - 2);
    };
    `;
    
    const explanation = await assistant.explainCode(complexCode, 'javascript');
    console.log(explanation + '\n');

    // تولید تست
    console.log('🧪 تولید تست‌های واحد:');
    const testCode = `
    function isEven(num) {
      return num % 2 === 0;
    }
    `;
    
    const unitTests = await assistant.generateUnitTests(testCode, 'javascript', 'jest');
    console.log(unitTests + '\n');

    // تولید API
    console.log('🌐 تولید API:');
    const apiSpec = `
    API برای مدیریت کاربران:
    - GET /users - لیست کاربران
    - POST /users - ایجاد کاربر جدید
    - GET /users/:id - دریافت کاربر
    - PUT /users/:id - به‌روزرسانی کاربر
    - DELETE /users/:id - حذف کاربر
    `;
    
    const apiCode = await assistant.generateAPI(apiSpec, 'javascript', 'express');
    console.log(apiCode.substring(0, 500) + '...\n');

  } catch (error) {
    console.error('خطا در اجرای دستیار کدنویسی:', error);
  }
}

// اجرای مثال
if (import.meta.url === `file://${process.argv[1]}`) {
  demonstrateCodeAssistant();
}

export default AICodeAssistant;