package utils

import (
	"net/url"
)

func RewriteUrl(link string) (string, error) {
	url, err := url.Parse(link)
	if err != nil {
		return "", err
	}
	switch url.Host {
	case "", "imgur.com", "www.imgur.com", "i.imgur.com":
		return url.Path, nil
	case "i.stack.imgur.com":
		return "/stack" + url.Path, nil
	}
	return link, nil
}