나만의 블로그를 만들어보자

const fs = require('fs').promises
const path = require('path')
const axios = require('axios')

// 설정
const VELOG_USERNAME = '여기에_본인의_velog_사용자명' // 본인 velog 사용자명으로 변경하세요
const OUTPUT_DIR = path.join(__dirname, 'content/posts')

/**
 * velog GraphQL API를 사용하여 사용자의 모든 글 목록을 가져옵니다
 */
async function fetchAllPostsList() {
  const query = `
    query Posts($username: String) {
      posts(username: $username) {
        id
        title
        url_slug
        released_at
        updated_at
        tags
        short_description
      }
    }
  `

  try {
    const response = await axios({
      url: 'https://v2.velog.io/graphql',
      method: 'post',
      data: {
        query,
        variables: { username: VELOG_USERNAME },
      },
    })

    if (response.data.errors) {
      console.error('GraphQL 오류:', response.data.errors)
      return []
    }

    return response.data.data.posts
  } catch (error) {
    console.error('글 목록 가져오기 실패:', error.message)
    return []
  }
}

/**
 * 특정 글의 전체 내용을 가져옵니다
 */
async function fetchPostContent(postId) {
  const query = `
    query Post($id: ID!) {
      post(id: $id) {
        id
        title
        released_at
        updated_at
        tags
        body
        short_description
        is_markdown
        is_private
      }
    }
  `

  try {
    const response = await axios({
      url: 'https://v2.velog.io/graphql',
      method: 'post',
      data: {
        query,
        variables: { id: postId },
      },
    })

    if (response.data.errors) {
      console.error(`글 내용 가져오기 오류 (ID: ${postId}):`, response.data.errors)
      return null
    }

    return response.data.data.post
  } catch (error) {
    console.error(`글 내용 가져오기 실패 (ID: ${postId}):`, error.message)
    return null
  }
}

/**
 * Markdown 파일로 저장 (요청한 형식으로)
 */
async function saveAsMarkdown(post) {
  // 요청된 형식으로 frontmatter 작성
  const frontMatter = `---
    title: ${post.title.replace(/'/g, "\\'")}
    date: '${new Date(post.released_at).toISOString().split('T')[0]}'
    tags: [${post.tags.map((tag) => `'${tag}'`).join(', ')}]
    draft: false
    summary: ${post.short_description?.replace(/'/g, "\\'") || ''}
    ---`

  const content = `${frontMatter}\n\n${post.body}`
  // frontmatter와 본문 사이에 빈 줄 추가

  // 파일명은 velog의 slug 사용
  const slug = post.url_slug || sanitizeFilename(post.title)
  const fileName = `${slug}.mdx`
  const filePath = path.join(OUTPUT_DIR, fileName)

  await fs.writeFile(filePath, content, 'utf-8')
  return fileName
}

/**
 * 파일명으로 사용할 수 없는 문자 제거
 */
function sanitizeFilename(name) {
  return name
    .toLowerCase()
    .replace(/[^\w\s-]/g, '') // 특수문자 제거
    .replace(/\s+/g, '-') // 공백을 대시로 변환
    .replace(/-+/g, '-') // 연속된 대시 하나로 변환
}

/**
 * 메인 함수
 */
async function main() {
  console.log(`👉 ${VELOG_USERNAME}님의 velog 글을 가져오는 중...`)

  // 출력 디렉토리 생성
  await fs.mkdir(OUTPUT_DIR, { recursive: true })

  // 글 목록 가져오기
  const postsList = await fetchAllPostsList()
  if (postsList.length === 0) {
    console.log('❌ 가져올 글이 없거나 오류가 발생했습니다.')
    return
  }

  console.log(`📋 총 ${postsList.length}개의 글을 찾았습니다.`)

  // 각 글의 내용 가져와서 저장
  let successCount = 0
  let failCount = 0

  for (let i = 0; i < postsList.length; i++) {
    const post = postsList[i]
    console.log(`🔄 (${i + 1}/${postsList.length}) "${post.title}" 처리 중...`)

    // 글 상세 내용 가져오기
    const postDetail = await fetchPostContent(post.id)

    if (postDetail) {
      try {
        const fileName = await saveAsMarkdown({ ...post, ...postDetail })
        console.log(`✅ 저장 완료: ${fileName}`)
        successCount++

        // API 호출 간격 조절 (서버 부하 방지)
        await new Promise((resolve) => setTimeout(resolve, 500))
      } catch (error) {
        console.error(`❌ 저장 실패 (${post.title}):`, error.message)
        failCount++
      }
    } else {
      failCount++
    }
  }

  // 메타데이터 저장
  const metadata = {
    username: VELOG_USERNAME,
    crawled_at: new Date().toISOString(),
    total_posts: postsList.length,
    success_count: successCount,
    fail_count: failCount,
  }

  await fs.writeFile(
    path.join(OUTPUT_DIR, '_metadata.json'),
    JSON.stringify(metadata, null, 2),
    'utf-8'
  )

  console.log('\n===== 크롤링 완료 =====')
  console.log(`📊 총 ${postsList.length}개 중 ${successCount}개 성공, ${failCount}개 실패`)
  console.log(`📁 저장 위치: ${OUTPUT_DIR}`)
}

// 실행
main().catch((error) => {
  console.error('❌ 오류 발생:', error)
})