video-maker/frontend/src/api.ts

import axios from "axios";
import { apiUrl } from "@/env";
import type { IUserProfile, IUserProfileUpdate, IUserProfileCreate, Video, VideoCreate, ArticleCreate, ImageDownload } from "@/interfaces";

function authHeaders(token: string) {
  return {
    headers: {
      Authorization: `Bearer ${token}`,
    },
  };
}

export const api = {
  async logInGetToken(username: string, password: string) {
    const params = new URLSearchParams();
    params.append("username", username);
    params.append("password", password);

    return axios.post(`${apiUrl}/api/v1/login/access-token`, params);
  },
  async getMe(token: string) {
    return axios.get<IUserProfile>(`${apiUrl}/api/v1/users/me`, authHeaders(token));
  },
  async updateMe(token: string, data: IUserProfileUpdate) {
    return axios.put<IUserProfile>(
      `${apiUrl}/api/v1/users/me`,
      data,
      authHeaders(token),
    );
  },
  async getUsers(token: string) {
    return axios.get<IUserProfile[]>(`${apiUrl}/api/v1/users/`, authHeaders(token));
  },
  async updateUser(token: string, userId: number, data: IUserProfileUpdate) {
    return axios.put(`${apiUrl}/api/v1/users/${userId}`, data, authHeaders(token));
  },
  async createUser(token: string, data: IUserProfileCreate) {
    return axios.post(`${apiUrl}/api/v1/users/create`, data, authHeaders(token));
  },
  async passwordRecovery(email: string) {
    return axios.post(`${apiUrl}/api/v1/password-recovery/${email}`);
  },
  async resetPassword(token: string, password: string) {
    return axios.post(`${apiUrl}/api/v1/reset-password/`, {
      token: token,
      new_password: password,
    });
  },
  async registerUser(data: IUserProfileCreate) {
    return axios.post(`${apiUrl}/api/v1/users/open`, data);
  },
  async testCeleryMsg(token: string, data: { msg: string }) {
    return axios.post<{ msg: string }>(`${apiUrl}/api/v1/utils/test-celery/msg`, data, authHeaders(token));
  },
  async testCeleryFile(token: string, file: File) {
    const formData = new FormData();
    formData.append("file", file)
    return axios.post<{ msg: string }>(`${apiUrl}/api/v1/utils/test-celery/file`, formData, authHeaders(token));
  },
  async uploadPlot(token: string, video_data: VideoCreate, file: File) {
    const formData = new FormData();
    formData.append("title", video_data.title)
    formData.append("anchor_id", video_data.anchor_id.toString())
    formData.append("lang_id", video_data.lang_id.toString())
    formData.append("upload_file", file)
    return axios.post<{ msg: string }>(`${apiUrl}/api/v1/videos/`, formData, authHeaders(token));
  },
  async uploadImage(token: string, file: File[]) {
    const formData = new FormData();
    for (let i = 0; i < file.length; i++) {
      const element = file[i];
      formData.append("upload_files", element);
    }
    return axios.post<{ filenames: string[] }>(`${apiUrl}/api/v1/images/sr`, formData, authHeaders(token));
  },
  // async getImage(token: string, data: ImageDownload) {
  //   axios({
  //     url: `${apiUrl}/api/v1/images/sr?stored_file_name=${data.stored_file_name}&file_name=${data.file_name}`,
  //     method: 'GET',
  //     responseType: 'blob',
  //     headers: {
  //       'Authorization': `Bearer ${token}`
  //     },
  //   }).then((response) => {
  //     const href = URL.createObjectURL(response.data);
  //     const link = document.createElement('a');
  //     link.href = href;
  //     link.setAttribute('download', `${data.file_name}_hr.png`);
  //     document.body.appendChild(link);
  //     link.click();
  //     document.body.removeChild(link);
  //     URL.revokeObjectURL(href);
  //   });
  // },
  async downloadImage(token: string, data: ImageDownload) {
    return axios({
      url: `${apiUrl}/api/v1/images/sr?stored_file_name=${data.stored_file_name}&file_name=${data.file_name}`,
      method: 'GET',
      responseType: 'blob',
      headers: {
        'Authorization': `Bearer ${token}`
      },
    }).then((response) => {
      return response;
    });
  },
  async uploadArticle(token: string, article_data: ArticleCreate) {
    const formData = new FormData();
    formData.append("title", article_data.title)
    formData.append("link", article_data.link)
    formData.append("content", article_data.content)
    return axios.post<{ msg: string }>(`${apiUrl}/api/v1/reputations/`, formData, authHeaders(token));
  },
  async getVideos(token: string) {
    return axios.get<Video[]>(`${apiUrl}/api/v1/videos/`, authHeaders(token));
  },
  async googleLogin(username: string) {
    const params = new URLSearchParams();
    params.append("username", username);
    params.append("password", "google");

    return axios.post(`${apiUrl}/api/v1/login/google/access-token`, params);
  },
};