api.pawol.nu/src/api/parole/controllers/parole.js

'use strict';

const { createCoreController } = require('@strapi/strapi').factories;

const VALID_LANGS = new Set(['fr', 'en', 'es', 'de', 'it'])

module.exports = createCoreController('api::parole.parole', ({strapi}) => ({
  async export(ctx) {
    const { type = 'pairs', lang, format = 'jsonl' } = ctx.query

    const langs = lang
      ? lang.split(',').map(l => l.trim()).filter(l => VALID_LANGS.has(l))
      : null

    if (lang && (!langs || langs.length === 0)) {
      return ctx.badRequest('Langue(s) invalide(s). Valeurs acceptées : fr, en, es, de, it.')
    }

    if (!['pairs', 'instruct'].includes(type)) {
      return ctx.badRequest('type invalide. Valeurs acceptées : pairs, instruct.')
    }

    const paroles = await strapi.service('api::parole.parole').fetchAllParoles()
    const { metadata, pairs } = strapi.service('api::parole.parole').buildExport(paroles, type, langs)

    if (format === 'json') {
      return ctx.send({ metadata, data: pairs })
    }

    // JSONL : première ligne = métadonnées, suivies des exemples d'entraînement.
    // Pour filtrer la ligne de métadonnées : jq 'select(._metadata | not)'
    const lines = [
      JSON.stringify({ _metadata: true, ...metadata }),
      ...pairs.map(p => JSON.stringify(p)),
    ]

    ctx.set('Content-Type', 'application/x-ndjson')
    ctx.set('Content-Disposition', `attachment; filename="pawol-nu-export-${Date.now()}.jsonl"`)
    ctx.body = lines.join('\n')
  },

  async bulkTranslate(ctx) {
    const result = await strapi.service('api::parole.parole').bulkTranslateMissing()
    return ctx.send(result)
  },

  async findOne(ctx) {
    const {id: documentId} = ctx.params
    const parole = await strapi.documents('api::parole.parole').findOne({
      documentId,
      populate: ['artistes']
    })

    return parole
  },
  async update(ctx) {
    const {body} = ctx.request
    const {data} = body

    const updatedParole = await strapi.documents('api::parole.parole').update({
      documentId: data.documentId,

      data: {
        titre: data.titre,
        transcription: data.transcription,
        traductions: data.traductions,
        traductionAuto: data.traductionAuto,
        artistes: data.artistes
      }
    })

    return updatedParole
  },
  async create(ctx) {
    const {body} = ctx.request
    const {data} = body

    if (!data?.user?.documentId || !data?.artistes?.[0]?.documentId) {
      return ctx.badRequest('Informations manquantes.')
    }

    strapi.service('api::parole.parole').validateParoles(data.titre, data.transcription)

    const user = await strapi.documents('plugin::users-permissions.user').findOne({
      documentId: body.data.user.documentId
    })

    if (!user) {
      return ctx.notFound('Utilisateur introuvable.')
    }

    if (user.id !== data.user.id || user.username !== data.user.username || user.email !== data.user.email) {
      return ctx.badRequest('Informations non valides.')
    }

    const artiste = await strapi.documents('api::artiste.artiste').findOne({
      documentId: data.artistes[0].documentId
    })

    if (!artiste) {
      return ctx.notFound('Artiste introuvable.')
    }

    const currentUserParole = await strapi.documents('api::parole.parole').findMany({
      fields: ['id'],
      filters: {
        user: {
          id: {
            $eq: user.id
          }
        },
        publishedAt: {
          $eq: null
        }
      }
    })


    if (user && user.canAutoTranslate && data.traductionAuto && data.traductions.francais && (!data.traductions.anglais || !data.traductions.espagnol || !data.traductions.allemand || !data.traductions.italien)) {
      const translated = await strapi.service('api::parole.parole').translateLyrics(data.traductions.francais)
      data.traductions = translated
    }

    const newParole = await strapi.documents('api::parole.parole').create({
      data: {
        titre: data.titre,
        transcription: data.transcription,
        traductions: data.traductions,
        traductionAuto: data.traductionAuto,
        artistes: [artiste.id],
        user: user.id
      }
    })

    const parolesIds = currentUserParole.map(({id}) => id)
    parolesIds.push(newParole.id)

    await strapi.documents('plugin::users-permissions.user').update({
      documentId: user.documentId,

      data: {
        paroles: parolesIds
      }
    })

    return newParole
  }
}))