refactor generate button backend

Author: sakuyacatcat

Makefile

@@ -14,6 +14,10 @@ clean:
 test:
 	yarn test
 
+.PHONY: test_watch
+test_watch:
+	yarn test:watch
+
 .PHONY: test_coverage
 test_coverage:
 	yarn test:coverage

app/api/frames/createStamp/route.tsx

@@ -1,12 +1,15 @@
 import { Button, createFrames } from 'frames.js/next';
-import { generateImageUrl } from '../../../service/stampService';
+import { StampService } from '../../../service/stampService';
 
 const frames = createFrames({
   basePath: '/api/frames/createStamp',
 });
 
 const handleRequest = frames(async (ctx) => {
-  const image_url = await generateImageUrl(ctx.message?.inputText);
+  const template =
+    'スタイルはキティちゃんのようにシンプルで愛らしいキャラクターのLINEスタンプを生成してください。ポーズはセリフに合わせて適切に変化させてください。セリフ：';
+  const service = new StampService(template);
+  const image_url = await service.generateImageUrl(ctx.message?.inputText);
 
   return {
     image: image_url,

app/repository/stampRepository.ts

@@ -1,19 +1,21 @@
 import OpenAI from 'openai';
 
-const openai = new OpenAI();
+export class StampRepository {
+  constructor(private openai = new OpenAI()) {}
 
-export async function generateImage(prompt: string) {
-  try {
-    const imageResponse = await openai.images.generate({
-      model: 'dall-e-3',
-      prompt: prompt,
-      n: 1,
-      size: '1024x1024',
-    });
-    const imageUrl = imageResponse.data[0]?.url || '';
-    return imageUrl;
-  } catch (error) {
-    console.error('Error generating image:', error);
-    throw error;
+  async generateImage(prompt: string): Promise<string> {
+    try {
+      const imageResponse = await this.openai.images.generate({
+        model: 'dall-e-3',
+        prompt: prompt,
+        n: 1,
+        size: '1024x1024',
+      });
+      const imageUrl = imageResponse.data[0]?.url || '';
+      return imageUrl;
+    } catch (error) {
+      console.error('Error generating image:', error);
+      throw new Error('Image generation failed');
+    }
   }
 }

app/service/stampService.ts

@@ -1,12 +1,19 @@
-import { generateImage } from '../repository/stampRepository';
+// stampService.js
+import { StampRepository } from '../repository/stampRepository';
 
-export async function generateImageUrl(input: string): Promise<string> {
-  const template =
-    'スタイルはキティちゃんのようにシンプルで愛らしいキャラクターのLINEスタンプを生成してください。ポーズはセリフに合わせて適切に変化させてください。セリフ：';
+export class StampService {
+  private template: string;
 
-  const prompt = template + input;
+  constructor(
+    template: string,
+    private stampRepository: StampRepository = new StampRepository()
+  ) {
+    this.template = template;
+  }
 
-  const imageUrl = await generateImage(prompt);
-
-  return imageUrl;
+  async generateImageUrl(input: string): Promise<string> {
+    const prompt = this.template + input;
+    const imageUrl = await this.stampRepository.generateImage(prompt);
+    return imageUrl;
+  }
 }

tests/stampRepository.test.ts_tmp

@@ -0,0 +1,54 @@
+// import OpenAI from 'openai';
+// import { StampRepository } from '../app/repository/stampRepository';
+
+// jest.mock('openai', () => {
+//   return {
+//     OpenAI: jest.fn().mockImplementation(() => {
+//       return {
+//         images: {
+//           generate: jest.fn().mockResolvedValue({
+//             data: [
+//               {
+//                 url: 'mockImageUrl',
+//               },
+//             ],
+//           }),
+//         },
+//       };
+//     }),
+//   };
+// });
+
+// describe('StampRepository', () => {
+//   let stampRepository: StampRepository;
+
+//   beforeEach(() => {
+//     stampRepository = new StampRepository();
+//   });
+
+//   it('should return an image URL when generateImage is called', async () => {
+//     const prompt = 'test prompt';
+//     const imageUrl = await stampRepository.generateImage(prompt);
+
+//     // 正しい画像URLが返されることを確認
+//     expect(imageUrl).toBe('mockImageUrl');
+//     // OpenAI.images.generateが適切な引数で呼び出されたことを確認
+//     expect(OpenAI.prototype.images.generate).toHaveBeenCalledWith({
+//       model: 'dall-e-3',
+//       prompt: prompt,
+//       n: 1,
+//       size: '1024x1024',
+//     });
+//   });
+
+//   it('should throw an error if the OpenAI API call fails', async () => {
+//     // OpenAI.images.generate メソッドを失敗させるようにモックを設定
+//     OpenAI.prototype.images.generate.mockRejectedValue(
+//       new Error('API call failed')
+//     );
+
+//     await expect(stampRepository.generateImage('test prompt')).rejects.toThrow(
+//       'Image generation failed'
+//     );
+//   });
+// });