@lobehub/chat 1.104.4 → 1.105.0

package/src/app/[variants]/(main)/profile/apikey/page.tsx

@@ -0,0 +1,32 @@
+import { notFound } from 'next/navigation';
+
+import { serverFeatureFlags } from '@/config/featureFlags';
+import { metadataModule } from '@/server/metadata';
+import { translation } from '@/server/translation';
+import { DynamicLayoutProps } from '@/types/next';
+import { RouteVariants } from '@/utils/server/routeVariants';
+
+import Page from '../../settings/system-agent';
+import Client from './Client';
+
+export const generateMetadata = async (props: DynamicLayoutProps) => {
+  const locale = await RouteVariants.getLocale(props);
+  const { t } = await translation('auth', locale);
+  return metadataModule.generate({
+    description: t('header.desc'),
+    title: t('tab.apikey'),
+    url: '/profile/apikey',
+  });
+};
+
+const page = () => {
+  const { showApiKeyManage } = serverFeatureFlags();
+
+  if (!showApiKeyManage) return notFound();
+
+  return <Client />;
+};
+
+Page.displayName = 'ApiKey';
+
+export default page;

package/src/app/[variants]/(main)/profile/hooks/useCategory.tsx

@@ -1,5 +1,5 @@
 import { Icon } from '@lobehub/ui';
-import { ChartColumnBigIcon, ShieldCheck, UserCircle } from 'lucide-react';
+import { ChartColumnBigIcon, KeyIcon, ShieldCheck, UserCircle } from 'lucide-react';
 import Link from 'next/link';
 import { useTranslation } from 'react-i18next';
 
@@ -7,12 +7,14 @@ import type { MenuProps } from '@/components/Menu';
 import { enableAuth } from '@/const/auth';
 import { isDeprecatedEdition } from '@/const/version';
 import { ProfileTabs } from '@/store/global/initialState';
+import { featureFlagsSelectors, useServerConfigStore } from '@/store/serverConfig';
 import { useUserStore } from '@/store/user';
 import { authSelectors } from '@/store/user/slices/auth/selectors';
 
 export const useCategory = () => {
   const { t } = useTranslation('auth');
   const [isLoginWithClerk] = useUserStore((s) => [authSelectors.isLoginWithClerk(s)]);
+  const { showApiKeyManage } = useServerConfigStore(featureFlagsSelectors);
 
   const cateItems: MenuProps['items'] = [
     {
@@ -43,6 +45,15 @@ export const useCategory = () => {
         </Link>
       ),
     },
+    !!showApiKeyManage && {
+      icon: <Icon icon={KeyIcon} />,
+      key: ProfileTabs.APIKey,
+      label: (
+        <Link href={'/profile/apikey'} onClick={(e) => e.preventDefault()}>
+          {t('tab.apikey')}
+        </Link>
+      ),
+    },
   ].filter(Boolean) as MenuProps['items'];
 
   return cateItems;

package/src/app/[variants]/(main)/settings/_layout/Desktop/index.tsx

@@ -36,7 +36,7 @@ const Layout = memo<LayoutProps>(({ children, category }) => {
       height={'100%'}
       horizontal={md}
       ref={ref}
-      style={{ background: theme.colorBgContainer, flex: '1', position: 'relative', width: '0' }}
+      style={{ background: theme.colorBgContainer, flex: '1', position: 'relative' }}
     >
       {md ? (
         <SideBar>{category}</SideBar>

package/src/config/featureFlags/schema.ts

@@ -16,6 +16,9 @@ export const FeatureFlagsSchema = z.object({
   openai_api_key: z.boolean().optional(),
   openai_proxy_url: z.boolean().optional(),
 
+  // profile
+  api_key_manage: z.boolean().optional(),
+
   create_session: z.boolean().optional(),
   edit_agent: z.boolean().optional(),
 
@@ -56,6 +59,8 @@ export const DEFAULT_FEATURE_FLAGS: IFeatureFlags = {
   openai_api_key: true,
   openai_proxy_url: true,
 
+  api_key_manage: false,
+
   create_session: true,
   edit_agent: true,
 
@@ -97,6 +102,8 @@ export const mapFeatureFlagsEnvToState = (config: IFeatureFlags) => {
     showOpenAIApiKey: config.openai_api_key,
     showOpenAIProxyUrl: config.openai_proxy_url,
 
+    showApiKeyManage: config.api_key_manage,
+
     enablePlugins: config.plugins,
     showDalle: config.dalle,
     showChangelog: config.changelog,

package/src/database/migrations/0029_add_apikey_manage.sql

@@ -0,0 +1,16 @@
+CREATE TABLE IF NOT EXISTS "api_keys" (
+	"id" integer PRIMARY KEY GENERATED BY DEFAULT AS IDENTITY (sequence name "api_keys_id_seq" INCREMENT BY 1 MINVALUE 1 MAXVALUE 2147483647 START WITH 1 CACHE 1),
+	"name" varchar(256) NOT NULL,
+	"key" varchar(256) NOT NULL,
+	"enabled" boolean DEFAULT true,
+	"expires_at" timestamp with time zone,
+	"last_used_at" timestamp with time zone,
+	"user_id" text NOT NULL,
+	"accessed_at" timestamp with time zone DEFAULT now() NOT NULL,
+	"created_at" timestamp with time zone DEFAULT now() NOT NULL,
+	"updated_at" timestamp with time zone DEFAULT now() NOT NULL,
+	CONSTRAINT "api_keys_key_unique" UNIQUE("key")
+);
+--> statement-breakpoint
+ALTER TABLE "rbac_roles" ADD COLUMN "metadata" jsonb DEFAULT '{}'::jsonb;--> statement-breakpoint
+ALTER TABLE "api_keys" ADD CONSTRAINT "api_keys_user_id_users_id_fk" FOREIGN KEY ("user_id") REFERENCES "public"."users"("id") ON DELETE cascade ON UPDATE no action;