mirror of
https://github.com/ArchiveBox/ArchiveBox
synced 2024-11-26 22:20:21 +00:00
111 lines
3.2 KiB
Python
111 lines
3.2 KiB
Python
__package__ = 'archivebox.core'
|
|
|
|
from django.shortcuts import render, redirect
|
|
|
|
from django.http import HttpResponse
|
|
from django.views import View, static
|
|
from django.conf import settings
|
|
|
|
from core.models import Snapshot
|
|
|
|
from ..index import load_main_index, load_main_index_meta
|
|
from ..config import (
|
|
OUTPUT_DIR,
|
|
VERSION,
|
|
FOOTER_INFO,
|
|
PUBLIC_INDEX,
|
|
PUBLIC_SNAPSHOTS,
|
|
)
|
|
from ..util import base_url
|
|
from .. main import add
|
|
|
|
|
|
class MainIndex(View):
|
|
template = 'main_index.html'
|
|
|
|
def get(self, request):
|
|
if not request.user.is_authenticated and not PUBLIC_INDEX:
|
|
return redirect(f'/admin/login/?next={request.path}')
|
|
|
|
all_links = load_main_index(out_dir=OUTPUT_DIR)
|
|
meta_info = load_main_index_meta(out_dir=OUTPUT_DIR)
|
|
|
|
context = {
|
|
'updated': meta_info['updated'],
|
|
'num_links': meta_info['num_links'],
|
|
'links': all_links,
|
|
'VERSION': VERSION,
|
|
'FOOTER_INFO': FOOTER_INFO,
|
|
}
|
|
|
|
return render(template_name=self.template, request=request, context=context)
|
|
|
|
|
|
class AddLinks(View):
|
|
template = 'add_links.html'
|
|
|
|
def get(self, request):
|
|
if not request.user.is_authenticated and not PUBLIC_INDEX:
|
|
return redirect(f'/admin/login/?next={request.path}')
|
|
|
|
context = {}
|
|
|
|
return render(template_name=self.template, request=request, context=context)
|
|
|
|
def post(self, request):
|
|
url = request.POST['url']
|
|
print(f'[+] Adding URL: {url}')
|
|
add(
|
|
import_str=url,
|
|
update_all=False,
|
|
out_dir=OUTPUT_DIR,
|
|
)
|
|
return redirect('/')
|
|
|
|
|
|
class LinkDetails(View):
|
|
def get(self, request, path):
|
|
# missing trailing slash -> redirect to index
|
|
if '/' not in path:
|
|
return redirect(f'{path}/index.html')
|
|
|
|
if not request.user.is_authenticated and not PUBLIC_SNAPSHOTS:
|
|
return redirect(f'/admin/login/?next={request.path}')
|
|
|
|
try:
|
|
slug, archivefile = path.split('/', 1)
|
|
except (IndexError, ValueError):
|
|
slug, archivefile = path.split('/', 1)[0], 'index.html'
|
|
|
|
all_pages = list(Snapshot.objects.all())
|
|
|
|
# slug is a timestamp
|
|
by_ts = {page.timestamp: page for page in all_pages}
|
|
try:
|
|
return static.serve(request, archivefile, by_ts[slug].link_dir, show_indexes=True)
|
|
except KeyError:
|
|
pass
|
|
|
|
# slug is a hash
|
|
by_hash = {page.url_hash: page for page in all_pages}
|
|
try:
|
|
timestamp = by_hash[slug].timestamp
|
|
return redirect(f'/archive/{timestamp}/{archivefile}')
|
|
except KeyError:
|
|
pass
|
|
|
|
# slug is a URL
|
|
by_url = {page.base_url: page for page in all_pages}
|
|
try:
|
|
# TODO: add multiple snapshot support by showing index of all snapshots
|
|
# for given url instead of redirecting to timestamp index
|
|
timestamp = by_url[base_url(path)].timestamp
|
|
return redirect(f'/archive/{timestamp}/index.html')
|
|
except KeyError:
|
|
pass
|
|
|
|
return HttpResponse(
|
|
'No archived link matches the given timestamp or hash.',
|
|
content_type="text/plain",
|
|
status=404,
|
|
)
|