Описание изменений

2025-03-07 21:49:36 +03:00
parent 5a074ad7e8
commit 4abc48a7be
10 changed files with 212 additions and 398 deletions
--- a/backend/routes/contracts.js
+++ b/backend/routes/contracts.js
@@ -7,10 +7,11 @@ router.get('/', (req, res) => {
  res.json({
    message: 'Contracts API endpoint',
    contracts: [
-      {
-        name: 'AccessToken',
-        address: process.env.ACCESS_TOKEN_ADDRESS,
-      },
+      // Удаляем AccessToken
+      // {
+      //   name: 'AccessToken',
+      //   address: process.env.ACCESS_TOKEN_ADDRESS,
+      // },
    ],
  });
 });
@@ -20,11 +21,12 @@ router.get('/details', requireRole('ADMIN'), (req, res) => {
  res.json({
    message: 'Contract details endpoint',
    contracts: [
-      {
-        name: 'AccessToken',
-        address: process.env.ACCESS_TOKEN_ADDRESS,
-        network: process.env.ETHEREUM_NETWORK_URL.includes('sepolia') ? 'Sepolia' : 'Unknown',
-      },
+      // Удаляем AccessToken
+      // {
+      //   name: 'AccessToken',
+      //   address: process.env.ACCESS_TOKEN_ADDRESS,
+      //   network: process.env.ETHEREUM_NETWORK_URL.includes('sepolia') ? 'Sepolia' : 'Unknown',
+      // },
    ],
  });
 });
--- a/backend/services/auth-service.js
+++ b/backend/services/auth-service.js
@@ -1,44 +1,36 @@
 const db = require('../db');
-const { getContract } = require('../utils/contracts');
 const logger = require('../utils/logger');
 const { ethers } = require('ethers');
-const path = require('path');

-// Инициализируем провайдер
-const provider = new ethers.JsonRpcProvider(process.env.RPC_URL);
+// В начале файла auth-service.js
+const getProvider = (network) => {
+  const primaryUrl = process.env[`RPC_URL_${network.toUpperCase()}`];
+  const backupUrls = {
+    eth: 'https://eth-mainnet.public.blastapi.io',
+    polygon: 'https://polygon-rpc.com',
+    bsc: 'https://bsc-dataseed.binance.org',
+    arbitrum: 'https://arb1.arbitrum.io/rpc'
+  };
  
-const contractsDir = path.join(__dirname, '../artifacts/contracts/AccessToken.sol');
+  try {
+    return new ethers.JsonRpcProvider(primaryUrl);
+  } catch (error) {
+    logger.warn(`Failed to connect to primary URL for ${network}, using backup`);
+    return new ethers.JsonRpcProvider(backupUrls[network]);
+  }
+};
+
+const providers = {
+  eth: getProvider('eth'),
+  polygon: getProvider('polygon'),
+  bsc: getProvider('bsc'),
+  arbitrum: getProvider('arbitrum')
+};

 /**
 * Сервис для работы с аутентификацией и авторизацией
 */
 class AuthService {
-  /**
-   * Проверяет наличие токена администратора для кошелька
-   * @param {string} walletAddress - Адрес кошелька
-   * @returns {Promise<boolean>} - Имеет ли кошелек токен администратора
-   */
-  async checkAdminToken(walletAddress) {
-    try {
-      if (!walletAddress) {
-        logger.error('Wallet address is undefined');
-        return false;
-      }
-
-      // Получаем контракт AccessToken
-      const accessToken = await getContract('AccessToken');
-      
-      // Проверяем роль пользователя
-      const role = await accessToken.checkRole(walletAddress);
-      
-      // 0 = ADMIN
-      return role === 0;
-    } catch (error) {
-      logger.error(`Error checking admin token: ${error.message}`);
-      return false;
-    }
-  }
-
  /**
   * Проверяет наличие токенов на кошельке и обновляет роль
   * @param {string} walletAddress - Адрес кошелька
@@ -60,8 +52,8 @@ class AuthService {
      
      const userId = userResult.rows[0].id;
      
-      // Проверяем наличие токена администратора
-      const isAdmin = await this.checkAdminToken(walletAddress);
+      // Проверяем наличие токенов на кошельке
+      const isAdmin = await this.checkAdminTokens(walletAddress);
      
      // Обновляем роль в базе данных
      await this.updateUserRole(userId, isAdmin ? 'admin' : 'user');
@@ -75,6 +67,62 @@ class AuthService {
    }
  }

+  /**
+   * Проверяет наличие токенов на кошельке
+   * @param {string} walletAddress - Адрес кошелька
+   * @returns {Promise<boolean>} - Имеет ли кошелек токены
+   */
+  async checkAdminTokens(walletAddress) {
+    try {
+      const tokenContracts = [
+        { address: "0xd95a45fc46a7300e6022885afec3d618d7d3f27c", network: "eth" },     // Ethereum
+        { address: "0x1d47f12ffA279BFE59Ab16d56fBb10d89AECdD5D", network: "bsc" },     // Binance Smart Chain
+        { address: "0xdce769b847a0a697239777d0b1c7dd33b6012ba0", network: "arbitrum" }, // Arbitrum
+        { address: "0x351f59de4fedbdf7601f5592b93db3b9330c1c1d", network: "polygon" }   // Polygon
+      ];
+
+      const MIN_BALANCE = ethers.parseUnits("1.0", 18); // 1 токен
+
+      for (const contract of tokenContracts) {
+        try {
+          const provider = providers[contract.network];
+          if (!provider) {
+            logger.warn(`Provider not found for network: ${contract.network}`);
+            continue;
+          }
+
+          // Проверка доступности провайдера
+          try {
+            await provider.getBlockNumber(); // Простой запрос для проверки соединения
+          } catch (providerError) {
+            logger.warn(`Provider for ${contract.network} is not responding: ${providerError.message}`);
+            continue;
+          }
+
+          const tokenContract = new ethers.Contract(contract.address, [
+            "function balanceOf(address owner) view returns (uint256)"
+          ], provider);
+
+          const balance = await tokenContract.balanceOf(walletAddress);
+          logger.info(`Balance for ${walletAddress} on ${contract.network}: ${balance.toString()}`);
+          
+          if (balance >= MIN_BALANCE) {
+            logger.info(`Admin token found on ${contract.network} for ${walletAddress}`);
+            return true; // Если найден хотя бы один токен, возвращаем true
+          }
+        } catch (error) {
+          logger.error(`Error checking balance on ${contract.network}: ${error.message}`);
+        }
+      }
+
+      logger.info(`No admin tokens found for ${walletAddress}`);
+      return false; // Если не найдено ни одного токена, возвращаем false
+    } catch (error) {
+      logger.error(`Error in checkAdminTokens: ${error.message}`);
+      return false;
+    }
+  }
+
  /**
   * Обновляет роль пользователя в базе данных
   * @param {number} userId - ID пользователя
@@ -154,7 +202,7 @@ class AuthService {
      
      return result.rows[0].id;
    } catch (error) {
-      logger.error(`Error getting user ID by identity: ${error.message}`);
+      logger.error(`Ошибка при получении ID пользователя по идентификатору: ${error.message}`);
      return null;
    }
  }
--- a/backend/utils/access-check.js
+++ b/backend/utils/access-check.js
@@ -1,5 +1,5 @@
 const db = require('../db');
-const logger = require('./logger');
+const logger = require('../utils/logger');
 const authService = require('../services/auth-service');

 /**
--- a/backend/utils/auth.js
+++ b/backend/utils/auth.js
@@ -5,7 +5,7 @@ const authService = require('../services/auth-service');
 const { USER_ROLES, IDENTITY_TYPES } = require('./constants');

 // Инициализация провайдера
-const provider = new ethers.JsonRpcProvider(process.env.RPC_URL);
+const provider = new ethers.JsonRpcProvider(process.env.RPC_URL_ETH);

 /**
 * Проверяет подпись сообщения
@@ -37,8 +37,8 @@ async function verifySignature(nonce, signature, address) {
 */
 async function checkUserRole(address) {
  try {
-    // Проверяем наличие токена администратора
-    const isAdmin = await authService.checkAdminToken(address);
+    // Проверяем наличие токенов администратора
+    const isAdmin = await authService.checkAdminTokens(address);
    return isAdmin;
  } catch (error) {
    console.error('Error checking user role:', error);
@@ -53,8 +53,8 @@ async function checkUserRole(address) {
 */
 async function checkAccess(walletAddress) {
  try {
-    // Проверяем наличие токена администратора
-    const isAdmin = await authService.checkAdminToken(walletAddress);
+    // Проверяем наличие токенов администратора
+    const isAdmin = await authService.checkAdminTokens(walletAddress);
    
    // Получаем или создаем пользователя
    const userId = await findOrCreateUser(walletAddress);
@@ -81,7 +81,11 @@ async function checkAccess(walletAddress) {
 async function findOrCreateUser(address) {
  try {
    // Проверяем, существует ли пользователь
-    const userResult = await db.query('SELECT * FROM users WHERE LOWER(address) = LOWER($1)', [address]);
+    const userResult = await db.query(`
+      SELECT u.id FROM users u
+      JOIN user_identities ui ON u.id = ui.user_id
+      WHERE ui.identity_type = 'wallet' AND LOWER(ui.identity_value) = LOWER($1)
+    `, [address]);
    
    let userId;
    let isAdmin = false;
@@ -97,12 +101,18 @@ async function findOrCreateUser(address) {
      
      // Создаем пользователя с ролью 'user'
      const newUserResult = await db.query(
-        'INSERT INTO users (address, role_id, created_at) VALUES (LOWER($1), $2, NOW()) RETURNING id',
-        [address, roleId]
+        'INSERT INTO users (role_id, created_at) VALUES ($1, NOW()) RETURNING id',
+        [roleId]
      );
      
      userId = newUserResult.rows[0].id;
      
+      // Добавляем идентификатор кошелька
+      await db.query(
+        'INSERT INTO user_identities (user_id, identity_type, identity_value, created_at) VALUES ($1, $2, $3, NOW())',
+        [userId, 'wallet', address.toLowerCase()]
+      );
+      
      // Проверяем, является ли пользователь администратором
      isAdmin = await checkUserRole(address);
      
@@ -115,25 +125,18 @@ async function findOrCreateUser(address) {
        }
      }
    } else {
-      // Если пользователь найден, получаем его ID и роль
+      // Если пользователь найден, получаем его ID
      userId = userResult.rows[0].id;
      
-      // Проверяем, является ли пользователь администратором по роли
-      const roleResult = await db.query('SELECT name FROM roles WHERE id = $1', [userResult.rows[0].role_id]);
-      isAdmin = roleResult.rows.length > 0 && roleResult.rows[0].name === 'admin';
+      // Проверяем, является ли пользователь администратором
+      isAdmin = await checkUserRole(address);
      
-      // Проверяем, является ли пользователь администратором по токену
-      const isAdminByToken = await checkUserRole(address);
-      
-      // Обновляем роль пользователя, если она изменилась
-      if (isAdminByToken !== isAdmin) {
-        const roleNameToSet = isAdminByToken ? 'admin' : 'user';
+      // Обновляем роль пользователя
+      const roleNameToSet = isAdmin ? 'admin' : 'user';
      const roleToSetResult = await db.query('SELECT id FROM roles WHERE name = $1', [roleNameToSet]);
      if (roleToSetResult.rows.length > 0) {
        const roleIdToSet = roleToSetResult.rows[0].id;
        await db.query('UPDATE users SET role_id = $1 WHERE id = $2', [roleIdToSet, userId]);
-          isAdmin = isAdminByToken;
-        }
      }
    }
    
--- a/frontend/src/components/Navigation.vue
+++ b/frontend/src/components/Navigation.vue
@@ -6,8 +6,6 @@
    
    <div class="navbar-menu">
      <div class="navbar-start">
-        <router-link to="/" class="navbar-item">Главная</router-link>
-        <router-link v-if="isAuthenticated" to="/chat" class="navbar-item">Чат</router-link>
      </div>
      
      <div class="navbar-end">
--- a/frontend/src/router/index.js
+++ b/frontend/src/router/index.js
@@ -1,7 +1,6 @@
 import { createRouter, createWebHistory } from 'vue-router';
 import { useAuthStore } from '../stores/auth';
 import HomeView from '../views/HomeView.vue';
-import ChatView from '../views/ChatView.vue';

 const routes = [
  {
@@ -10,12 +9,6 @@ const routes = [
    component: HomeView,
    meta: { requiresAuth: false },
  },
-  {
-    path: '/chat',
-    name: 'chat',
-    component: ChatView,
-    meta: { requiresAuth: true }
-  }
 ];

 const router = createRouter({
--- a/frontend/src/views/ChatView.vue
+++ b/frontend/src/views/ChatView.vue
@@ -1,243 +0,0 @@
-<template>
-  <div class="chat-view">
-    <div class="chat-container">
-      <h2>Чат с ИИ-ассистентом</h2>
-      <div class="chat-messages" ref="chatMessages">
-        <div
-          v-for="(message, index) in messages"
-          :key="index"
-          :class="['message', message.sender === 'user' ? 'user-message' : 'ai-message']"
-        >
-          <div class="message-content" v-html="message.text"></div>
-          <div class="message-time">{{ formatTime(message.timestamp) }}</div>
-        </div>
-      </div>
-
-      <div class="chat-input">
-        <textarea
-          v-model="userInput"
-          placeholder="Введите ваше сообщение..."
-          @keydown.enter.prevent="sendMessage"
-        ></textarea>
-        <button class="send-btn" @click="sendMessage" :disabled="!userInput.trim() || isLoading">
-          {{ isLoading ? 'Отправка...' : 'Отправить' }}
-        </button>
-      </div>
-    </div>
-  </div>
-</template>
-
-<script setup>
-import { ref, onMounted, watch, nextTick } from 'vue';
-import axios from 'axios';
-import { useAuthStore } from '../stores/auth';
-
-const userInput = ref('');
-const messages = ref([
-  {
-    sender: 'ai',
-    text: 'Привет! Я ИИ-ассистент DApp for Business. Чем я могу помочь вам сегодня?',
-    timestamp: new Date(),
-  },
-]);
-const chatMessages = ref(null);
-const isLoading = ref(false);
-const authStore = useAuthStore();
-
-// Прокрутка чата вниз при добавлении новых сообщений
-watch(
-  messages,
-  () => {
-    nextTick(() => {
-      if (chatMessages.value) {
-        chatMessages.value.scrollTop = chatMessages.value.scrollHeight;
-      }
-    });
-  },
-  { deep: true }
-);
-
-// Проверка прав администратора
-onMounted(async () => {
-  if (authStore.isAdmin) {
-    messages.value.push({
-      sender: 'ai',
-      text: 'Вы имеете права администратора.',
-      timestamp: new Date(),
-    });
-  } else {
-    messages.value.push({
-      sender: 'ai',
-      text: 'У вас нет прав администратора.',
-      timestamp: new Date(),
-    });
-  }
-});
-
-// Функция для отправки сообщения
-async function sendMessage() {
-  if (!userInput.value.trim() || isLoading.value) return;
-
-  // Добавляем сообщение пользователя в чат
-  const userMessage = userInput.value.trim();
-  messages.value.push({
-    sender: 'user',
-    text: userMessage,
-    timestamp: new Date(),
-  });
-
-  userInput.value = '';
-  isLoading.value = true;
-
-  try {
-    console.log('Отправка сообщения:', userMessage);
-    
-    // Отправляем запрос к API
-    const response = await axios.post(
-      '/api/chat/message',
-      {
-        message: userMessage,
-        language: 'ru', // Укажите язык, если необходимо
-      },
-      {
-        withCredentials: true, // Важно для передачи куков
-      }
-    );
-
-    console.log('Ответ от сервера:', response.data);
-
-    // Добавляем ответ от ИИ в чат
-    messages.value.push({
-      sender: 'ai',
-      text: response.data.reply || 'Извините, я не смог обработать ваш запрос.',
-      timestamp: new Date(),
-    });
-  } catch (error) {
-    console.error('Error sending message:', error);
-    messages.value.push({
-      sender: 'ai',
-      text: 'Извините, произошла ошибка при обработке вашего запроса. Пожалуйста, попробуйте позже.',
-      timestamp: new Date(),
-    });
-  } finally {
-    isLoading.value = false;
-  }
-}
-
-// Функция для форматирования времени
-function formatTime(timestamp) {
-  if (!timestamp) return '';
-
-  const date = new Date(timestamp);
-  return date.toLocaleTimeString([], { hour: '2-digit', minute: '2-digit' });
-}
-</script>
-
-<style scoped>
-.chat-view {
-  height: 100%;
-  display: flex;
-  flex-direction: column;
-  padding: 1rem;
-}
-
-.chat-container {
-  flex: 1;
-  display: flex;
-  flex-direction: column;
-  background-color: white;
-  border-radius: 8px;
-  box-shadow: 0 2px 10px rgba(0, 0, 0, 0.1);
-  overflow: hidden;
-  max-width: 800px;
-  margin: 0 auto;
-  width: 100%;
-}
-
-h2 {
-  padding: 1rem;
-  margin: 0;
-  border-bottom: 1px solid #eee;
-  font-size: 1.5rem;
-  color: #333;
-}
-
-.chat-messages {
-  flex: 1;
-  overflow-y: auto;
-  padding: 1rem;
-  display: flex;
-  flex-direction: column;
-  gap: 1rem;
-  min-height: 400px;
-}
-
-.message {
-  max-width: 80%;
-  padding: 0.75rem 1rem;
-  border-radius: 8px;
-  position: relative;
-}
-
-.user-message {
-  align-self: flex-end;
-  background-color: #e3f2fd;
-  color: #0d47a1;
-}
-
-.ai-message {
-  align-self: flex-start;
-  background-color: #f5f5f5;
-  color: #333;
-}
-
-.message-content {
-  word-break: break-word;
-}
-
-.message-time {
-  font-size: 0.75rem;
-  color: #999;
-  margin-top: 0.25rem;
-  text-align: right;
-}
-
-.chat-input {
-  display: flex;
-  padding: 1rem;
-  border-top: 1px solid #eee;
-  background-color: white;
-}
-
-textarea {
-  flex: 1;
-  padding: 0.75rem;
-  border: 1px solid #ddd;
-  border-radius: 4px;
-  resize: none;
-  height: 60px;
-  font-family: inherit;
-  font-size: 1rem;
-}
-
-.send-btn {
-  margin-left: 0.5rem;
-  padding: 0 1.5rem;
-  background-color: #1976d2;
-  color: white;
-  border: none;
-  border-radius: 4px;
-  cursor: pointer;
-  font-size: 1rem;
-  transition: background-color 0.2s;
-}
-
-.send-btn:hover {
-  background-color: #1565c0;
-}
-
-.send-btn:disabled {
-  background-color: #ccc;
-  cursor: not-allowed;
-}
-</style>
--- a/frontend/src/views/ConversationsView.vue
+++ b/frontend/src/views/ConversationsView.vue
@@ -1,77 +0,0 @@
-<template>
-  <div class="conversations-view">
-    <conversation-list @select-conversation="handleSelectConversation" />
-
-    <div v-if="selectedConversationId" class="conversation-content">
-      <message-thread :conversation-id="selectedConversationId" ref="messageThread" />
-      <message-input @send-message="handleSendMessage" />
-    </div>
-
-    <div v-else class="empty-conversation">
-      <div class="empty-state">
-        <h3>Выберите диалог</h3>
-        <p>
-          Выберите существующий диалог или создайте новый, чтобы начать общение с ИИ-ассистентом.
-        </p>
-      </div>
-    </div>
-  </div>
-</template>
-
-<script setup>
-import { ref } from 'vue';
-import ConversationList from '../components/chat/ConversationList.vue';
-import MessageThread from '../components/chat/MessageThread.vue';
-import MessageInput from '../components/chat/MessageInput.vue';
-
-const selectedConversationId = ref(null);
-const messageThread = ref(null);
-
-function handleSelectConversation(id) {
-  selectedConversationId.value = id;
-}
-
-function handleSendMessage(message) {
-  if (messageThread.value) {
-    messageThread.value.sendMessage(message);
-  }
-}
-</script>
-
-<style scoped>
-.conversations-view {
-  display: flex;
-  height: calc(100vh - 64px); /* Высота экрана минус высота навигации */
-}
-
-.conversation-content {
-  flex-grow: 1;
-  display: flex;
-  flex-direction: column;
-  height: 100%;
-}
-
-.empty-conversation {
-  flex-grow: 1;
-  display: flex;
-  align-items: center;
-  justify-content: center;
-  background-color: #f9f9f9;
-}
-
-.empty-state {
-  text-align: center;
-  max-width: 400px;
-  padding: 2rem;
-}
-
-.empty-state h3 {
-  margin-bottom: 1rem;
-  color: #333;
-}
-
-.empty-state p {
-  color: #666;
-  line-height: 1.5;
-}
-</style>
--- a/frontend/src/views/HomeView.vue
+++ b/frontend/src/views/HomeView.vue
@@ -108,6 +108,21 @@ onMounted(() => {
  } else {
    userLanguage.value = userLang.split('-')[0];
  }
+
+  // Проверка прав администратора
+  if (auth.isAdmin) {
+    messages.value.push({
+      sender: 'ai',
+      text: 'Вы имеете права администратора.',
+      timestamp: new Date(),
+    });
+  } else {
+    messages.value.push({
+      sender: 'ai',
+      text: 'У вас нет прав администратора.',
+      timestamp: new Date(),
+    });
+  }
 });

 // Функция для отправки сообщения
--- a/frontend/src/views/TokenAccess.vue
+++ b/frontend/src/views/TokenAccess.vue
@@ -0,0 +1,75 @@
+<template>
+    <div class="token-access">
+      <h1>Настройка прав доступа</h1>
+      <div>
+        <label for="blockchain">Выберите блокчейн:</label>
+        <select v-model="selectedBlockchain" id="blockchain">
+          <option v-for="(blockchain, index) in blockchains" :key="index" :value="blockchain.value">
+            {{ blockchain.name }}
+          </option>
+        </select>
+      </div>
+      <form @submit.prevent="checkTokenBalance">
+        <div>
+          <label for="contractAddress">Адрес смарт-контракта:</label>
+          <input v-model="contractAddress" id="contractAddress" placeholder="Введите адрес смарт-контракта" required />
+        </div>
+        <div>
+          <label for="requiredAmount">Объем токенов:</label>
+          <input v-model="requiredAmount" type="number" id="requiredAmount" placeholder="Введите объем токенов" required />
+        </div>
+        <button type="submit">Проверить баланс</button>
+      </form>
+      <p v-if="message">{{ message }}</p>
+    </div>
+  </template>
+  
+  <script>
+  export default {
+    data() {
+      return {
+        selectedBlockchain: 'mainnet', // Значение по умолчанию
+        blockchains: [
+          { name: 'Ethereum', value: 'mainnet' },
+          { name: 'Polygon', value: 'polygon' },
+          { name: 'Binance Smart Chain', value: 'bsc' },
+          { name: 'Arbitrum', value: 'arbitrum' }
+        ],
+        contractAddress: '',
+        requiredAmount: '',
+        message: ''
+      };
+    },
+    methods: {
+      async checkTokenBalance() {
+        const response = await fetch(`/api/admin/check-balance`, {
+          method: 'POST',
+          headers: {
+            'Content-Type': 'application/json'
+          },
+          body: JSON.stringify({
+            walletAddress: this.$store.state.walletAddress, // Предполагается, что адрес кошелька хранится в хранилище
+            contractAddress: this.contractAddress,
+            requiredAmount: this.requiredAmount,
+            blockchain: this.selectedBlockchain
+          })
+        });
+  
+        const data = await response.json();
+        this.message = data.message;
+      }
+    }
+  };
+  </script>
+  
+  <style scoped>
+  .token-access {
+    max-width: 400px;
+    margin: auto;
+    padding: 20px;
+    border: 1px solid #ccc;
+    border-radius: 5px;
+    background-color: #f9f9f9;
+  }
+  </style>
+