actions.go

/*
 * wp-import imports posts from WordPress into Write.as / WriteFreely.
 * Copyright © 2019, 2024 Musing Studio LLC.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 */

package wpimport

import (
	"bytes"
	"fmt"
	"github.com/frankbille/go-wxr-import"
	"github.com/writeas/go-writeas/v2"
	"github.com/writeas/godown"
	"github.com/writeas/wp-import/core"
	"log"
	"os"
	"regexp"
)

var (
	commentReg = regexp.MustCompile("(?m)<!-- ([^m ]|m[^o ]|mo[^r ]|mor[^e ])+ -->\n?")
)

func ImportWordPress(dstBlog string, raw []byte) error {
	d := wxr.ParseWxr(raw)
	log.Printf("Found %d channels.\n", len(d.Channels))

	postsCount := 0

	for _, ch := range d.Channels {
		log.Printf("Channel: %s\n", ch.Title)

		/*
			// Create the blog
			c := &writeas.CollectionParams{
				Title:       ch.Title,
				Description: ch.Description,
			}
			log.Printf("Creating %s...\n", ch.Title)
			coll, err := core.Client.CreateCollection(c)
			if err != nil {
				return err
			}
			log.Printf("Done!\n")
		*/

		log.Printf("Found %d items.\n", len(ch.Items))
		for _, wpp := range ch.Items {
			if wpp.PostType != "post" {
				continue
			}

			// Convert to Markdown
			b := bytes.NewBufferString("")
			r := bytes.NewReader([]byte(wpp.Content))
			err := godown.Convert(b, r, nil)
			con := b.String()

			// Remove unneeded WordPress comments that take up space, like <!-- wp:paragraph -->
			con = commentReg.ReplaceAllString(con, "")

			// Append tags
			tags := ""
			sep := ""
			for _, cat := range wpp.Categories {
				if cat.Domain != "post_tag" {
					continue
				}
				tags += sep + "#" + cat.DisplayName
				sep = " "
			}
			if tags != "" {
				con += "\n\n" + tags
			}

			p := &writeas.PostParams{
				Title:      wpp.Title,
				Slug:       wpp.PostName,
				Content:    con,
				Created:    &wpp.PostDateGmt,
				Updated:    &wpp.PostDateGmt,
				Font:       "norm",
				Language:   &ch.Language,
				Collection: dstBlog,
				//Collection: coll.Alias,
			}
			log.Printf("Creating %s", p.Title)
			_, err = core.Client.CreatePost(p)
			if err != nil {
				fmt.Fprintf(os.Stderr, "create post: %s\n", err)
				continue
			}

			postsCount++
		}
		log.Printf("Created %d posts.\n", postsCount)
	}
	return nil
}

func errQuit(m string) {
	fmt.Fprintf(os.Stderr, m+"\n")
	os.Exit(1)
}