Merge pull request #158 from Dogtiti/feature/auth-script

feat: add auth bash

Author: Dogtiti

setup-auth.sh

@@ -0,0 +1,203 @@
+#!/bin/bash
+
+# Color escape sequences
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+NC='\033[0m' # No Color
+
+# Check if we are running in a terminal that supports colors
+if [ -t 1 ]; then
+  # Use colors
+  RED=$(printf '\033[31m')
+  GREEN=$(printf '\033[32m')
+  NC=$(printf '\033[0m')
+fi
+
+# Check if a string matches the "sk-" key pattern
+is_valid_sk_key() {
+  local api_key=$1
+  local pattern="^sk-[a-zA-Z0-9]{48}$"
+  [[ $api_key =~ $pattern ]] && return 0 || return 1
+}
+
+# Set value for NEXT_PUBLIC_WEB_SEARCH_ENABLED
+select_web_search_enabled() {
+  PS3="${GREEN}Do you want to enable web search?${NC} "
+  options=("true" "false")
+  select opt in "${options[@]}"; do
+    case $opt in
+      "true")
+        NEXT_PUBLIC_WEB_SEARCH_ENABLED=true
+        read_variable "${GREEN}Enter your SERP API Key (required):${NC} " "^.+$" "SERP_API_KEY"
+        break
+        ;;
+      "false")
+        NEXT_PUBLIC_WEB_SEARCH_ENABLED=false
+        break
+        ;;
+      *) echo "${RED}Please enter a valid option.${NC}" ;;
+    esac
+  done
+}
+
+# Ask for user input and validate variable values
+read_variable() {
+  local prompt="$1"
+  local pattern="$2"
+  local var_name="$3"
+  local var_val=""
+  while true; do
+    read -p "$prompt" var_val
+    if [[ -z "$var_val" ]]; then
+      echo -e "${RED}Error: Please enter a valid value for $var_name.${NC}"
+    elif [[ ! $var_val =~ $pattern ]]; then
+      echo -e "${RED}Error: Invalid format for $var_name.${NC}"
+    else
+      eval "$var_name=$var_val"
+      echo -e "${GREEN}$var_name set. ✔${NC}"
+      break
+    fi
+  done
+}
+
+# Get user input for OPENAI_API_KEY
+read_variable "${GREEN}Enter your OpenAI API Key (required):${NC} " "^sk-[a-zA-Z0-9]{48}$" "OPENAI_API_KEY"
+
+# Get user input for DATABASE_URL
+read_variable "${GREEN}Enter your database URL (required):${NC} " "^.+$" "DATABASE_URL"
+
+# Get user input for GITHUB_CLIENT_ID
+read_variable "${GREEN}Enter your Github Client ID (required):${NC} " "^.+$" "GITHUB_CLIENT_ID"
+
+# Get user input for GITHUB_CLIENT_SECRET
+read_variable "${GREEN}Enter your Github Client Secret (required):${NC} " "^.+$" "GITHUB_CLIENT_SECRET"
+
+# Get user input for NEXT_PUBLIC_WEB_SEARCH_ENABLED
+select_web_search_enabled
+
+echo -e "${GREEN}All required variables set. ✔${NC}"
+
+# Generate a random string for NEXTAUTH_SECRET
+NEXTAUTH_SECRET=$(openssl rand -base64 32)
+
+# Disable guest mode when auth enable
+NEXT_PUBLIC_GUEST_KEY=''
+
+ENV="NEXTAUTH_SECRET=$NEXTAUTH_SECRET\n\
+OPENAI_API_KEY=$OPENAI_API_KEY\n\
+DATABASE_URL=${DATABASE_URL}\n\
+GITHUB_CLIENT_ID=${GITHUB_CLIENT_ID}\n\
+GITHUB_CLIENT_SECRET=${GITHUB_CLIENT_SECRET}\n\
+NEXT_PUBLIC_GUEST_KEY=${NEXT_PUBLIC_GUEST_KEY}\n\
+NEXTAUTH_SECRET=${NEXTAUTH_SECRET}\n\
+SERP_API_KEY=$SERP_API_KEY\n\
+NEXT_PUBLIC_WEB_SEARCH_ENABLED=$NEXT_PUBLIC_WEB_SEARCH_ENABLED\n"
+
+printf $ENV > .env.auth
+
+
+# 模型定义
+modelDefinitions="generator client {
+  provider = \"prisma-client-js\"
+}
+
+datasource db {
+  provider     = \"mysql\"
+  url          = env(\"DATABASE_URL\")
+  relationMode = \"prisma\"
+}
+
+// Necessary for Next auth
+model Account {
+  id                String  @id @default(cuid())
+  userId            String
+  type              String
+  provider          String
+  providerAccountId String
+  refresh_token     String? @db.Text
+  access_token      String? @db.Text
+  expires_at        Int?
+  token_type        String?
+  scope             String?
+  id_token          String? @db.Text
+  session_state     String?
+  user              User    @relation(fields: [userId], references: [id], onDelete: Cascade)
+
+  @@unique([provider, providerAccountId])
+  @@index([userId])
+}
+
+model Session {
+  id           String   @id @default(cuid())
+  sessionToken String   @unique
+  userId       String
+  expires      DateTime
+  user         User     @relation(fields: [userId], references: [id], onDelete: Cascade)
+
+  @@index([userId])
+}
+
+model User {
+  id             String    @id @default(cuid())
+  name           String?
+  email          String?   @unique
+  emailVerified  DateTime?
+  image          String?
+  role           String?
+  subscriptionId String?   @db.Text
+  customerId     String?   @db.Text
+
+  createDate DateTime @default(now())
+
+  accounts Account[]
+  sessions Session[]
+  Agent    Agent[]
+
+  @@index([email])
+}
+
+model VerificationToken {
+  identifier String
+  token      String   @unique
+  expires    DateTime
+
+  @@unique([identifier, token])
+}
+
+model Agent {
+  id     String @id @default(cuid())
+  userId String
+  name   String @db.Text
+  goal   String @db.Text
+
+  deleteDate DateTime?
+  createDate DateTime  @default(now())
+
+  user  User        @relation(fields: [userId], references: [id], onDelete: Cascade)
+  tasks AgentTask[]
+
+  @@index([userId, deleteDate, createDate])
+}
+
+model AgentTask {
+  id      String  @id @default(cuid())
+  taskId  String?
+  parentTaskId String?
+  agentId String
+  type    String
+  status  String?
+  value   String  @db.Text
+  info    String? @db.Text
+  sort    Int
+
+  deleteDate DateTime?
+  createDate DateTime  @default(now())
+
+  agent Agent @relation(fields: [agentId], references: [id], onDelete: Cascade)
+
+  @@index([agentId])
+  @@index([type])
+}"
+
+# Generate prisma schema
+echo "$modelDefinitions" > prisma/schema.prisma